swagger-api#4639 [Python] Proxy support in generated Python module

oflebbe · Feb 14, 2017 · 1f5ed79 · 1f5ed79
1 parent 2447412
commit 1f5ed79
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 9 deletions.
diff --git a/modules/swagger-codegen/src/main/resources/python/configuration.mustache b/modules/swagger-codegen/src/main/resources/python/configuration.mustache
@@ -79,6 +79,8 @@ class Configuration(object):
         self.cert_file = None
         # client key file
         self.key_file = None
+        # http proxy
+        self.proxy = None
 
     @property
     def logger_file(self):

diff --git a/modules/swagger-codegen/src/main/resources/python/rest.mustache b/modules/swagger-codegen/src/main/resources/python/rest.mustache
@@ -76,15 +76,31 @@ class RESTClientObject(object):
         # key file
         key_file = Configuration().key_file
 
-        # https pool manager
-        self.pool_manager = urllib3.PoolManager(
-            num_pools=pools_size,
-            maxsize=maxsize,
-            cert_reqs=cert_reqs,
-            ca_certs=ca_certs,
-            cert_file=cert_file,
-            key_file=key_file
-        )
+        # http proxy
+        proxy = Configuration().proxy
+
+        if proxy is None :
+
+            # https pool manager
+            self.pool_manager = urllib3.PoolManager(
+                num_pools=pools_size,
+                maxsize=maxsize,
+                cert_reqs=cert_reqs,
+                ca_certs=ca_certs,
+                cert_file=cert_file,
+                key_file=key_file
+            )
+        else :
+           self.pool_manager = urllib3.ProxyManager(
+                proxy,
+                num_pools=pools_size,
+                maxsize=maxsize,
+                cert_reqs=cert_reqs,
+                ca_certs=ca_certs,
+                cert_file=cert_file,
+                key_file=key_file
+            )
+
 
     def request(self, method, url, query_params=None, headers=None,
                 body=None, post_params=None, _preload_content=True, _request_timeout=None):