forked from extern/httpie-cli
21 lines
698 B
Python
21 lines
698 B
Python
|
from typing import Tuple, Union
|
||
|
|
|
||
|
|
from httpie.cli.dicts import RequestDataDict, RequestFilesDict
|
||
|
|
from requests_toolbelt import MultipartEncoder
|
||
|
|
|
||
|
|
|
||
|
|
# Multipart uploads smaller than this size gets buffered (otherwise streamed).
|
||
|
|
# NOTE: Unbuffered upload requests cannot be displayed on the terminal.
|
||
|
|
UPLOAD_BUFFER = 1024 * 100
|
||
|
|
|
||
|
|
|
||
|
|
def get_multipart_data(
|
||
|
|
data: RequestDataDict,
|
||
|
|
files: RequestFilesDict
|
||
|
|
) -> Tuple[Union[MultipartEncoder, bytes], str]:
|
||
|
|
fields = list(data.items()) + list(files.items())
|
||
|
|
encoder = MultipartEncoder(fields=fields)
|
||
|
|
content_type = encoder.content_type
|
||
|
|
data = encoder.to_string() if encoder.len < UPLOAD_BUFFER else encoder
|
||
|
|
return data, content_type
|