forked from encode/httpx
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Add keepalive_expiry to Limits config * keepalive_expiry should be optional. In line with httpcore. * HTTPTransport and AsyncHTTPTransport * Update docs for httpx.HTTPTransport() * Update type hints * Fix docs typo * Additional mount example * Tweak context manager methods * Add 'httpx.HTTPTransport(proxy=...)' * Use explicit keyword arguments throughout httpx.HTTPTransport Co-authored-by: Florimond Manca <[email protected]>
- Loading branch information
1 parent
1816393
commit 89fb0cb
Showing
6 changed files
with
242 additions
and
70 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,174 @@ | ||
| """ | ||
| Custom transports, with nicely configured defaults. | ||
| The following additional keyword arguments are currently supported by httpcore... | ||
| * uds: str | ||
| * local_address: str | ||
| * retries: int | ||
| * backend: str ("auto", "asyncio", "trio", "curio", "anyio", "sync") | ||
| Example usages... | ||
| # Disable HTTP/2 on a single specfic domain. | ||
| mounts = { | ||
| "all://": httpx.HTTPTransport(http2=True), | ||
| "all://*example.org": httpx.HTTPTransport() | ||
| } | ||
| # Using advanced httpcore configuration, with connection retries. | ||
| transport = httpx.HTTPTransport(retries=1) | ||
| client = httpx.Client(transport=transport) | ||
| # Using advanced httpcore configuration, with unix domain sockets. | ||
| transport = httpx.HTTPTransport(uds="socket.uds") | ||
| client = httpx.Client(transport=transport) | ||
| """ | ||
| import typing | ||
| from types import TracebackType | ||
|
|
||
| import httpcore | ||
|
|
||
| from .._config import DEFAULT_LIMITS, Limits, Proxy, create_ssl_context | ||
| from .._types import CertTypes, VerifyTypes | ||
|
|
||
| T = typing.TypeVar("T", bound="HTTPTransport") | ||
| A = typing.TypeVar("A", bound="AsyncHTTPTransport") | ||
| Headers = typing.List[typing.Tuple[bytes, bytes]] | ||
| URL = typing.Tuple[bytes, bytes, typing.Optional[int], bytes] | ||
|
|
||
|
|
||
| class HTTPTransport(httpcore.SyncHTTPTransport): | ||
| def __init__( | ||
| self, | ||
| verify: VerifyTypes = True, | ||
| cert: CertTypes = None, | ||
| http2: bool = False, | ||
| limits: Limits = DEFAULT_LIMITS, | ||
| trust_env: bool = True, | ||
| proxy: Proxy = None, | ||
| uds: str = None, | ||
| local_address: str = None, | ||
| retries: int = 0, | ||
| backend: str = "sync", | ||
| ) -> None: | ||
| ssl_context = create_ssl_context(verify=verify, cert=cert, trust_env=trust_env) | ||
|
|
||
| if proxy is None: | ||
| self._pool = httpcore.SyncConnectionPool( | ||
| ssl_context=ssl_context, | ||
| max_connections=limits.max_connections, | ||
| max_keepalive_connections=limits.max_keepalive_connections, | ||
| keepalive_expiry=limits.keepalive_expiry, | ||
| http2=http2, | ||
| uds=uds, | ||
| local_address=local_address, | ||
| retries=retries, | ||
| backend=backend, | ||
| ) | ||
| else: | ||
| self._pool = httpcore.SyncHTTPProxy( | ||
| proxy_url=proxy.url.raw, | ||
| proxy_headers=proxy.headers.raw, | ||
| proxy_mode=proxy.mode, | ||
| ssl_context=ssl_context, | ||
| max_connections=limits.max_connections, | ||
| max_keepalive_connections=limits.max_keepalive_connections, | ||
| keepalive_expiry=limits.keepalive_expiry, | ||
| http2=http2, | ||
| backend=backend, | ||
| ) | ||
|
|
||
| def __enter__(self: T) -> T: # Use generics for subclass support. | ||
| self._pool.__enter__() | ||
| return self | ||
|
|
||
| def __exit__( | ||
| self, | ||
| exc_type: typing.Type[BaseException] = None, | ||
| exc_value: BaseException = None, | ||
| traceback: TracebackType = None, | ||
| ) -> None: | ||
| self._pool.__exit__(exc_type, exc_value, traceback) | ||
|
|
||
| def request( | ||
| self, | ||
| method: bytes, | ||
| url: URL, | ||
| headers: Headers = None, | ||
| stream: httpcore.SyncByteStream = None, | ||
| ext: dict = None, | ||
| ) -> typing.Tuple[int, Headers, httpcore.SyncByteStream, dict]: | ||
| return self._pool.request(method, url, headers=headers, stream=stream, ext=ext) | ||
|
|
||
| def close(self) -> None: | ||
| self._pool.close() | ||
|
|
||
|
|
||
| class AsyncHTTPTransport(httpcore.AsyncHTTPTransport): | ||
| def __init__( | ||
| self, | ||
| verify: VerifyTypes = True, | ||
| cert: CertTypes = None, | ||
| http2: bool = False, | ||
| limits: Limits = DEFAULT_LIMITS, | ||
| trust_env: bool = True, | ||
| proxy: Proxy = None, | ||
| uds: str = None, | ||
| local_address: str = None, | ||
| retries: int = 0, | ||
| backend: str = "auto", | ||
| ) -> None: | ||
| ssl_context = create_ssl_context(verify=verify, cert=cert, trust_env=trust_env) | ||
|
|
||
| if proxy is None: | ||
| self._pool = httpcore.AsyncConnectionPool( | ||
| ssl_context=ssl_context, | ||
| max_connections=limits.max_connections, | ||
| max_keepalive_connections=limits.max_keepalive_connections, | ||
| keepalive_expiry=limits.keepalive_expiry, | ||
| http2=http2, | ||
| uds=uds, | ||
| local_address=local_address, | ||
| retries=retries, | ||
| backend=backend, | ||
| ) | ||
| else: | ||
| self._pool = httpcore.AsyncHTTPProxy( | ||
| proxy_url=proxy.url.raw, | ||
| proxy_headers=proxy.headers.raw, | ||
| proxy_mode=proxy.mode, | ||
| ssl_context=ssl_context, | ||
| max_connections=limits.max_connections, | ||
| max_keepalive_connections=limits.max_keepalive_connections, | ||
| keepalive_expiry=limits.keepalive_expiry, | ||
| http2=http2, | ||
| backend=backend, | ||
| ) | ||
|
|
||
| async def __aenter__(self: A) -> A: # Use generics for subclass support. | ||
| await self._pool.__aenter__() | ||
| return self | ||
|
|
||
| async def __aexit__( | ||
| self, | ||
| exc_type: typing.Type[BaseException] = None, | ||
| exc_value: BaseException = None, | ||
| traceback: TracebackType = None, | ||
| ) -> None: | ||
| await self._pool.__aexit__(exc_type, exc_value, traceback) | ||
|
|
||
| async def arequest( | ||
| self, | ||
| method: bytes, | ||
| url: URL, | ||
| headers: Headers = None, | ||
| stream: httpcore.AsyncByteStream = None, | ||
| ext: dict = None, | ||
| ) -> typing.Tuple[int, Headers, httpcore.AsyncByteStream, dict]: | ||
| return await self._pool.arequest( | ||
| method, url, headers=headers, stream=stream, ext=ext | ||
| ) | ||
|
|
||
| async def aclose(self) -> None: | ||
| await self._pool.aclose() |
Oops, something went wrong.