dongzhaorui 3 年 前
コミット
d646a0d1a6
1 ファイル変更6 行追加3 行削除
  1. 6 3
      zbytb/utils/socks5.py

+ 6 - 3
zbytb/utils/socks5.py

@@ -27,13 +27,16 @@ class Socks5Proxy:
                 self._proxies = self._fetch_proxies()
 
     def _fetch_proxies(self):
-        proxy = requests.get(self._url, headers=self._auth).json()
-        return proxy.get("data")
+        _proxy = {}
+        try:
+            _proxy = requests.get(self._url, headers=self._auth, timeout=10).json()
+        finally:
+            return _proxy.get("data")
 
     def __call__(self, enable_proxy: bool = False, *args, **kwargs):
         self._enable_proxy = enable_proxy
         if self._enable_proxy:
-            logger.info("[加载socks5代理]")
+            logger.info("[socks5代理 - 开启]")
             self._proxies = self._fetch_proxies()
         return self