Использование MultipartPostHandler для POST-данных формы с Python
Проблема: при размещении данных с помощью Python urllib2 все данные кодируются в виде URL и отправляются в виде Content-Type: application/x-www-form-urlencoded. При загрузке файлов вместо Content-Type следует установить multipart / form-data, а содержимое должно быть закодировано в MIME. Обсуждение этой проблемы здесь: http://code.activestate.com/recipes/146306/
Чтобы обойти это ограничение, некоторые проницательные программисты создали библиотеку с именем MultipartPostHandler, которая создает OpenerDirector, который вы можете использовать с urllib2, чтобы в большинстве случаев автоматически выполнять POST с multipart / form-data. Копия этой библиотеки находится здесь: http://peerit.blogspot.com/2007/07/multipartposthandler-doesnt-work-for.html
Я новичок в Python и не могу заставить эту библиотеку работать. Я написал по существу следующий код. Когда я перехватываю его в локальном HTTP-прокси, я вижу, что данные по-прежнему имеют URL-кодировку, а не MIME-кодировку из нескольких частей. Пожалуйста, помогите мне понять, что я делаю неправильно, или лучший способ сделать это. Спасибо:-)
FROM_ADDR = 'my@email.com'
try:
data = open(file, 'rb').read()
except:
print "Error: could not open file %s for reading" % file
print "Check permissions on the file or folder it resides in"
sys.exit(1)
# Build the POST request
url = "http://somedomain.com/?action=analyze"
post_data = {}
post_data['analysisType'] = 'file'
post_data['executable'] = data
post_data['notification'] = 'email'
post_data['email'] = FROM_ADDR
# MIME encode the POST payload
opener = urllib2.build_opener(MultipartPostHandler.MultipartPostHandler)
urllib2.install_opener(opener)
request = urllib2.Request(url, post_data)
request.set_proxy('127.0.0.1:8080', 'http') # For testing with Burp Proxy
# Make the request and capture the response
try:
response = urllib2.urlopen(request)
print response.geturl()
except urllib2.URLError, e:
print "File upload failed..."
РЕДАКТИРОВАТЬ1: Спасибо за ваш ответ. Мне известно о решении ActiveState httplib для этого (я связался с ним выше). Я бы предпочел абстрагироваться от этой проблемы и использовать минимальный объем кода, чтобы продолжать использовать urllib2, как я был. Есть идеи, почему нож не устанавливается и не используется?
6 ответов
Кажется, что самый простой и совместимый способ обойти эту проблему - это использовать модуль 'poster'.
# test_client.py
from poster.encode import multipart_encode
from poster.streaminghttp import register_openers
import urllib2
# Register the streaming http handlers with urllib2
register_openers()
# Start the multipart/form-data encoding of the file "DSC0001.jpg"
# "image1" is the name of the parameter, which is normally set
# via the "name" parameter of the HTML <input> tag.
# headers contains the necessary Content-Type and Content-Length
# datagen is a generator object that yields the encoded parameters
datagen, headers = multipart_encode({"image1": open("DSC0001.jpg")})
# Create the Request object
request = urllib2.Request("http://localhost:5000/upload_image", datagen, headers)
# Actually do the request, and get the response
print urllib2.urlopen(request).read()
Это сработало отлично, и мне не пришлось гадить с httplib. Модуль доступен здесь: http://atlee.ca/software/poster/index.html
Нашел этот рецепт для публикации нескольких частей, используя httplib напрямую (без внешних библиотек)
import httplib
import mimetypes
def post_multipart(host, selector, fields, files):
content_type, body = encode_multipart_formdata(fields, files)
h = httplib.HTTP(host)
h.putrequest('POST', selector)
h.putheader('content-type', content_type)
h.putheader('content-length', str(len(body)))
h.endheaders()
h.send(body)
errcode, errmsg, headers = h.getreply()
return h.file.read()
def encode_multipart_formdata(fields, files):
LIMIT = '----------lImIt_of_THE_fIle_eW_$'
CRLF = '\r\n'
L = []
for (key, value) in fields:
L.append('--' + LIMIT)
L.append('Content-Disposition: form-data; name="%s"' % key)
L.append('')
L.append(value)
for (key, filename, value) in files:
L.append('--' + LIMIT)
L.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename))
L.append('Content-Type: %s' % get_content_type(filename))
L.append('')
L.append(value)
L.append('--' + LIMIT + '--')
L.append('')
body = CRLF.join(L)
content_type = 'multipart/form-data; boundary=%s' % LIMIT
return content_type, body
def get_content_type(filename):
return mimetypes.guess_type(filename)[0] or 'application/octet-stream'
Просто используйте python-запросы, он установит правильные заголовки и сделает за вас загрузку:
import requests
files = {"form_input_field_name": open("filename", "rb")}
requests.post("http://httpbin.org/post", files=files)
Я столкнулся с той же проблемой, и мне нужно было сделать многочастную форму без использования внешних библиотек. Я написал целый блог о проблемах, с которыми столкнулся.
В итоге я использовал измененную версию http://code.activestate.com/recipes/146306/. Код в этом URL фактически добавляет содержимое файла в виде строки, что может вызвать проблемы с двоичными файлами. Вот мой рабочий код.
import mimetools
import mimetypes
import io
import http
import json
form = MultiPartForm()
form.add_field("form_field", "my awesome data")
# Add a fake file
form.add_file(key, os.path.basename(filepath),
fileHandle=codecs.open("/path/to/my/file.zip", "rb"))
# Build the request
url = "http://www.example.com/endpoint"
schema, netloc, url, params, query, fragments = urlparse.urlparse(url)
try:
form_buffer = form.get_binary().getvalue()
http = httplib.HTTPConnection(netloc)
http.connect()
http.putrequest("POST", url)
http.putheader('Content-type',form.get_content_type())
http.putheader('Content-length', str(len(form_buffer)))
http.endheaders()
http.send(form_buffer)
except socket.error, e:
raise SystemExit(1)
r = http.getresponse()
if r.status == 200:
return json.loads(r.read())
else:
print('Upload failed (%s): %s' % (r.status, r.reason))
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 get_binary(self):
"""Return a binary buffer containing the form data, including attached files."""
part_boundary = '--' + self.boundary
binary = io.BytesIO()
needsCLRF = False
# Add the form fields
for name, value in self.form_fields:
if needsCLRF:
binary.write('\r\n')
needsCLRF = True
block = [part_boundary,
'Content-Disposition: form-data; name="%s"' % name,
'',
value
]
binary.write('\r\n'.join(block))
# Add the files to upload
for field_name, filename, content_type, body in self.files:
if needsCLRF:
binary.write('\r\n')
needsCLRF = True
block = [part_boundary,
str('Content-Disposition: file; name="%s"; filename="%s"' % \
(field_name, filename)),
'Content-Type: %s' % content_type,
''
]
binary.write('\r\n'.join(block))
binary.write('\r\n')
binary.write(body)
# add closing boundary marker,
binary.write('\r\n--' + self.boundary + '--\r\n')
return binary
Я какое совпадение, 2 года, 6 месяцев назад создаю проект
https://pypi.python.org/pypi/MultipartPostHandler2, который исправляет MultipartPostHandler для систем utf-8. Я также сделал некоторые незначительные улучшения, вы можете проверить это:)
Чтобы ответить на вопрос OP о том, почему оригинальный код не работал, переданный обработчик не был экземпляром класса. Линия
# MIME encode the POST payload
opener = urllib2.build_opener(MultipartPostHandler.MultipartPostHandler)
должен прочесть
opener = urllib2.build_opener(MultipartPostHandler.MultipartPostHandler())