Compare commits
2 commits
2bac617dc9
...
b4fbb863af
Author | SHA1 | Date | |
---|---|---|---|
b4fbb863af | |||
973c7e1ef1 |
3 changed files with 0 additions and 106 deletions
|
@ -17,7 +17,6 @@ from . import vtt
|
||||||
|
|
||||||
from .api import *
|
from .api import *
|
||||||
from .file import *
|
from .file import *
|
||||||
from .form import *
|
|
||||||
from .format import *
|
from .format import *
|
||||||
from .geo import *
|
from .geo import *
|
||||||
from .html import *
|
from .html import *
|
||||||
|
|
|
@ -18,7 +18,6 @@ import requests
|
||||||
|
|
||||||
from . import __version__
|
from . import __version__
|
||||||
from .utils import json
|
from .utils import json
|
||||||
from .form import MultiPartForm
|
|
||||||
|
|
||||||
__all__ = ['getAPI', 'API']
|
__all__ = ['getAPI', 'API']
|
||||||
|
|
||||||
|
|
104
ox/form.py
104
ox/form.py
|
@ -1,104 +0,0 @@
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
# vi:si:et:sw=4:sts=4:ts=4
|
|
||||||
# GPL 2014
|
|
||||||
from __future__ import print_function
|
|
||||||
|
|
||||||
import itertools
|
|
||||||
import mimetypes
|
|
||||||
import os
|
|
||||||
import hashlib
|
|
||||||
import sys
|
|
||||||
|
|
||||||
|
|
||||||
__all__ = ['MultiPartForm']
|
|
||||||
|
|
||||||
# from /usr/lib/python3.4/email/generator.py
|
|
||||||
# Helper used by Generator._make_boundary
|
|
||||||
_width = len(repr(sys.maxsize-1))
|
|
||||||
_fmt = '%%0%dd' % _width
|
|
||||||
|
|
||||||
def _make_boundary():
|
|
||||||
# Craft a random boundary.
|
|
||||||
boundary = ('=' * 15) + hashlib.sha1(os.urandom(32)).hexdigest() + '=='
|
|
||||||
return boundary
|
|
||||||
|
|
||||||
class MultiPartForm(object):
|
|
||||||
"""Accumulate the data to be used when posting a form."""
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.form_fields = []
|
|
||||||
self.files = []
|
|
||||||
self.boundary = _make_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."""
|
|
||||||
if isinstance(name, bytes):
|
|
||||||
name = name.decode('utf-8')
|
|
||||||
if isinstance(value, bytes):
|
|
||||||
value = value.decode('utf-8')
|
|
||||||
self.form_fields.append((name, value))
|
|
||||||
return
|
|
||||||
|
|
||||||
def add_file(self, fieldname, filename, fileHandle, mimetype=None):
|
|
||||||
"""Add a file to be uploaded."""
|
|
||||||
if isinstance(fieldname, bytes):
|
|
||||||
fieldname = fieldname.decode('utf-8')
|
|
||||||
if isinstance(filename, bytes):
|
|
||||||
filename = filename.decode('utf-8')
|
|
||||||
|
|
||||||
if hasattr(fileHandle, 'read'):
|
|
||||||
body = fileHandle.read()
|
|
||||||
else:
|
|
||||||
body = fileHandle
|
|
||||||
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):
|
|
||||||
body = self.body()
|
|
||||||
return body
|
|
||||||
|
|
||||||
def body(self):
|
|
||||||
"""Return a byte 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('')
|
|
||||||
flattened = [part if isinstance(part, bytes) else part.encode('utf-8') for part in flattened]
|
|
||||||
return b'\r\n'.join(flattened)
|
|
||||||
|
|
Loading…
Reference in a new issue