diff --git a/backend/pashub_fetcher/token_getter.py b/backend/pashub_fetcher/token_getter.py index 5534d114..3f6c63e6 100644 --- a/backend/pashub_fetcher/token_getter.py +++ b/backend/pashub_fetcher/token_getter.py @@ -7,22 +7,23 @@ from utils.logger import setup_logger logger = setup_logger() -def get_token_from_local_storage(email: str, password: str) -> str: +def get_token_from_local_storage(email: str, password: str, record_video: bool = False) -> str: logger.info("Starting Playwright flow") - # For local testing / debugging, we save videos of the flow - video_dir = os.path.join(os.path.dirname(__file__), "videos") - os.makedirs(video_dir, exist_ok=True) - with sync_playwright() as p: browser = p.chromium.launch( headless=True, args=["--no-sandbox", "--disable-dev-shm-usage"], ) + video_dir = None + if record_video: + video_dir = os.path.join(os.path.dirname(__file__), "videos") + os.makedirs(video_dir, exist_ok=True) + context = browser.new_context( record_video_dir=video_dir, - record_video_size={"width": 1280, "height": 720}, + record_video_size={"width": 1280, "height": 720} if record_video else None, ) page = context.new_page() @@ -71,8 +72,8 @@ def get_token_from_local_storage(email: str, password: str) -> str: raise Exception(f"Unexpected error: {str(e)}") finally: - logger.info("Closing browser context (saving video)...") context.close() browser.close() - logger.info(f"Video(s) saved in: {video_dir}") + if record_video and video_dir: + logger.info(f"Video(s) saved in: {video_dir}")