diff --git a/backend/pashub_fetcher/handler/handler.py b/backend/pashub_fetcher/handler/handler.py index 3d22ef44..a5cbe55a 100644 --- a/backend/pashub_fetcher/handler/handler.py +++ b/backend/pashub_fetcher/handler/handler.py @@ -40,7 +40,19 @@ def handler(event: Mapping[str, Any], context: Any) -> Dict[str, str]: logger.info(f"Login likely successful. URL: {page.url}") - return {"status": "ok"} + token = page.evaluate( + """() => { + return localStorage.getItem('token'); + }""" + ) + + if not token: + logger.error("Login succeeded but no token found") + return {"status": "error", "message": "No token found"} + + logger.info(f"Token extracted successfully: {token}") + + return {"status": "ok", "token": token} except PlaywrightTimeoutError as e: logger.error(f"Timeout during login flow: {str(e)}")