formdata.py

# -*- coding: utf-8 -*-
import uuid
import os
import mimetypes

class FormData:
    def __init__(self):
        self.boundary = uuid.uuid4().hex
        self.out = ''

    def add_text(self, name, text):
        self.out += '--%s\r\n' % self.boundary
        self.out += 'Content-Disposition: form-data; name="%s"\r\n\r\n' % name
        self.out += text + '\r\n'

    def add_file(self, name, filename):
        content = open(filename, 'rb').read()
        self.out += '--%s\r\n' % self.boundary
        self.out += 'Content-Disposition: form-data; name="%s"; filename="%s"\r\n' % (name, os.path.basename(filename))
        extension = os.path.splitext(filename)[1]
        self.out += 'Content-Type: %s\r\n\r\n' % (mimetypes.types_map[extension] if extension in mimetypes.types_map else 'application/octet-stream')
        self.out += content + '\r\n'

    def __str__(self):
        return self.out + '--' + self.boundary + '--'