diff --git a/lib/utils/browser_use/__init__.py b/lib/utils/browser_use/__init__.py index 11168e1..c38ca03 100644 --- a/lib/utils/browser_use/__init__.py +++ b/lib/utils/browser_use/__init__.py @@ -2,11 +2,10 @@ from lib.utils.browser_use.func import * # Initialize configuration proxy_url = setup_proxy() -storage_state_path = setup_storage_state() # Create browser profile -def GetProfile(): - setup_storage_state() +async def GetProfile(): + storage_state_path = await setup_storage_state() profile = BrowserProfile( # Security settings disable_security=True, diff --git a/lib/utils/browser_use/func.py b/lib/utils/browser_use/func.py index f44b2dd..31eee31 100644 --- a/lib/utils/browser_use/func.py +++ b/lib/utils/browser_use/func.py @@ -20,7 +20,7 @@ def setup_proxy(): return None -def setup_storage_state(): +async def setup_storage_state(): """Setup browser storage state for session persistence.""" # Get the script directory to ensure correct path resolution script_dir = Path(__file__).parent.parent.parent.parent diff --git a/main.py b/main.py index 8b150a1..ccc70bf 100644 --- a/main.py +++ b/main.py @@ -84,7 +84,7 @@ signal.signal(signal.SIGINT, signal_handler) # ── URL별로 Browser를 새로 띄우는 함수 ── async def scan_one_url(url: str, skip_html_check: bool = False): - setup_storage_state() + await setup_storage_state() target_url = url if url.startswith("http") else f"https://{url}" print(f"🚀 Starting scan for: {target_url}") @@ -103,7 +103,7 @@ async def scan_one_url(url: str, skip_html_check: bool = False): # BrowserSession에 profile 전달 session = BrowserSession( playwright=(await async_patchright().start()), - browser_profile=browser_use.GetProfile(), + browser_profile=await browser_use.GetProfile(), ) # Agent 생성 및 실행 (단일 try-except with 백오프)