z230

IWRS/Drugs/run_all.py

@@ -505,14 +505,21 @@ def download_shipment_details(page, study):
     df = pd.read_excel(report_path, header=header_row)
     df = df.dropna(how="all")
     df = df[df["Location"].astype(str).str.contains("Czech", na=False, case=False)]
-    cz_ids = df["Shipment ID"].astype(str).str.strip().tolist()
-    print(f"    CZ zásilek ke stažení: {len(cz_ids)}")
+    cz_shipments = list(zip(
+        df["Shipment ID"].astype(str).str.strip(),
+        df["IRT Shipment Status"].astype(str).str.strip() if "IRT Shipment Status" in df.columns else [""] * len(df),
+    ))
+    print(f"    CZ zásilek ke stažení: {len(cz_shipments)}")
 
     page.goto(f"{BASE_URL}/report/shipment_details_report")
     page.wait_for_load_state("networkidle", timeout=120000)
 
-    for shipment in cz_ids:
+    skipped = 0
+    for shipment, status in cz_shipments:
         filename = os.path.join(out_dir, f"shipment_details_{shipment}.xlsx")
+        if os.path.exists(filename) and status.upper() == "RECEIVED":
+            skipped += 1
+            continue  # finální stav, soubor se nemění
         input_field = page.locator('input[placeholder="search"], input[type="text"]').first
         input_field.click()
         input_field.fill(shipment)
@@ -523,11 +530,13 @@ def download_shipment_details(page, study):
         with page.expect_download(timeout=120000) as dl:
             page.get_by_role("button", name="Download XLS").click()
         dl.value.save_as(filename)
-        print(f"    [{shipment}] OK")
+        print(f"    [{shipment}] ({status}) OK")
 
         page.get_by_role("button", name="Clear").click()
         page.wait_for_load_state("networkidle", timeout=120000)
 
+    print(f"    Přeskočeno (RECEIVED): {skipped}")
+
 
 # ── main ─────────────────────────────────────────────────────────────────────