junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

default to not storing videos

Browse source
This commit is contained in:
Daniel Roth 2026-04-20 11:08:33 +00:00
parent b633de5feb
commit 2865320890
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
backend/pashub_fetcher/token_getter.py
View file

@ -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}")