File size: 3,975 Bytes
7d3d63c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | import asyncio
import os
from pathlib import Path
from playwright.async_api import Page, async_playwright
from metrics.metrics import metrics_counter_inc
from neo.utils import logger
_BEDROCK_PROJECT = os.environ.get("BEDROCK_PROJECT", "")
def is_bedrock_env() -> bool:
return _BEDROCK_PROJECT != ""
async def handle_new_page(page: Page):
"""
Handle new page events and execute custom logic
"""
print(f"New page created: {page.url}")
async def launch_chrome_debug(use_chrome_channel: bool = False, headless: bool = False):
"""
Launch Chrome browser with remote debugging enabled on port 9222
Returns the browser instance when launched successfully
"""
try:
extension_path = Path(os.path.dirname(__file__)).joinpath("browser_extension/error_capture") # type: ignore
playwright = await async_playwright().start()
disable_security_args = [
"--disable-web-security",
"--disable-site-isolation-trials",
"--disable-features=IsolateOrigins,site-per-process",
]
workspace = "/workspace" if is_bedrock_env() else "./workspace"
user_data_dir = os.path.join(workspace, "browser", "user_data")
# 删除浏览器单例锁文件（如果存在），避免从NAS恢复的旧锁文件导致冲突
# 使用 lexists 而不是 exists，因为这些文件可能是指向不存在目标的符号链接
singleton_files = ["SingletonLock", "SingletonSocket", "SingletonCookie"]
for filename in singleton_files:
file_path = os.path.join(user_data_dir, filename)
try:
if os.path.lexists(file_path):
os.remove(file_path)
logger.info(f"已删除浏览器单例文件: {file_path}")
except Exception as e:
logger.warning(f"删除浏览器单例文件失败 {file_path}: {str(e)}")
context = await playwright.chromium.launch_persistent_context(
user_data_dir=user_data_dir,
headless=headless,
viewport={"width": 1280, "height": 720},
args=[
"--no-sandbox",
"--disable-blink-features=AutomationControlled",
"--disable-infobars",
"--disable-background-timer-throttling",
"--disable-popup-blocking",
"--disable-backgrounding-occluded-windows",
"--disable-renderer-backgrounding",
"--disable-window-activation",
"--disable-focus-on-load",
"--no-first-run",
"--no-default-browser-check",
"--window-position=0,0",
]
+ disable_security_args
+ [
f"--disable-extensions-except={extension_path}",
f"--load-extension={extension_path}",
"--disable-web-security",
"--disable-site-isolation-trials",
"--remote-debugging-port=9222",
"--remote-debugging-address=127.0.0.1", # 仅允许本地访问，防止外部连接
],
channel="chromium" if not use_chrome_channel else "chrome",
# proxy={"server": "http://data-capture-online.xaminim.com:3160", "username": "default-user", "password": "default"},
)
metrics_counter_inc("agent_browser_launch", {"status": "success"})
# 监听新页面事件
context.on("page", handle_new_page)
# 处理已经打开的页面
for page in context.pages:
await handle_new_page(page)
# Keep browser process alive
while True:
await asyncio.sleep(1000)
except Exception as e:
logger.exception(f"Failed to launch Chrome browser: {str(e)}")
metrics_counter_inc("agent_browser_launch", {"status": "failed"})
raise
if __name__ == "__main__":
asyncio.run(launch_chrome_debug())
|