[Enhancement] setup_storage_state 및 GetProfile 함수를 비동기 처리로 변경

This commit is contained in:
imnyang 2025-06-15 22:37:35 +09:00
commit 7ae5834d82
3 changed files with 5 additions and 6 deletions

View file

@ -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,

View file

@ -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

View file

@ -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 백오프)