From da21e4870e6ebcd637f635064caf07f06a4a9680 Mon Sep 17 00:00:00 2001 From: Justin Zandbergen Date: Mon, 26 Sep 2022 10:11:14 +0200 Subject: [PATCH 1/2] Allow for OpenAPI connection pool configuration. --- tuya_iot/openapi.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tuya_iot/openapi.py b/tuya_iot/openapi.py index 10e6f23..29edd1e 100644 --- a/tuya_iot/openapi.py +++ b/tuya_iot/openapi.py @@ -62,9 +62,16 @@ def __init__( access_secret: str, auth_type: AuthType = AuthType.SMART_HOME, lang: str = "en", + pool_connections: int = 10, + pool_maxsize: int = 10, ) -> None: """Init TuyaOpenAPI.""" self.session = requests.session() + self.adapter = requests.adapters.HTTPAdapter( + pool_connections=pool_connections, + pool_maxsize=pool_maxsize + ) + self.session.mount("https://", self.adapter) self.endpoint = endpoint self.access_id = access_id From 5db2687f8105924e91720b4a5166d20baf6fcd49 Mon Sep 17 00:00:00 2001 From: Justin Zandbergen Date: Mon, 26 Sep 2022 11:25:13 +0200 Subject: [PATCH 2/2] Add `max_retries` set default value to 3. --- tuya_iot/openapi.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tuya_iot/openapi.py b/tuya_iot/openapi.py index 29edd1e..90b515b 100644 --- a/tuya_iot/openapi.py +++ b/tuya_iot/openapi.py @@ -64,12 +64,14 @@ def __init__( lang: str = "en", pool_connections: int = 10, pool_maxsize: int = 10, + max_retries: int = 3 ) -> None: """Init TuyaOpenAPI.""" self.session = requests.session() self.adapter = requests.adapters.HTTPAdapter( - pool_connections=pool_connections, - pool_maxsize=pool_maxsize + pool_connections = pool_connections, + pool_maxsize = pool_maxsize, + max_retries = max_retries ) self.session.mount("https://", self.adapter)