Merge branch 'main' of https://gitea.butubb.cn/butubb/vzan_autoreplay

main.py

@@ -2,7 +2,7 @@ import json
 import os
 import time
 import requests
-from seleniumwire import webdriver  # pip install selenium-wire
+from seleniumwire import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
@@ -44,10 +44,26 @@ def get_auth_from_browser():
     driver.requests.clear()
 
     driver.get("https://live.vzan.com/admin/index.html")
-    input("请手动登录网站，登录完成后按回车继续...")
+    
+    log("请在浏览器窗口中手动登录...", "INFO")
+
+    # 自动检测是否登录成功
+    timeout = 120  # 最多等 120 秒
+    start_time = time.time()
+    while True:
+        if "admin/index.html" in driver.current_url:
+            # 登录后会进入后台页面
+            log("检测到已进入后台管理页面，继续获取认证信息", "INFO")
+            break
+        if time.time() - start_time > timeout:
+            driver.quit()
+            raise Exception("登录超时，请检查账号密码或验证码")
+        time.sleep(2)
+
     driver.get(ADMIN_URL)
     log("等待页面加载完成...", "INFO")
 
+
     # 捕获 headers 和 cookies
     timeout = 60
     start_time = time.time()