✨ add httpx transport config

Author: yanyongyu

githubkit/config.py

@@ -24,6 +24,8 @@ class Config:
     ssl_verify: Union[bool, "ssl.SSLContext"]
     trust_env: bool  # effects the `httpx` proxy and ssl cert
     proxy: Optional[ProxyTypes]
+    transport: Optional[httpx.BaseTransport]
+    async_transport: Optional[httpx.AsyncBaseTransport]
     cache_strategy: BaseCacheStrategy
     http_cache: bool
     throttler: BaseThrottler
@@ -113,6 +115,8 @@ def get_config(
     ssl_verify: Union[bool, "ssl.SSLContext"] = True,
     trust_env: bool = True,
     proxy: Optional[ProxyTypes] = None,
+    transport: Optional[httpx.BaseTransport] = None,
+    async_transport: Optional[httpx.AsyncBaseTransport] = None,
     cache_strategy: Optional[BaseCacheStrategy] = None,
     http_cache: bool = True,
     throttler: Optional[BaseThrottler] = None,
@@ -129,6 +133,8 @@ def get_config(
         ssl_verify,
         trust_env,
         proxy,
+        transport,
+        async_transport,
         build_cache_strategy(cache_strategy),
         http_cache,
         build_throttler(throttler),

githubkit/core.py

@@ -88,6 +88,8 @@ def __init__(
         ssl_verify: Union[bool, "ssl.SSLContext"] = ...,
         trust_env: bool = True,
         proxy: Optional[ProxyTypes] = None,
+        transport: Optional[httpx.BaseTransport] = None,
+        async_transport: Optional[httpx.AsyncBaseTransport] = None,
         cache_strategy: Optional[BaseCacheStrategy] = None,
         http_cache: bool = True,
         throttler: Optional[BaseThrottler] = None,
@@ -110,6 +112,8 @@ def __init__(
         ssl_verify: Union[bool, "ssl.SSLContext"] = ...,
         trust_env: bool = True,
         proxy: Optional[ProxyTypes] = None,
+        transport: Optional[httpx.BaseTransport] = None,
+        async_transport: Optional[httpx.AsyncBaseTransport] = None,
         cache_strategy: Optional[BaseCacheStrategy] = None,
         http_cache: bool = True,
         throttler: Optional[BaseThrottler] = None,
@@ -132,6 +136,8 @@ def __init__(
         ssl_verify: Union[bool, "ssl.SSLContext"] = ...,
         trust_env: bool = True,
         proxy: Optional[ProxyTypes] = None,
+        transport: Optional[httpx.BaseTransport] = None,
+        async_transport: Optional[httpx.AsyncBaseTransport] = None,
         cache_strategy: Optional[BaseCacheStrategy] = None,
         http_cache: bool = True,
         throttler: Optional[BaseThrottler] = None,
@@ -153,6 +159,8 @@ def __init__(
         ssl_verify: Union[bool, "ssl.SSLContext"] = True,
         trust_env: bool = True,
         proxy: Optional[ProxyTypes] = None,
+        transport: Optional[httpx.BaseTransport] = None,
+        async_transport: Optional[httpx.AsyncBaseTransport] = None,
         cache_strategy: Optional[BaseCacheStrategy] = None,
         http_cache: bool = True,
         throttler: Optional[BaseThrottler] = None,
@@ -174,6 +182,8 @@ def __init__(
             ssl_verify=ssl_verify,
             trust_env=trust_env,
             proxy=proxy,
+            transport=transport,
+            async_transport=async_transport,
             cache_strategy=cache_strategy,
             http_cache=http_cache,
             throttler=throttler,
@@ -241,11 +251,14 @@ def _create_sync_client(self) -> httpx.Client:
         if self.config.http_cache:
             return hishel.CacheClient(
                 **self._get_client_defaults(),
+                transport=self.config.transport,
                 storage=self.config.cache_strategy.get_hishel_storage(),
                 controller=self.config.cache_strategy.get_hishel_controller(),
             )
 
-        return httpx.Client(**self._get_client_defaults())
+        return httpx.Client(
+            **self._get_client_defaults(), transport=self.config.transport
+        )
 
     # get or create sync client
     @contextmanager
@@ -263,11 +276,14 @@ def _create_async_client(self) -> httpx.AsyncClient:
         if self.config.http_cache:
             return hishel.AsyncCacheClient(
                 **self._get_client_defaults(),
+                transport=self.config.async_transport,
                 storage=self.config.cache_strategy.get_async_hishel_storage(),
                 controller=self.config.cache_strategy.get_hishel_controller(),
             )
 
-        return httpx.AsyncClient(**self._get_client_defaults())
+        return httpx.AsyncClient(
+            **self._get_client_defaults(), transport=self.config.async_transport
+        )
 
     # get or create async client
     @asynccontextmanager

githubkit/github.py

@@ -79,6 +79,8 @@ def __init__(
             ssl_verify: Union[bool, "ssl.SSLContext"] = ...,
             trust_env: bool = True,
             proxy: Optional[ProxyTypes] = None,
+            transport: Optional[httpx.BaseTransport] = None,
+            async_transport: Optional[httpx.AsyncBaseTransport] = None,
             cache_strategy: Optional["BaseCacheStrategy"] = None,
             http_cache: bool = True,
             throttler: Optional["BaseThrottler"] = None,
@@ -101,6 +103,8 @@ def __init__(
             ssl_verify: Union[bool, "ssl.SSLContext"] = ...,
             trust_env: bool = True,
             proxy: Optional[ProxyTypes] = None,
+            transport: Optional[httpx.BaseTransport] = None,
+            async_transport: Optional[httpx.AsyncBaseTransport] = None,
             cache_strategy: Optional["BaseCacheStrategy"] = None,
             http_cache: bool = True,
             throttler: Optional["BaseThrottler"] = None,
@@ -123,6 +127,8 @@ def __init__(
             ssl_verify: Union[bool, "ssl.SSLContext"] = ...,
             trust_env: bool = True,
             proxy: Optional[ProxyTypes] = None,
+            transport: Optional[httpx.BaseTransport] = None,
+            async_transport: Optional[httpx.AsyncBaseTransport] = None,
             cache_strategy: Optional["BaseCacheStrategy"] = None,
             http_cache: bool = True,
             throttler: Optional["BaseThrottler"] = None,