feat: 票据合规预审助手首版 v1.0.0（verify-invoice-default）

examples/simulator_browser_rpa/scripts/service/browser_session.py

@@ -0,0 +1,126 @@
+"""浏览器会话启动（系统 Chrome/Edge + async Playwright persistent context）。"""
+
+from __future__ import annotations
+
+import logging
+import os
+from typing import Any, List, Optional, Tuple
+
+logger = logging.getLogger(__name__)
+
+DEFAULT_PAGE_TIMEOUT_MS = 15_000
+
+CHROME_LAUNCH_ARGS: List[str] = [
+    "--start-maximized",
+    "--disable-blink-features=AutomationControlled",
+]
+
+STEALTH_INIT_SCRIPT = """
+(() => {
+  try { Object.defineProperty(navigator, 'webdriver', { get: () => undefined }); } catch (e) {}
+})();
+"""
+
+
+def chrome_launch_args() -> List[str]:
+    return list(CHROME_LAUNCH_ARGS)
+
+
+def _headless_from_env() -> bool:
+    v = (os.getenv("OPENCLAW_BROWSER_HEADLESS") or "0").strip().lower()
+    return v in ("1", "true", "yes", "on")
+
+
+def find_chrome_executable() -> str | None:
+    try:
+        from jiangchang_skill_core.runtime_env import find_chrome_executable as _find
+
+        chrome = _find()
+        if chrome and os.path.isfile(chrome):
+            return chrome
+    except ImportError:
+        pass
+
+    candidates = [
+        os.path.expandvars(r"%ProgramFiles%\Google\Chrome\Application\chrome.exe"),
+        os.path.expandvars(r"%ProgramFiles(x86)%\Google\Chrome\Application\chrome.exe"),
+        os.path.expandvars(r"%ProgramFiles%\Microsoft\Edge\Application\msedge.exe"),
+        os.path.expandvars(r"%ProgramFiles(x86)%\Microsoft\Edge\Application\msedge.exe"),
+    ]
+    for path in candidates:
+        if path and os.path.isfile(path):
+            return path
+    return None
+
+
+def _clear_node_options() -> None:
+    os.environ.pop("NODE_OPTIONS", None)
+
+
+async def start_browser_session(
+    profile_dir: str,
+    *,
+    headless: Optional[bool] = None,
+) -> Tuple[Any, Any, Any]:
+    """
+    启动持久化 Chrome profile，返回 (playwright, context, page)。
+
+    URL 不在此文件打开；由 simulator_playwright.py 负责 page.goto。
+    """
+    from playwright.async_api import async_playwright
+
+    _clear_node_options()
+    hl = headless if headless is not None else _headless_from_env()
+    chrome = find_chrome_executable()
+    if not chrome:
+        raise RuntimeError(
+            "ERROR:MISSING_BROWSER 未检测到 Chrome 或 Edge 浏览器，请先安装系统浏览器后重试。"
+        )
+
+    args = chrome_launch_args()
+    logger.info(
+        "browser_launch chrome=%s profile_dir=%s args=%s headless=%s",
+        chrome,
+        profile_dir,
+        args,
+        hl,
+    )
+
+    pw = await async_playwright().start()
+    context = None
+    try:
+        context = await pw.chromium.launch_persistent_context(
+            user_data_dir=profile_dir,
+            headless=hl,
+            executable_path=chrome,
+            locale="zh-CN",
+            no_viewport=True,
+            args=args,
+            ignore_default_args=["--enable-automation"],
+        )
+        await context.add_init_script(STEALTH_INIT_SCRIPT)
+
+        page = context.pages[0] if context.pages else await context.new_page()
+        page.set_default_timeout(DEFAULT_PAGE_TIMEOUT_MS)
+        return pw, context, page
+    except Exception:
+        if context is not None:
+            try:
+                await context.close()
+            except Exception:
+                pass
+        await pw.stop()
+        raise
+
+
+async def close_browser_session(playwright: Any, context: Any) -> None:
+    try:
+        if context is not None:
+            await context.close()
+    except Exception:
+        pass
+    try:
+        if playwright is not None:
+            await playwright.stop()
+    except Exception:
+        pass