diff --git a/backend/pashub_fetcher/token_getter.py b/backend/pashub_fetcher/token_getter.py index 3f6c63e6..ebc7506f 100644 --- a/backend/pashub_fetcher/token_getter.py +++ b/backend/pashub_fetcher/token_getter.py @@ -21,10 +21,13 @@ def get_token_from_local_storage(email: str, password: str, record_video: bool = 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} if record_video else None, - ) + if record_video: + context = browser.new_context( + record_video_dir=video_dir, + record_video_size={"width": 1280, "height": 720}, + ) + else: + context = browser.new_context() page = context.new_page() try: