flash1
This commit is contained in:
nafer committed
1 parent
61e8f7b490
commit
5b9e9ffc29
1 file changed
+82
-90
@@ -85,101 +85,93 @@ async def run_browser_task(task_id: str, task: str, max_steps: int, model: str):
|
|||||||
import urllib.request
|
import urllib.request
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
print(f"[{task_id}] Starting Chrome with Remote Debugging on port {REMOTE_DEBUGGING_PORT}...")
|
|
||||||
|
|
||||||
# Список возможных путей к Chrome
|
|
||||||
chrome_exe = None
|
|
||||||
possible_paths = [
|
|
||||||
"/root/.cache/ms-playwright/chromium-1208/chrome-linux64/chrome",
|
|
||||||
"/root/.cache/ms-playwright/chromium-*/chrome-linux64/chrome",
|
|
||||||
"/ms-playwright/*/chrome-linux/chrome",
|
|
||||||
"/usr/bin/google-chrome",
|
|
||||||
"/usr/bin/chromium",
|
|
||||||
"/usr/bin/chromium-browser",
|
|
||||||
]
|
|
||||||
|
|
||||||
for path_pattern in possible_paths:
|
|
||||||
matches = glob.glob(path_pattern)
|
|
||||||
if matches:
|
|
||||||
chrome_exe = matches[0]
|
|
||||||
break
|
|
||||||
|
|
||||||
if not chrome_exe:
|
|
||||||
# Попробуем найти через shutil
|
|
||||||
import shutil
|
|
||||||
chrome_exe = shutil.which("google-chrome") or shutil.which("chromium") or shutil.which("chromium-browser")
|
|
||||||
|
|
||||||
if not chrome_exe:
|
|
||||||
raise RuntimeError(f"Chrome not found. Please ensure playwright browsers are installed.")
|
|
||||||
|
|
||||||
print(f"[{task_id}] Using Chrome executable: {chrome_exe}")
|
|
||||||
|
|
||||||
# Создаем уникальную директорию для профиля (чтобы избежать блокировки)
|
|
||||||
import uuid
|
|
||||||
unique_id = str(uuid.uuid4())[:8]
|
|
||||||
user_data_dir = f"/tmp/browser-data-{unique_id}"
|
|
||||||
os.makedirs(user_data_dir, exist_ok=True)
|
|
||||||
|
|
||||||
# Запускаем Chrome с remote debugging
|
|
||||||
chrome_cmd = [
|
|
||||||
chrome_exe,
|
|
||||||
f"--remote-debugging-port={REMOTE_DEBUGGING_PORT}",
|
|
||||||
f"--remote-debugging-address=0.0.0.0",
|
|
||||||
f"--user-data-dir={user_data_dir}",
|
|
||||||
"--no-first-run",
|
|
||||||
"--no-default-browser-check",
|
|
||||||
"--disable-extensions",
|
|
||||||
"--disable-gpu",
|
|
||||||
"--no-sandbox",
|
|
||||||
"--disable-dev-shm-usage",
|
|
||||||
"--headless",
|
|
||||||
"about:blank"
|
|
||||||
]
|
|
||||||
|
|
||||||
print(f"[{task_id}] Executing command: {' '.join(chrome_cmd)}")
|
|
||||||
|
|
||||||
# Запускаем процесс
|
|
||||||
process = subprocess.Popen(
|
|
||||||
chrome_cmd,
|
|
||||||
stdout=subprocess.PIPE,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
text=True
|
|
||||||
)
|
|
||||||
|
|
||||||
# Ждем пока Chrome запустится и CDP будет готов
|
|
||||||
cdp_ready = False
|
|
||||||
cdp_url = f"http://127.0.0.1:{REMOTE_DEBUGGING_PORT}"
|
cdp_url = f"http://127.0.0.1:{REMOTE_DEBUGGING_PORT}"
|
||||||
|
|
||||||
for i in range(30): # Ждем до 30 секунд
|
# 1. Проверяем, не запущен ли уже Chrome с CDP
|
||||||
await asyncio.sleep(1)
|
print(f"[{task_id}] Checking if Chrome is already running with CDP at {cdp_url}...")
|
||||||
|
chrome_already_running = False
|
||||||
|
try:
|
||||||
|
response = urllib.request.urlopen(f"{cdp_url}/json/version", timeout=2)
|
||||||
|
if response.status == 200:
|
||||||
|
print(f"[{task_id}] Chrome is already running, connecting to existing instance.")
|
||||||
|
chrome_already_running = True
|
||||||
|
except Exception:
|
||||||
|
print(f"[{task_id}] Chrome not detected at {cdp_url}. Starting new instance...")
|
||||||
|
|
||||||
|
if not chrome_already_running:
|
||||||
|
# Список возможных путей к Chrome
|
||||||
|
chrome_exe = None
|
||||||
|
possible_paths = [
|
||||||
|
"/root/.cache/ms-playwright/chromium-1208/chrome-linux64/chrome",
|
||||||
|
"/root/.cache/ms-playwright/chromium-*/chrome-linux64/chrome",
|
||||||
|
"/ms-playwright/*/chrome-linux/chrome",
|
||||||
|
"/usr/bin/google-chrome",
|
||||||
|
"/usr/bin/chromium",
|
||||||
|
"/usr/bin/chromium-browser",
|
||||||
|
]
|
||||||
|
|
||||||
# Проверяем, не упал ли процесс
|
for path_pattern in possible_paths:
|
||||||
if process.poll() is not None:
|
matches = glob.glob(path_pattern)
|
||||||
stdout, stderr = process.communicate()
|
if matches:
|
||||||
raise RuntimeError(
|
chrome_exe = matches[0]
|
||||||
f"Chrome process exited unexpectedly with code {process.returncode}.\n"
|
|
||||||
f"STDOUT: {stdout}\n"
|
|
||||||
f"STDERR: {stderr}"
|
|
||||||
)
|
|
||||||
|
|
||||||
try:
|
|
||||||
response = urllib.request.urlopen(f"{cdp_url}/json/version", timeout=2)
|
|
||||||
if response.status == 200:
|
|
||||||
cdp_ready = True
|
|
||||||
print(f"[{task_id}] CDP is ready at {cdp_url}")
|
|
||||||
break
|
break
|
||||||
except Exception:
|
|
||||||
if i % 5 == 0:
|
if not chrome_exe:
|
||||||
print(f"[{task_id}] Waiting for CDP at {cdp_url}... (attempt {i})")
|
import shutil
|
||||||
|
chrome_exe = shutil.which("google-chrome") or shutil.which("chromium") or shutil.which("chromium-browser")
|
||||||
if not cdp_ready:
|
|
||||||
# Получаем логи Chrome
|
if not chrome_exe:
|
||||||
stdout, stderr = process.communicate(timeout=5)
|
raise RuntimeError(f"Chrome not found. Please ensure playwright browsers are installed.")
|
||||||
raise RuntimeError(
|
|
||||||
f"Chrome started but CDP not ready after 30s at {cdp_url}.\n"
|
print(f"[{task_id}] Using Chrome executable: {chrome_exe}")
|
||||||
f"STDOUT: {stdout}\n"
|
|
||||||
f"STDERR: {stderr}"
|
# Используем постоянную директорию для сохранения сессий (логинов)
|
||||||
|
user_data_dir = "/app/browser-data/main_profile"
|
||||||
|
os.makedirs(user_data_dir, exist_ok=True)
|
||||||
|
|
||||||
|
# Запускаем Chrome с remote debugging
|
||||||
|
chrome_cmd = [
|
||||||
|
chrome_exe,
|
||||||
|
f"--remote-debugging-port={REMOTE_DEBUGGING_PORT}",
|
||||||
|
f"--remote-debugging-address=0.0.0.0",
|
||||||
|
f"--user-data-dir={user_data_dir}",
|
||||||
|
"--remote-allow-origins=*",
|
||||||
|
"--no-first-run",
|
||||||
|
"--no-default-browser-check",
|
||||||
|
"--disable-extensions",
|
||||||
|
"--disable-gpu",
|
||||||
|
"--no-sandbox",
|
||||||
|
"--disable-dev-shm-usage",
|
||||||
|
"--headless",
|
||||||
|
"about:blank"
|
||||||
|
]
|
||||||
|
|
||||||
|
print(f"[{task_id}] Executing command: {' '.join(chrome_cmd)}")
|
||||||
|
|
||||||
|
# Запускаем процесс (без блокировки, в фоне)
|
||||||
|
subprocess.Popen(
|
||||||
|
chrome_cmd,
|
||||||
|
stdout=subprocess.DEVNULL,
|
||||||
|
stderr=subprocess.DEVNULL,
|
||||||
|
start_new_session=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Ждем пока Chrome запустится и CDP будет готов
|
||||||
|
cdp_ready = False
|
||||||
|
for i in range(30): # Ждем до 30 секунд
|
||||||
|
await asyncio.sleep(1)
|
||||||
|
try:
|
||||||
|
response = urllib.request.urlopen(f"{cdp_url}/json/version", timeout=2)
|
||||||
|
if response.status == 200:
|
||||||
|
cdp_ready = True
|
||||||
|
print(f"[{task_id}] CDP is ready at {cdp_url}")
|
||||||
|
break
|
||||||
|
except Exception:
|
||||||
|
if i % 5 == 0:
|
||||||
|
print(f"[{task_id}] Waiting for CDP at {cdp_url}... (attempt {i})")
|
||||||
|
|
||||||
|
if not cdp_ready:
|
||||||
|
raise RuntimeError(f"Failed to start Chrome or CDP not ready after 30s at {cdp_url}")
|
||||||
|
|
||||||
# Подключаемся через CDP
|
# Подключаемся через CDP
|
||||||
from browser_use import BrowserConfig
|
from browser_use import BrowserConfig
|
||||||
|
|||||||
Reference in new issue
Block a user