Update utils.py to use urllib3.ProxyManager

When using ProxyManager instead of Poolmanager the proxy settings will be used when provided in proxy_url. Proxy_url can be retrieved from the environment variables proxy_url = os.environ.get('http_proxy') or os.environ.get('https_proxy')
wimglenn · Dec 6, 2023 · 5e57bcd · 5e57bcd
1 parent 64d216a
commit 5e57bcd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/aocd/utils.py b/aocd/utils.py
@@ -32,7 +32,12 @@ class HttpClient:
     # aocd users should not need to use this class directly.
 
     def __init__(self):
-        self.pool_manager = urllib3.PoolManager(headers={"User-Agent": USER_AGENT})
+
+        proxy_url = os.environ.get('http_proxy') or os.environ.get('https_proxy')
+        if proxy_url:
+            self.pool_manager = urllib3.ProxyManager(proxy_url, headers={"User-Agent": USER_AGENT})
+        else:
+            self.pool_manager = urllib3.PoolManager(headers={"User-Agent": USER_AGENT})
         self.req_count = {"GET": 0, "POST": 0}
         self._max_t = 3.0
         self._cooloff = 0.16