修复代理判断逻辑

auqhjjqdo · Mar 16, 2023 · 2f514df · 2f514df
1 parent 4607b12
commit 2f514df
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/main.py b/main.py
@@ -20,7 +20,7 @@
 class LiveRecoder:
     def __init__(self, config: dict, user: dict):
         self.interval = config['interval']
-        self.proxy = config.get('proxy', request.getproxies().get('http'))
+        self.proxy = config.get('proxy')
 
         self.platform = user['platform']
         self.id = user['id']
@@ -50,10 +50,13 @@ def get_client(self):
             'headers': self.headers,
             'cookies': self.cookies
         }
-        if 'socks' in self.proxy:
-            kwargs['transport'] = AsyncProxyTransport.from_url(self.proxy)
+        if self.proxy:
+            if 'socks' in self.proxy:
+                kwargs['transport'] = AsyncProxyTransport.from_url(self.proxy)
+            else:
+                kwargs['proxies'] = self.proxy
         else:
-            kwargs['proxies'] = self.proxy
+            self.proxy = request.getproxies().get('http')
         return httpx.AsyncClient(**kwargs)
 
     @staticmethod