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.
52 lines
1.5 KiB
52 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)
|
|
|