diff --git a/browser_login/login.py b/browser_login/login.py index 22d0faa..e47380b 100644 --- a/browser_login/login.py +++ b/browser_login/login.py @@ -21,7 +21,6 @@ from websocket import create_connection as raw_ws_create_connection, WebSocketBa # ── 加载 .env ───────────────────────────────────────────────────────────────── load_dotenv(Path(__file__).parent / ".env") -patch_drission_ws_handshake() # 强制 Python 的 websocket 客户端忽略本地代理,防止出现 Handshake status 404 Not Found os.environ["NO_PROXY"] = "localhost,127.0.0.1,::1" @@ -86,6 +85,9 @@ def patch_drission_ws_handshake() -> None: dp_driver_module._DTSK_WS_PATCHED = True +patch_drission_ws_handshake() + + def get_docker_tmp_root() -> Path: """ 指定 DrissionPage 在 Docker 中的临时根目录。