Support custom httpx transports

lightkube/config/client_adapter.py

@@ -12,15 +12,25 @@
 
 
 def Client(
-    config: SingleConfig, timeout: httpx.Timeout, trust_env: bool = True
+    config: SingleConfig,
+    timeout: httpx.Timeout,
+    trust_env: bool = True,
+    transport: httpx.BaseTransport = None,
 ) -> httpx.Client:
-    return httpx.Client(**httpx_parameters(config, timeout, trust_env))
+    return httpx.Client(
+        transport=transport, **httpx_parameters(config, timeout, trust_env)
+    )
 
 
 def AsyncClient(
-    config: SingleConfig, timeout: httpx.Timeout, trust_env: bool = True
+    config: SingleConfig,
+    timeout: httpx.Timeout,
+    trust_env: bool = True,
+    transport: httpx.AsyncBaseTransport = None,
 ) -> httpx.AsyncClient:
-    return httpx.AsyncClient(**httpx_parameters(config, timeout, trust_env))
+    return httpx.AsyncClient(
+        transport=transport, **httpx_parameters(config, timeout, trust_env)
+    )
 
 
 def httpx_parameters(config: SingleConfig, timeout: httpx.Timeout, trust_env: bool):

lightkube/core/async_client.py

@@ -47,6 +47,7 @@ class AsyncClient:
     * **dry_run** - *(optional)* Apply server-side dry-run and guarantee that modifications will not
         be persisted in storage. Setting this field to `True` is equivalent of passing `--dry-run=server`
         to `kubectl` commands.
+    * **transport** - *(optional)* Custom httpx transport
     """
 
     def __init__(
@@ -58,6 +59,7 @@ def __init__(
         field_manager: str = None,
         trust_env: bool = True,
         dry_run: bool = False,
+        transport: httpx.AsyncBaseTransport = None,
     ):
         self._client = GenericAsyncClient(
             config,
@@ -67,6 +69,7 @@ def __init__(
             field_manager=field_manager,
             trust_env=trust_env,
             dry_run=dry_run,
+            transport=transport,
         )
 
     @property

lightkube/core/client.py

@@ -52,6 +52,7 @@ class Client:
     * **dry_run** - *(optional)* Apply server-side dry-run and guarantee that modifications will not
         be persisted in storage. Setting this field to `True` is equivalent of passing `--dry-run=server`
         to `kubectl` commands.
+    * **transport** - *(optional)* Custom httpx transport
     """
 
     def __init__(
@@ -63,6 +64,7 @@ def __init__(
         field_manager: str = None,
         trust_env: bool = True,
         dry_run: bool = False,
+        transport: httpx.BaseTransport = None,
     ):
         self._client = GenericSyncClient(
             config,
@@ -72,6 +74,7 @@ def __init__(
             field_manager=field_manager,
             trust_env=trust_env,
             dry_run=dry_run,
+            transport=transport,
         )
 
     @property

lightkube/core/generic_client.py

@@ -84,6 +84,7 @@ def __init__(
         trust_env: bool = True,
         field_manager: str = None,
         dry_run: bool = False,
+        transport: Union[httpx.BaseTransport, httpx.AsyncBaseTransport] = None,
     ):
         self._timeout = httpx.Timeout(10) if timeout is None else timeout
         self._watch_timeout = httpx.Timeout(self._timeout)
@@ -97,7 +98,9 @@ def __init__(
             config = config.get()
 
         self.config = config
-        self._client = self.AdapterClient(config, timeout, trust_env=trust_env)
+        self._client = self.AdapterClient(
+            config, timeout, trust_env=trust_env, transport=transport
+        )
         self._field_manager = field_manager
         self._dry_run = dry_run
         self.namespace = namespace if namespace else config.namespace

setup.py

@@ -3,7 +3,7 @@
 
 setup(
     name='lightkube',
-    version="0.15.4",
+    version="0.15.5",
     description='Lightweight kubernetes client library',
     long_description=Path("README.md").read_text(),
     long_description_content_type="text/markdown",