z230

IWRS/download_ip_destruction.py

@@ -13,78 +13,64 @@ STUDY = "77242113UCO3001"
 OUTPUT_DIR = f"xls_ip_destruction_{STUDY}"
 # ────────────────────────────────────────────────────────────────────────────
 
-os.makedirs(OUTPUT_DIR, exist_ok=True)
+def run(page, study):
+    output_dir = f"xls_ip_destruction_{study}"
+    os.makedirs(output_dir, exist_ok=True)
+
+    page.goto(f"{BASE_URL}/report/ip_destruction_form")
+    page.wait_for_load_state("networkidle", timeout=120000)
+
+    page.locator('input[placeholder="search"], input[type="text"]').first.click()
+    page.wait_for_timeout(1000)
+    baskets = [b.strip() for b in page.locator('mat-option').all_inner_texts()
+               if b.strip() and b.strip() != "No results found"]
+    print(f"  Nalezeno {len(baskets)} kosiku: {baskets}")
+    page.keyboard.press("Escape")
+    page.wait_for_timeout(500)
+
+    if not baskets:
+        print("  Zadne destruction kosite — preskakuji.")
+        return
+
+    for basket in baskets:
+        filename = os.path.join(output_dir, f"ip_destruction_basket_{basket}.xlsx")
+        if os.path.exists(filename):
+            print(f"  [{basket}] Preskakuji — existuje.")
+            continue
+        print(f"  [{basket}] Stahuji...")
+        input_field = page.locator('input[placeholder="search"], input[type="text"]').first
+        input_field.click()
+        input_field.fill(basket)
+        page.wait_for_timeout(500)
+        page.locator('mat-option').first.dispatch_event('click')
+        page.wait_for_load_state("networkidle", timeout=120000)
+
+        with page.expect_download(timeout=120000) as dl:
+            page.get_by_role("button", name="Download XLS").click()
+        dl.value.save_as(filename)
+        print(f"  [{basket}] OK")
+
+        page.get_by_role("button", name="Clear").click()
+        page.wait_for_load_state("networkidle", timeout=120000)
+
+    print("  Destruction hotovo.")
 
 
-def download_ip_destruction():
+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()
-
-        # Přihlášení
         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")
-
-        # Výběr 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 report
-        page.goto(f"{BASE_URL}/report/ip_destruction_form")
-        page.wait_for_load_state("networkidle", timeout=15000)
-
-        # Přečti dostupné košíky
-        page.locator('input[placeholder="search"], input[type="text"]').first.click()
-        page.wait_for_timeout(1000)
-        baskets = [b.strip() for b in page.locator('mat-option').all_inner_texts()
-                   if b.strip() != "No results found"]
-        print(f"Nalezeno {len(baskets)} košíků: {baskets}")
-        page.keyboard.press("Escape")
-        page.wait_for_timeout(500)
-
-        if not baskets:
-            print("Žádné destruction košíky nenalezeny — přeskakuji.")
-            browser.close()
-            return
-
-        for basket in baskets:
-            filename = os.path.join(OUTPUT_DIR, f"ip_destruction_basket_{basket}.xlsx")
-            if os.path.exists(filename):
-                print(f"[{basket}] Přeskakuji — soubor již existuje.")
-                continue
-            print(f"[{basket}] Stahuji...")
-
-            # Otevři dropdown a vyber košík přes dispatch_event
-            input_field = page.locator('input[placeholder="search"], input[type="text"]').first
-            input_field.click()
-            input_field.fill(basket)
-            page.wait_for_timeout(500)
-            page.locator('mat-option').first.dispatch_event('click')
-
-            # Počkej na načtení dat
-            page.wait_for_load_state("networkidle", timeout=30000)
-
-            # Stáhni XLS
-            with page.expect_download(timeout=30000) as dl:
-                page.get_by_role("button", name="Download XLS").click()
-
-            download = dl.value
-            download.save_as(filename)
-            print(f"[{basket}] Uloženo → {filename}")
-
-            # Reset pro další košík
-            page.get_by_role("button", name="Clear").click()
-            page.wait_for_load_state("networkidle", timeout=15000)
-
+        run(page, STUDY)
         browser.close()
-        print("\nHotovo!")
-
-
-download_ip_destruction()