Merge pull request #68 from monkeydone/main

Support terminal proxy in code
pythongosssss · Oct 23, 2024 · ac5601e · ac5601e
2 parents 4f1a857 + 5b160bd
commit ac5601e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pysssss.py b/pysssss.py
@@ -175,7 +175,13 @@ async def download_to_file(url, destination, update_callback, is_ext_subpath=Tru
     if is_ext_subpath:
         destination = get_ext_dir(destination)
     try:
-        async with session.get(url) as response:
+        proxy = os.getenv("HTTP_PROXY") or os.getenv("http_proxy")
+        print("proxy:", proxy)
+        proxy_auth = None
+        if proxy:
+            proxy_auth = aiohttp.BasicAuth(os.getenv("PROXY_USER", ""), os.getenv("PROXY_PASS", ""))
+
+        async with session.get(url, proxy=proxy, proxy_auth=proxy_auth) as response:            
             size = int(response.headers.get('content-length', 0)) or None
 
             with tqdm(