optional timeouts

fauna · Jul 26, 2023 · 847ce15 · 847ce15
1 parent 97e737f
commit 847ce15
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/fauna/client/client.py b/fauna/client/client.py
@@ -140,10 +140,14 @@ def __init__(
         if http_read_timeout is not None:
           read_timeout_s = http_read_timeout.total_seconds()
 
-        write_timeout_s = http_write_timeout.total_seconds()
-        connect_timeout_s = http_connect_timeout.total_seconds()
-        pool_timeout_s = http_pool_timeout.total_seconds()
-        idle_timeout_s = http_idle_timeout.total_seconds()
+        write_timeout_s: Optional[float] = http_write_timeout.total_seconds(
+        ) if http_write_timeout is not None else None
+        connect_timeout_s: Optional[float] = http_connect_timeout.total_seconds(
+        ) if http_connect_timeout is not None else None
+        pool_timeout_s: Optional[float] = http_pool_timeout.total_seconds(
+        ) if http_pool_timeout is not None else None
+        idle_timeout_s: Optional[float] = http_idle_timeout.total_seconds(
+        ) if http_idle_timeout is not None else None
 
         import httpx
         from fauna.http.httpx_client import HTTPXClient