junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-07-12 13:29:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'main' into feature/read-site-notes-pdf

Browse source
This commit is contained in:
Daniel Roth 2026-04-20 11:19:49 +00:00
commit c11c85a974
1 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
backend/pashub_fetcher/token_getter.py
View file

@ -7,23 +7,27 @@ from utils.logger import setup_logger
logger = 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") 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: with sync_playwright() as p:
browser = p.chromium.launch( browser = p.chromium.launch(
headless=True, headless=True,
args=["--no-sandbox", "--disable-dev-shm-usage"], args=["--no-sandbox", "--disable-dev-shm-usage"],
) )
context = browser.new_context( video_dir = None
record_video_dir=video_dir, if record_video:
record_video_size={"width": 1280, "height": 720}, video_dir = os.path.join(os.path.dirname(__file__), "videos")
) os.makedirs(video_dir, exist_ok=True)
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() page = context.new_page()
try: try:
@ -71,8 +75,8 @@ def get_token_from_local_storage(email: str, password: str) -> str:
raise Exception(f"Unexpected error: {str(e)}") raise Exception(f"Unexpected error: {str(e)}")
finally: finally:
logger.info("Closing browser context (saving video)...")
context.close() context.close()
browser.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}")