[Python] support api key refresh in configuration module (#3594)

OpenAPITools · Aug 15, 2019 · 99fcfbc · 99fcfbc
1 parent a4811c7
commit 99fcfbc
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 0 deletions.
diff --git a/modules/openapi-generator/src/main/resources/python/configuration.mustache b/modules/openapi-generator/src/main/resources/python/configuration.mustache
@@ -61,6 +61,9 @@ class Configuration(six.with_metaclass(TypeWithDefault, object)):
         self.api_key_prefix = api_key_prefix
         """dict to store API prefix (e.g. Bearer)
         """
+        self.refresh_api_key_hook = None
+        """function hook to refresh API key if expired
+        """
         self.username = username
         """Username for HTTP basic authentication
         """
@@ -238,6 +241,8 @@ class Configuration(six.with_metaclass(TypeWithDefault, object)):
         :param identifier: The identifier of apiKey.
         :return: The token for api key authentication.
         """
+        if self.refresh_api_key_hook is not None:
+            self.refresh_api_key_hook(self)
         key = self.api_key.get(identifier)
         if key:
             prefix = self.api_key_prefix.get(identifier)

diff --git a/samples/client/petstore/python-asyncio/petstore_api/configuration.py b/samples/client/petstore/python-asyncio/petstore_api/configuration.py
@@ -66,6 +66,9 @@ def __init__(self, host="http://petstore.swagger.io:80/v2",
         self.api_key_prefix = api_key_prefix
         """dict to store API prefix (e.g. Bearer)
         """
+        self.refresh_api_key_hook = None
+        """function hook to refresh API key if expired
+        """
         self.username = username
         """Username for HTTP basic authentication
         """
@@ -223,6 +226,8 @@ def get_api_key_with_prefix(self, identifier):
         :param identifier: The identifier of apiKey.
         :return: The token for api key authentication.
         """
+        if self.refresh_api_key_hook is not None:
+            self.refresh_api_key_hook(self)
         key = self.api_key.get(identifier)
         if key:
             prefix = self.api_key_prefix.get(identifier)

diff --git a/samples/client/petstore/python-tornado/petstore_api/configuration.py b/samples/client/petstore/python-tornado/petstore_api/configuration.py
@@ -67,6 +67,9 @@ def __init__(self, host="http://petstore.swagger.io:80/v2",
         self.api_key_prefix = api_key_prefix
         """dict to store API prefix (e.g. Bearer)
         """
+        self.refresh_api_key_hook = None
+        """function hook to refresh API key if expired
+        """
         self.username = username
         """Username for HTTP basic authentication
         """
@@ -227,6 +230,8 @@ def get_api_key_with_prefix(self, identifier):
         :param identifier: The identifier of apiKey.
         :return: The token for api key authentication.
         """
+        if self.refresh_api_key_hook is not None:
+            self.refresh_api_key_hook(self)
         key = self.api_key.get(identifier)
         if key:
             prefix = self.api_key_prefix.get(identifier)

diff --git a/samples/client/petstore/python/petstore_api/configuration.py b/samples/client/petstore/python/petstore_api/configuration.py
@@ -67,6 +67,9 @@ def __init__(self, host="http://petstore.swagger.io:80/v2",
         self.api_key_prefix = api_key_prefix
         """dict to store API prefix (e.g. Bearer)
         """
+        self.refresh_api_key_hook = None
+        """function hook to refresh API key if expired
+        """
         self.username = username
         """Username for HTTP basic authentication
         """
@@ -227,6 +230,8 @@ def get_api_key_with_prefix(self, identifier):
         :param identifier: The identifier of apiKey.
         :return: The token for api key authentication.
         """
+        if self.refresh_api_key_hook is not None:
+            self.refresh_api_key_hook(self)
         key = self.api_key.get(identifier)
         if key:
             prefix = self.api_key_prefix.get(identifier)

diff --git a/samples/openapi3/client/petstore/python/petstore_api/configuration.py b/samples/openapi3/client/petstore/python/petstore_api/configuration.py
@@ -67,6 +67,9 @@ def __init__(self, host="http://petstore.swagger.io:80/v2",
         self.api_key_prefix = api_key_prefix
         """dict to store API prefix (e.g. Bearer)
         """
+        self.refresh_api_key_hook = None
+        """function hook to refresh API key if expired
+        """
         self.username = username
         """Username for HTTP basic authentication
         """
@@ -227,6 +230,8 @@ def get_api_key_with_prefix(self, identifier):
         :param identifier: The identifier of apiKey.
         :return: The token for api key authentication.
         """
+        if self.refresh_api_key_hook is not None:
+            self.refresh_api_key_hook(self)
         key = self.api_key.get(identifier)
         if key:
             prefix = self.api_key_prefix.get(identifier)