修复self.proxy为None时产生的报错 (#50)

auqhjjqdo · Oct 5, 2023 · c3dbfda · c3dbfda
1 parent 0022e03
commit c3dbfda
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/live_recorder.py b/live_recorder.py
@@ -71,11 +71,17 @@ async def request(self, method, url, **kwargs):
             raise ConnectionError(f'{self.flag}直播检测请求错误\n{repr(error)}')
 
     def get_client(self):
+        # 检查是否有设置代理
+        if self.proxy:
+            transport = AsyncProxyTransport.from_url(self.proxy)
+        else:
+            transport = None
+
         return httpx.AsyncClient(
             http2=True,
             timeout=self.interval,
             limits=httpx.Limits(max_keepalive_connections=100, keepalive_expiry=self.interval * 2),
-            transport=AsyncProxyTransport.from_url(self.proxy),
+            transport=transport,
             headers=self.headers,
             cookies=self.cookies
         )