mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError
|
|
|
|
from utils.logger import setup_logger
|
|
|
|
logger = setup_logger()
|
|
|
|
|
|
def get_token_from_local_storage(email: str, password: str) -> str:
|
|
logger.info("Starting Playwright flow")
|
|
|
|
with sync_playwright() as p:
|
|
browser = p.chromium.launch(
|
|
headless=True,
|
|
args=["--no-sandbox", "--disable-dev-shm-usage"],
|
|
)
|
|
page = browser.new_page()
|
|
|
|
try:
|
|
logger.info("Navigating to site...")
|
|
page.goto("https://pashub.net/", timeout=30000)
|
|
|
|
logger.info("Filling login form...")
|
|
page.fill("#email", email)
|
|
page.fill("#password", password)
|
|
|
|
logger.info("Submitting login...")
|
|
page.click("#btn-login")
|
|
|
|
page.wait_for_timeout(3000)
|
|
|
|
if "login" in page.url.lower():
|
|
raise Exception("Login failed (still on login page)")
|
|
|
|
logger.info(f"Login likely successful. URL: {page.url}")
|
|
|
|
token = page.evaluate(
|
|
"""() => {
|
|
return localStorage.getItem('token');
|
|
}"""
|
|
)
|
|
|
|
if not token:
|
|
raise Exception("Login succeeded but no token found")
|
|
|
|
return token
|
|
|
|
except PlaywrightTimeoutError as e:
|
|
raise Exception(f"Timeout during login flow: {str(e)}")
|
|
|
|
except Exception as e:
|
|
raise Exception(f"Unexpected error: {str(e)}")
|
|
|
|
finally:
|
|
browser.close()
|