diff --git a/spider/task.py b/spider/task.py index 5bd4acd..00c9e28 100644 --- a/spider/task.py +++ b/spider/task.py @@ -24,6 +24,7 @@ import const from const import BUCKET, BASE_PATH from exceptions import AuthException, OperationFailed from miniofile import client, put_object +from spider.proxy_valid import ProxyChecker def sleep(a, b=None): @@ -852,7 +853,9 @@ def playwright_m_login(username, password, code_2fa=None): with lock: with sync_playwright() as playwright: update_windows_distinguish() - proxy_url = get_proxy_from_api() + proxy_url = ProxyChecker(timeout=8).get_valid_proxy_url() + if proxy_url is None: + raise OperationFailed("获取代理失败") logger.info(f"使用proxy={proxy_url}") browser = playwright.chromium.launch(