102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import httplib2
|
|
|
|
import mimetypes
|
|
import mimetools
|
|
import itertools
|
|
|
|
class MultiPartForm(object):
|
|
"""Accumulate the data to be used when posting a form."""
|
|
|
|
def __init__(self):
|
|
self.form_fields = []
|
|
self.files = []
|
|
self.boundary = mimetools.choose_boundary()
|
|
return
|
|
|
|
def get_content_type(self):
|
|
return 'multipart/form-data; boundary=%s' % self.boundary
|
|
|
|
def add_field(self, name, value):
|
|
"""Add a simple field to the form data."""
|
|
self.form_fields.append((name, value))
|
|
return
|
|
|
|
def add_file(self, fieldname, filename, fileHandle, mimetype=None):
|
|
"""Add a file to be uploaded."""
|
|
body = fileHandle.read()
|
|
if mimetype is None:
|
|
mimetype = mimetypes.guess_type(filename)[0] or 'application/octet-stream'
|
|
self.files.append((fieldname, filename, mimetype, body))
|
|
return
|
|
|
|
def __str__(self):
|
|
"""Return a string representing the form data, including attached files."""
|
|
# Build a list of lists, each containing "lines" of the
|
|
# request. Each part is separated by a boundary string.
|
|
# Once the list is built, return a string where each
|
|
# line is separated by '\r\n'.
|
|
parts = []
|
|
part_boundary = '--' + self.boundary
|
|
|
|
# Add the form fields
|
|
parts.extend(
|
|
[ part_boundary,
|
|
'Content-Disposition: form-data; name="%s"' % name,
|
|
'',
|
|
value,
|
|
]
|
|
for name, value in self.form_fields
|
|
)
|
|
|
|
# Add the files to upload
|
|
parts.extend(
|
|
[ part_boundary,
|
|
'Content-Disposition: file; name="%s"; filename="%s"' % \
|
|
(field_name, filename),
|
|
'Content-Type: %s' % content_type,
|
|
'',
|
|
body,
|
|
]
|
|
for field_name, filename, content_type, body in self.files
|
|
)
|
|
|
|
# Flatten the list and add closing boundary marker,
|
|
# then return CR+LF separated data
|
|
flattened = list(itertools.chain(*parts))
|
|
flattened.append('--' + self.boundary + '--')
|
|
flattened.append('')
|
|
return '\r\n'.join(flattened)
|
|
|
|
|
|
def get(url):
|
|
h = httplib2.Http()
|
|
resp, content = h.request(url)
|
|
return resp, content
|
|
|
|
def post(url, data):
|
|
h = httplib2.Http()
|
|
form = MultiPartForm()
|
|
|
|
for field, value in data.iteritems():
|
|
if hasattr(value, "read"):
|
|
form.add_file(field, value.name, value)
|
|
else:
|
|
form.add_field(field, str(value))
|
|
|
|
body = str(form)
|
|
headers = {
|
|
'Content-type': form.get_content_type(),
|
|
'Content-length': str(len(body))
|
|
}
|
|
|
|
try:
|
|
response = h.request(url, 'POST', headers=headers, body=body)
|
|
except httplib2.HttpLib2Error, err:
|
|
response = err
|
|
|
|
return response
|
|
|