新内容

browser_login/auto_launcher.py

@@ -0,0 +1,80 @@
+"""
+ERP 数据同步器 - 自动化浏览器拉起与登录守护模块
+目标:
+1. 自动寻找本地安装的 Chrome 浏览器。
+2. 以 9222 端口和独立的用户数据目录启动（互不干扰，持久化登录状态）。
+3. 弹出 ERP 登录页面，等待用户手动登录（解决滑块验证码等问题）。
+4. 登录成功后，将浏览器挂在后台作为保活引擎。
+"""
+import sys
+import time
+from pathlib import Path
+from DrissionPage import ChromiumOptions, ChromiumPage
+from config import DATA_DIR
+
+# 独立的用户数据目录，放在外部持久化目录，这样重启不会丢失登录状态
+USER_DATA_DIR = DATA_DIR / "browser_login" / "chrome_user_data"
+HOME_URL = "https://yunmes.tftykj.cn/"
+
+def start_and_wait_login():
+    print("🚀 [1/3] 正在配置内置 Chrome 浏览器引擎...")
+    
+    # 初始化配置
+    co = ChromiumOptions()
+    co.set_local_port(9222)
+    # 指定一个独立的用户数据存放文件夹
+    co.set_user_data_path(str(USER_DATA_DIR))
+    # 忽略证书错误等常规反爬配置
+    co.ignore_certificate_errors()
+    
+    print("🌍 [2/3] 正在拉起浏览器并前往 ERP 登录页...")
+    try:
+        # 这一步会自动寻找你电脑上的 Chrome，如果没有运行在 9222 端口，它会自动帮你启动一个！
+        page = ChromiumPage(co)
+    except Exception as e:
+        print(f"❌ 启动浏览器失败，请确保电脑安装了 Chrome 浏览器！报错信息: {e}")
+        return None
+        
+    page.get(HOME_URL)
+    
+    print("\n" + "="*50)
+    print("👀 [等待人工介入] 请在弹出的浏览器窗口中完成登录操作！")
+    print("💡 提示: 输入账号密码、通过滑块验证码，直到进入 ERP 系统主界面。")
+    print("="*50 + "\n")
+    
+    # 循环检查登录状态
+    # 怎么判断登录成功？ERP 登录前 URL 通常带有 Login 等字样，或者登录后页面会出现类似“退出”、“注销”或者用户名的元素
+    # 我们这里通过检测页面中是否出现了主菜单的特有元素，或者通过监听一个登录后的特有接口来判断
+    is_logged_in = False
+    
+    while not is_logged_in:
+        time.sleep(2)
+        # 假设登录后页面会出现“首页”或者用户的头像/名字（这里的 xpath 需要根据你们 ERP 登录后的实际情况微调，我们先用一个保险的：看看有没有业务统计报表的菜单）
+        # 如果还在登录页，肯定找不到这个元素
+        try:
+            # 这里的元素用来验证是否已经成功进入系统内部
+            menu_ele = page.ele('xpath://*[@id="app"]/div/div[1]/div[1]/div[2]/div/div[1]/div/div[10]/div/p', timeout=1)
+            if menu_ele:
+                is_logged_in = True
+        except:
+            pass
+            
+        if not is_logged_in:
+            print("⏳ 等待登录中...")
+            
+    print("\n✅ [3/3] 检测到登录成功！")
+    print("🔒 登录状态已保存，你可以随时关闭或者最小化这个浏览器窗口。")
+    print("🤖 爬虫引擎已挂载至后台，可以开始点击前端界面的【同步数据】按钮了！\n")
+    
+    return page
+
+if __name__ == "__main__":
+    # 单独运行此文件即可拉起浏览器
+    page = start_and_wait_login()
+    if page:
+        # 为了不让脚本退出，这里写个死循环保活（在真正的桌面软件中，这里就是启动 Flask 后端和 Webview 窗口的地方）
+        try:
+            while True:
+                time.sleep(10)
+        except KeyboardInterrupt:
+            print("👋 守护进程已退出。")