[Python] support api key refresh in configuration module

swagger-api · Aug 6, 2019 · f243386 · f243386
1 parent 0d372e5
commit f243386
Show file tree

Hide file tree

Showing 7 changed files with 23 additions and 3 deletions.
diff --git a/modules/swagger-codegen/src/main/resources/python/configuration.mustache b/modules/swagger-codegen/src/main/resources/python/configuration.mustache
@@ -40,6 +40,8 @@ class Configuration(object):
         self.api_key = {}
         # dict to store API prefix (e.g. Bearer)
         self.api_key_prefix = {}
+        # function to refresh API key if expired
+        self.refresh_api_key_hook = None
         # Username for HTTP basic authentication
         self.username = ""
         # Password for HTTP basic authentication
@@ -194,6 +196,9 @@ class Configuration(object):
         :param identifier: The identifier of apiKey.
         :return: The token for api key authentication.
         """
+        if self.refresh_api_key_hook:
+            self.refresh_api_key_hook(self)
+
         if (self.api_key.get(identifier) and
                 self.api_key_prefix.get(identifier)):
             return self.api_key_prefix[identifier] + ' ' + self.api_key[identifier]  # noqa: E501

diff --git a/samples/client/petstore/python-asyncio/.swagger-codegen/VERSION b/samples/client/petstore/python-asyncio/.swagger-codegen/VERSION
@@ -1 +1 @@
-2.4.3-SNAPSHOT
+2.4.8-SNAPSHOT
diff --git a/samples/client/petstore/python-asyncio/petstore_api/configuration.py b/samples/client/petstore/python-asyncio/petstore_api/configuration.py
@@ -49,6 +49,8 @@ def __init__(self):
         self.api_key = {}
         # dict to store API prefix (e.g. Bearer)
         self.api_key_prefix = {}
+        # function to refresh API key if expired
+        self.refresh_api_key_hook = None
         # Username for HTTP basic authentication
         self.username = ""
         # Password for HTTP basic authentication
@@ -203,6 +205,9 @@ 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:
+            self.refresh_api_key_hook(self)
+
         if (self.api_key.get(identifier) and
                 self.api_key_prefix.get(identifier)):
             return self.api_key_prefix[identifier] + ' ' + self.api_key[identifier]  # noqa: E501

diff --git a/samples/client/petstore/python-tornado/.swagger-codegen/VERSION b/samples/client/petstore/python-tornado/.swagger-codegen/VERSION
@@ -1 +1 @@
-2.4.3-SNAPSHOT
+2.4.8-SNAPSHOT
diff --git a/samples/client/petstore/python-tornado/petstore_api/configuration.py b/samples/client/petstore/python-tornado/petstore_api/configuration.py
@@ -49,6 +49,8 @@ def __init__(self):
         self.api_key = {}
         # dict to store API prefix (e.g. Bearer)
         self.api_key_prefix = {}
+        # function to refresh API key if expired
+        self.refresh_api_key_hook = None
         # Username for HTTP basic authentication
         self.username = ""
         # Password for HTTP basic authentication
@@ -203,6 +205,9 @@ 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:
+            self.refresh_api_key_hook(self)
+
         if (self.api_key.get(identifier) and
                 self.api_key_prefix.get(identifier)):
             return self.api_key_prefix[identifier] + ' ' + self.api_key[identifier]  # noqa: E501

diff --git a/samples/client/petstore/python/.swagger-codegen/VERSION b/samples/client/petstore/python/.swagger-codegen/VERSION
@@ -1 +1 @@
-2.4.3-SNAPSHOT
+2.4.8-SNAPSHOT
diff --git a/samples/client/petstore/python/petstore_api/configuration.py b/samples/client/petstore/python/petstore_api/configuration.py
@@ -49,6 +49,8 @@ def __init__(self):
         self.api_key = {}
         # dict to store API prefix (e.g. Bearer)
         self.api_key_prefix = {}
+        # function to refresh API key if expired
+        self.refresh_api_key_hook = None
         # Username for HTTP basic authentication
         self.username = ""
         # Password for HTTP basic authentication
@@ -203,6 +205,9 @@ 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:
+            self.refresh_api_key_hook(self)
+
         if (self.api_key.get(identifier) and
                 self.api_key_prefix.get(identifier)):
             return self.api_key_prefix[identifier] + ' ' + self.api_key[identifier]  # noqa: E501