mirror of https://github.com/huashengdun/webssh
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.5 KiB
53 lines
1.5 KiB
import mimetypes
|
|
import os.path
|
|
from uuid import uuid4
|
|
from webssh.settings import base_dir
|
|
|
|
|
|
def encode_multipart_formdata(fields, files):
|
|
"""
|
|
fields is a sequence of (name, value) elements for regular form fields.
|
|
files is a sequence of (name, filename, value) elements for data to be
|
|
uploaded as files.
|
|
Return (content_type, body) ready for httplib.HTTP instance
|
|
"""
|
|
boundary = uuid4().hex
|
|
CRLF = '\r\n'
|
|
L = []
|
|
for (key, value) in fields:
|
|
L.append('--' + boundary)
|
|
L.append('Content-Disposition: form-data; name="%s"' % key)
|
|
L.append('')
|
|
L.append(value)
|
|
for (key, filename, value) in files:
|
|
L.append('--' + boundary)
|
|
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('--' + boundary + '--')
|
|
L.append('')
|
|
body = CRLF.join(L)
|
|
content_type = 'multipart/form-data; boundary=%s' % boundary
|
|
return content_type, body
|
|
|
|
|
|
def get_content_type(filename):
|
|
return mimetypes.guess_type(filename)[0] or 'application/octet-stream'
|
|
|
|
|
|
def read_file(path, encoding='utf-8'):
|
|
with open(path, 'rb') as f:
|
|
data = f.read()
|
|
if encoding is None:
|
|
return data
|
|
return data.decode(encoding)
|
|
|
|
|
|
def make_tests_data_path(filename):
|
|
return os.path.join(base_dir, 'tests', 'data', filename)
|