75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""Unified Chromium launch with stealth + persistent profile for RPA helpers."""
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
from typing import TYPE_CHECKING
|
||
|
||
from playwright.sync_api import sync_playwright
|
||
|
||
from service.browser_service import resolve_chromium_channel
|
||
from util.playwright_stealth import (
|
||
STEALTH_INIT_SCRIPT,
|
||
persistent_context_launch_parts,
|
||
stealth_enabled,
|
||
)
|
||
|
||
if TYPE_CHECKING:
|
||
from playwright.sync_api import BrowserContext, Page
|
||
|
||
|
||
def launch_browser_with_profile(
|
||
profile_dir: str,
|
||
*,
|
||
channel: str | None = None,
|
||
headless: bool = False,
|
||
extra_args: list[str] | None = None,
|
||
locale: str = "zh-CN",
|
||
) -> tuple["BrowserContext", "Page"]:
|
||
"""启动浏览器（Chrome 优先 / Edge fallback），返回 (context, page)。
|
||
|
||
调用方应在 finally 中调用 close_browser(context)。
|
||
"""
|
||
os.makedirs(profile_dir, exist_ok=True)
|
||
|
||
resolved_channel = channel if channel is not None else resolve_chromium_channel()
|
||
if not resolved_channel:
|
||
raise RuntimeError("no chrome/edge found on system")
|
||
|
||
_pw = sync_playwright().start()
|
||
try:
|
||
args, ignore_automation = persistent_context_launch_parts(extra_args=extra_args)
|
||
launch_kwargs: dict = dict(
|
||
user_data_dir=profile_dir,
|
||
headless=headless,
|
||
channel=resolved_channel,
|
||
no_viewport=True,
|
||
locale=locale,
|
||
args=args,
|
||
)
|
||
if ignore_automation is not None:
|
||
launch_kwargs["ignore_default_args"] = ignore_automation
|
||
ctx = _pw.chromium.launch_persistent_context(**launch_kwargs)
|
||
if stealth_enabled():
|
||
ctx.add_init_script(STEALTH_INIT_SCRIPT)
|
||
page = ctx.pages[0] if ctx.pages else ctx.new_page()
|
||
setattr(ctx, "_openclaw_pw", _pw)
|
||
return ctx, page
|
||
except Exception:
|
||
_pw.stop()
|
||
raise
|
||
|
||
|
||
def close_browser(ctx) -> None:
|
||
"""关闭 context 并清理 Playwright manager。"""
|
||
pw = getattr(ctx, "_openclaw_pw", None)
|
||
try:
|
||
ctx.close()
|
||
except Exception:
|
||
pass
|
||
if pw is not None:
|
||
try:
|
||
pw.stop()
|
||
except Exception:
|
||
pass
|