extern/httpie-cli
1
0
Fork 1
mirror of https://github.com/httpie/cli.git synced 2024-11-21 23:33:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improve startup time when pyOpenSSL is available on the environment (#1233)

Browse Source
This commit is contained in:
Batuhan Taskaya 2021-12-17 11:00:22 +03:00 committed by GitHub
parent c237e15108
commit 1bd8422fb5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
httpie/uploads.py
View File

@ -1,10 +1,12 @@
import zlib
from typing import Callable, IO, Iterable, Tuple, Union
from typing import Callable, IO, Iterable, Tuple, Union, TYPE_CHECKING
from urllib.parse import urlencode
import requests
from requests.utils import super_len
from requests_toolbelt import MultipartEncoder
if TYPE_CHECKING:
from requests_toolbelt import MultipartEncoder
from .cli.dicts import MultipartRequestDataDict, RequestDataDict
@ -23,7 +25,7 @@ class ChunkedUploadStream:
class ChunkedMultipartUploadStream:
chunk_size = 100 * 1024
def __init__(self, encoder: MultipartEncoder):
def __init__(self, encoder: 'MultipartEncoder'):
self.encoder = encoder
def __iter__(self) -> Iterable[Union[str, bytes]]:
@ -35,12 +37,12 @@ class ChunkedMultipartUploadStream:
def prepare_request_body(
body: Union[str, bytes, IO, MultipartEncoder, RequestDataDict],
body: Union[str, bytes, IO, 'MultipartEncoder', RequestDataDict],
body_read_callback: Callable[[bytes], bytes],
content_length_header_value: int = None,
chunked=False,
offline=False,
) -> Union[str, bytes, IO, MultipartEncoder, ChunkedUploadStream]:
) -> Union[str, bytes, IO, 'MultipartEncoder', ChunkedUploadStream]:
is_file_like = hasattr(body, 'read')
@ -85,6 +87,7 @@ def prepare_request_body(
body.read = new_read
if chunked:
from requests_toolbelt import MultipartEncoder
if isinstance(body, MultipartEncoder):
body = ChunkedMultipartUploadStream(
encoder=body,
@ -102,7 +105,9 @@ def get_multipart_data_and_content_type(
data: MultipartRequestDataDict,
boundary: str = None,
content_type: str = None,
) -> Tuple[MultipartEncoder, str]:
) -> Tuple['MultipartEncoder', str]:
from requests_toolbelt import MultipartEncoder
encoder = MultipartEncoder(
fields=data.items(),
boundary=boundary,