7d7862f4b5
Import-surface docs and docstrings describe the automation as a migration script; status-doc references in CHANGELOG/ADR-009/ROADMAP point at repo files. Screenshot/verify tools write to /tmp/shopdb-shots (created on import) instead of a machine-specific directory.
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
"""Drive the dev UI: open the map, click Export PDF, save the download + a shot."""
|
|
import os
|
|
import json
|
|
import urllib.request
|
|
from playwright.sync_api import sync_playwright
|
|
|
|
UI = "http://localhost:5173"
|
|
API = "http://localhost:5001/api"
|
|
USERNAME = "270015376"
|
|
PASSWORD = "changeme"
|
|
OUT = "/tmp/shopdb-shots"
|
|
os.makedirs(OUT, exist_ok=True)
|
|
|
|
|
|
def login():
|
|
body = json.dumps({"username": USERNAME, "password": PASSWORD}).encode()
|
|
httprequest = urllib.request.Request(f"{API}/auth/login", data=body,
|
|
headers={"Content-Type": "application/json"})
|
|
return json.load(urllib.request.urlopen(httprequest))["data"]
|
|
|
|
|
|
def main():
|
|
data = login()
|
|
seed = f"""
|
|
localStorage.setItem('token', {json.dumps(data['access_token'])});
|
|
localStorage.setItem('refreshToken', {json.dumps(data['refresh_token'])});
|
|
localStorage.setItem('user', {json.dumps(json.dumps(data['user']))});
|
|
"""
|
|
with sync_playwright() as p:
|
|
browser = p.chromium.launch()
|
|
context = browser.new_context(viewport={"width": 1500, "height": 1000},
|
|
accept_downloads=True)
|
|
context.add_init_script(seed)
|
|
page = context.new_page()
|
|
page.on("console", lambda m: print("CONSOLE", m.type, m.text) if m.type == "error" else None)
|
|
|
|
page.goto(f"{UI}/map", wait_until="networkidle", timeout=30000)
|
|
# Apply a filter to prove the PDF respects it: select Equipment.
|
|
page.wait_for_timeout(1500)
|
|
page.select_option("select", label=lambda: True) if False else None
|
|
page.screenshot(path=f"{OUT}/shot_map_page.png", full_page=False)
|
|
|
|
with page.expect_download(timeout=30000) as dl_info:
|
|
page.get_by_role("button", name="Export PDF").click()
|
|
download = dl_info.value
|
|
pdf_path = f"{OUT}/{download.suggested_filename}"
|
|
download.save_as(pdf_path)
|
|
print("SAVED_PDF", pdf_path)
|
|
browser.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|