Add support for pre-configured pool manager

Author: sstallion

CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+- Added support for pre-configured pool managers
+
 ## [2.0.0] - 2024-08-22
 
 - Migrated to [openapi-generator](https://github.com/OpenAPITools/openapi-generator)

sensorpush_api/configuration.py

@@ -84,6 +84,7 @@ def __init__(self, host=None,
                  server_index=None, server_variables=None,
                  server_operation_index=None, server_operation_variables=None,
                  ssl_ca_cert=None,
+                 pool_manager=None
                  ) -> None:
         """Constructor
         """
@@ -199,6 +200,10 @@ def __init__(self, host=None,
         """date format
         """
 
+        self.pool_manager = pool_manager
+        """Provide a pre-configured pool manager
+        """
+
     def __deepcopy__(self, memo):
         cls = self.__class__
         result = cls.__new__(cls)

sensorpush_api/rest.py

@@ -52,29 +52,31 @@ def __init__(self, configuration, pools_size=4, maxsize=None, trust_env=False) -
         if maxsize is None:
             maxsize = configuration.connection_pool_maxsize
 
-        ssl_context = ssl.create_default_context(cafile=configuration.ssl_ca_cert)
-        if configuration.cert_file:
-            ssl_context.load_cert_chain(
-                configuration.cert_file, keyfile=configuration.key_file
-            )
-
-        if not configuration.verify_ssl:
-            ssl_context.check_hostname = False
-            ssl_context.verify_mode = ssl.CERT_NONE
-
-        connector = aiohttp.TCPConnector(
-            limit=maxsize,
-            ssl=ssl_context
-        )
-
         self.proxy = configuration.proxy
         self.proxy_headers = configuration.proxy_headers
 
-        # https pool manager
-        self.pool_manager = aiohttp.ClientSession(
-            connector=connector,
-            trust_env=trust_env,
-        )
+        self.pool_manager = configuration.pool_manager
+        if self.pool_manager is None:
+            ssl_context = ssl.create_default_context(cafile=configuration.ssl_ca_cert)
+            if configuration.cert_file:
+                ssl_context.load_cert_chain(
+                    configuration.cert_file, keyfile=configuration.key_file
+                )
+
+            if not configuration.verify_ssl:
+                ssl_context.check_hostname = False
+                ssl_context.verify_mode = ssl.CERT_NONE
+
+            connector = aiohttp.TCPConnector(
+                limit=maxsize,
+                ssl=ssl_context
+            )
+
+            # https pool manager
+            self.pool_manager = aiohttp.ClientSession(
+                connector=connector,
+                trust_env=trust_env,
+            )
 
     async def close(self):
         await self.pool_manager.close()