更新

2026-07-17 09:24:47 +08:00
commit 530e7f839d
4353 changed files with 731879 additions and 0 deletions
@@ -0,0 +1,131 @@
+"""Playwright 浏览器引导：浏览器固定装在应用目录 playwright-browsers。"""
+from __future__ import annotations
+
+import os
+import subprocess
+import sys
+from pathlib import Path
+
+from app_paths import app_root, is_frozen
+
+ROOT = app_root()
+
+
+def browsers_dir() -> Path:
+    return ROOT / "playwright-browsers"
+
+
+def scrub_playwright_env() -> Path:
+    target = browsers_dir()
+    target.mkdir(parents=True, exist_ok=True)
+    for key in list(os.environ.keys()):
+        if key.upper().startswith("PLAYWRIGHT"):
+            os.environ.pop(key, None)
+    os.environ["PLAYWRIGHT_BROWSERS_PATH"] = str(target.resolve())
+    return target
+
+
+def find_chrome_exe(root: Path | None = None) -> Path | None:
+    base = root or browsers_dir()
+    if not base.is_dir():
+        return None
+    for folder in sorted(base.glob("chromium-*"), reverse=True):
+        for sub in ("chrome-win64", "chrome-win"):
+            exe = folder / sub / "chrome.exe"
+            if exe.is_file():
+                return exe.resolve()
+    return None
+
+
+def find_system_chrome() -> Path | None:
+    candidates = [
+        Path(os.environ.get("PROGRAMFILES", r"C:\Program Files")) / "Google/Chrome/Application/chrome.exe",
+        Path(os.environ.get("PROGRAMFILES(X86)", r"C:\Program Files (x86)")) / "Google/Chrome/Application/chrome.exe",
+        Path(os.environ.get("LOCALAPPDATA", "")) / "Google/Chrome/Application/chrome.exe",
+    ]
+    for path in candidates:
+        if path.is_file():
+            return path.resolve()
+    return None
+
+
+def python_exe() -> str:
+    if is_frozen():
+        return str(Path(sys.executable))
+    for rel in (".venv/Scripts/python.exe", "backend/.venv/Scripts/python.exe"):
+        p = ROOT / rel.replace("/", os.sep)
+        if p.is_file():
+            return str(p)
+    return str(Path(sys.executable))
+
+
+def install_chromium(target: Path) -> None:
+    print("正在安装 Playwright Chromium（约 130MB）…")
+    print(f"  {target.resolve()}\n")
+    env = os.environ.copy()
+    env["PLAYWRIGHT_BROWSERS_PATH"] = str(target.resolve())
+
+    if is_frozen():
+        from playwright._impl._driver import compute_driver_executable, get_driver_env
+
+        node, cli = compute_driver_executable()
+        env.update(get_driver_env())
+        env["PLAYWRIGHT_BROWSERS_PATH"] = str(target.resolve())
+        proc = subprocess.run([node, cli, "install", "chromium"], env=env)
+    else:
+        proc = subprocess.run(
+            [python_exe(), "-m", "playwright", "install", "chromium"],
+            env=env,
+            cwd=str(ROOT),
+        )
+
+    if proc.returncode != 0:
+        hint = "抖音IM凭证采集器.exe --install-browser" if is_frozen() else f"{python_exe()} -m playwright install chromium"
+        raise RuntimeError(f"Chromium 安装失败，请检查网络。\n手动执行：{hint}")
+
+
+def resolve_chrome_exe() -> Path:
+    target = scrub_playwright_env()
+
+    exe = find_chrome_exe(target)
+    if exe:
+        return exe
+
+    local = Path(os.environ.get("LOCALAPPDATA", "")) / "ms-playwright"
+    exe = find_chrome_exe(local)
+    if exe:
+        print(f"发现已有 Chromium：{exe}")
+        return exe
+
+    install_chromium(target)
+    exe = find_chrome_exe(target)
+    if exe:
+        return exe
+
+    system = find_system_chrome()
+    if system:
+        print(f"将使用本机 Google Chrome：{system}")
+        return system
+
+    if is_frozen():
+        raise RuntimeError("找不到 chrome.exe，请运行「修复浏览器.bat」或 exe --install-browser")
+    raise RuntimeError(f"找不到 chrome.exe，请运行：{ROOT / '修复Playwright浏览器.bat'}")
+
+
+def ensure_browser_ready() -> Path:
+    exe = resolve_chrome_exe()
+    print(f"浏览器就绪：{exe}")
+    return exe
+
+
+def browser_launch_kwargs() -> dict:
+    scrub_playwright_env()
+    exe = resolve_chrome_exe()
+    print(f"启动浏览器：{exe}")
+    return {
+        "executable_path": str(exe),
+        "env": {"PLAYWRIGHT_BROWSERS_PATH": str(browsers_dir().resolve())},
+    }
+
+
+fix_playwright_env = scrub_playwright_env