z230

IWRS/download_shipments_report.py

@@ -13,43 +13,35 @@ STUDY = "77242113UCO3001"
 OUTPUT_DIR = f"xls_shipments_{STUDY}"
 # ────────────────────────────────────────────────────────────────────────────
 
-os.makedirs(OUTPUT_DIR, exist_ok=True)
+def run(page, study):
+    output_dir = f"xls_shipments_{study}"
+    os.makedirs(output_dir, exist_ok=True)
+
+    page.goto(f"{BASE_URL}/report/shipments_report")
+    page.wait_for_load_state("networkidle", timeout=120000)
+
+    filename = os.path.join(output_dir, f"shipments_report_{study}.xlsx")
+    with page.expect_download(timeout=120000) as dl:
+        page.get_by_role("button", name="Download XLS").click()
+    dl.value.save_as(filename)
+    print(f"  Shipments report OK -> {filename}")
 
 
-def download_shipments():
+if __name__ == "__main__":
+    from playwright.sync_api import sync_playwright
     with sync_playwright() as p:
         browser = p.chromium.launch(headless=False)
         context = browser.new_context(accept_downloads=True)
         page = context.new_page()
-
-        # Prihlaseni
         page.goto(BASE_URL)
         page.wait_for_load_state("networkidle")
         page.get_by_label("Email *").fill(EMAIL)
         page.get_by_label("Password *").fill(PASSWORD)
         page.locator('#login__submit').click()
         page.wait_for_load_state("networkidle")
-
-        # Vyber studie
         page.get_by_label("Study *").click()
         page.get_by_role("option", name=STUDY).click()
         page.get_by_role("button", name="SELECT").click()
         page.wait_for_load_state("networkidle")
-
-        # Naviguj na Shipments Report
-        page.goto(f"{BASE_URL}/report/shipments_report")
-        page.wait_for_load_state("networkidle", timeout=15000)
-
-        # Stahni report
-        filename = os.path.join(OUTPUT_DIR, f"shipments_report_{STUDY}.xlsx")
-        with page.expect_download(timeout=30000) as dl:
-            page.get_by_role("button", name="Download XLS").click()
-
-        dl.value.save_as(filename)
-        print(f"Ulozeno -> {filename}")
-
+        run(page, STUDY)
         browser.close()
-        print("Hotovo!")
-
-
-download_shipments()