Merge pull request #73 from tony-romanovych/master

Extend DiscoveryBase with `api_factory` setting
HubSpot · Dec 1, 2021 · 5ab761d · 5ab761d
2 parents ff3a0f7 + 8c53084
commit 5ab761d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-4.0.5
+4.0.6
diff --git a/hubspot/discovery/discovery_base.py b/hubspot/discovery/discovery_base.py
@@ -6,8 +6,13 @@ def __init__(self, config):
         self.config = config
 
     def _configure_api_client(self, api_client_package, api_name):
+        api_factory = self.config.get("api_factory") or self._default_api_factory
+        config = {k: v for k, v in self.config.items() if k != "api_factory" and v}
+        return api_factory(api_client_package, api_name, config)
+
+    @staticmethod
+    def _default_api_factory(api_client_package, api_name, config):
         configuration = api_client_package.Configuration()
-        config = {k: v for k, v in self.config.items() if v is not None}
         if "api_key" in config:
             configuration.api_key["hapikey"] = config["api_key"]
         if "access_token" in config: