Ruud
14 years ago
|
Before Width: | Height: | Size: 1015 B |
|
Before Width: | Height: | Size: 451 B After Width: | Height: | Size: 451 B |
|
Before Width: | Height: | Size: 228 B After Width: | Height: | Size: 228 B |
|
After Width: | Height: | Size: 502 B |
|
After Width: | Height: | Size: 624 B |
|
Before Width: | Height: | Size: 159 B After Width: | Height: | Size: 159 B |
|
Before Width: | Height: | Size: 283 B After Width: | Height: | Size: 283 B |
|
After Width: | Height: | Size: 674 B |
|
Before Width: | Height: | Size: 726 B |
@ -0,0 +1,88 @@ |
|||||
|
#!/usr/bin/python |
||||
|
|
||||
|
#### |
||||
|
# 06/2010 Nic Wolfe <nic@wolfeden.ca> |
||||
|
# 02/2006 Will Holcomb <wholcomb@gmail.com> |
||||
|
# |
||||
|
# This library is free software; you can redistribute it and/or |
||||
|
# modify it under the terms of the GNU Lesser General Public |
||||
|
# License as published by the Free Software Foundation; either |
||||
|
# version 2.1 of the License, or (at your option) any later version. |
||||
|
# |
||||
|
# This library is distributed in the hope that it will be useful, |
||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
|
# Lesser General Public License for more details. |
||||
|
# |
||||
|
|
||||
|
import urllib |
||||
|
import urllib2 |
||||
|
import mimetools, mimetypes |
||||
|
import os, sys |
||||
|
|
||||
|
# Controls how sequences are uncoded. If true, elements may be given multiple values by |
||||
|
# assigning a sequence. |
||||
|
doseq = 1 |
||||
|
|
||||
|
class MultipartPostHandler(urllib2.BaseHandler): |
||||
|
handler_order = urllib2.HTTPHandler.handler_order - 10 # needs to run first |
||||
|
|
||||
|
def http_request(self, request): |
||||
|
data = request.get_data() |
||||
|
if data is not None and type(data) != str: |
||||
|
v_files = [] |
||||
|
v_vars = [] |
||||
|
try: |
||||
|
for(key, value) in data.items(): |
||||
|
if type(value) in (file, list, tuple): |
||||
|
v_files.append((key, value)) |
||||
|
else: |
||||
|
v_vars.append((key, value)) |
||||
|
except TypeError: |
||||
|
systype, value, traceback = sys.exc_info() |
||||
|
raise TypeError, "not a valid non-string sequence or mapping object", traceback |
||||
|
|
||||
|
if len(v_files) == 0: |
||||
|
data = urllib.urlencode(v_vars, doseq) |
||||
|
else: |
||||
|
boundary, data = MultipartPostHandler.multipart_encode(v_vars, v_files) |
||||
|
contenttype = 'multipart/form-data; boundary=%s' % boundary |
||||
|
if(request.has_header('Content-Type') |
||||
|
and request.get_header('Content-Type').find('multipart/form-data') != 0): |
||||
|
print "Replacing %s with %s" % (request.get_header('content-type'), 'multipart/form-data') |
||||
|
request.add_unredirected_header('Content-Type', contenttype) |
||||
|
|
||||
|
request.add_data(data) |
||||
|
return request |
||||
|
|
||||
|
@staticmethod |
||||
|
def multipart_encode(vars, files, boundary = None, buffer = None): |
||||
|
if boundary is None: |
||||
|
boundary = mimetools.choose_boundary() |
||||
|
if buffer is None: |
||||
|
buffer = '' |
||||
|
for(key, value) in vars: |
||||
|
buffer += '--%s\r\n' % boundary |
||||
|
buffer += 'Content-Disposition: form-data; name="%s"' % key |
||||
|
buffer += '\r\n\r\n' + value + '\r\n' |
||||
|
for(key, fd) in files: |
||||
|
|
||||
|
# allow them to pass in a file or a tuple with name & data |
||||
|
if type(fd) == file: |
||||
|
name_in = fd.name |
||||
|
fd.seek(0) |
||||
|
data_in = fd.read() |
||||
|
elif type(fd) in (tuple, list): |
||||
|
name_in, data_in = fd |
||||
|
|
||||
|
filename = os.path.basename(name_in) |
||||
|
contenttype = mimetypes.guess_type(filename)[0] or 'application/octet-stream' |
||||
|
buffer += '--%s\r\n' % boundary |
||||
|
buffer += 'Content-Disposition: form-data; name="%s"; filename="%s"\r\n' % (key, filename) |
||||
|
buffer += 'Content-Type: %s\r\n' % contenttype |
||||
|
# buffer += 'Content-Length: %s\r\n' % file_size |
||||
|
buffer += '\r\n' + data_in + '\r\n' |
||||
|
buffer += '--%s--\r\n\r\n' % boundary |
||||
|
return boundary, buffer |
||||
|
|
||||
|
https_request = http_request |
||||