z230

RozkladScanu/rozloz_brozuru.py

@@ -12,6 +12,7 @@ Pouziti:
 
 import sys
 import copy
+from pathlib import Path
 from pypdf import PdfReader, PdfWriter
 from pypdf.generic import RectangleObject
 
@@ -67,10 +68,29 @@ def rozloz(input_path, output_path):
 
 
 if __name__ == "__main__":
-    if len(sys.argv) < 2:
-        print("Pouziti: python rozloz_brozuru.py <vstup.pdf> [vystup.pdf]")
-        sys.exit(1)
+    BASE = Path(__file__).parent
+    VSTUP  = BASE / "vstup"
+    VYSTUP = BASE / "vystup"
+    VYSTUP.mkdir(exist_ok=True)
 
-    inp = sys.argv[1]
-    out = sys.argv[2] if len(sys.argv) > 2 else inp.replace(".pdf", "_serazena.pdf")
-    rozloz(inp, out)
+    if len(sys.argv) >= 2:
+        # explicitni argumenty zachovany pro zpetnou kompatibilitu
+        inp = sys.argv[1]
+        out = sys.argv[2] if len(sys.argv) > 2 else inp.replace(".pdf", "_serazena.pdf")
+        rozloz(inp, out)
+    else:
+        pdfs = [f for f in VSTUP.glob("*.pdf") if "[ZPRACOVANO]" not in f.name]
+        if not pdfs:
+            print("Zadne PDF soubory ke zpracovani ve vstup/")
+            sys.exit(0)
+
+        for pdf in pdfs:
+            out = VYSTUP / (pdf.stem + "_serazena.pdf")
+            print(f"\n=== {pdf.name} ===")
+            try:
+                rozloz(str(pdf), str(out))
+                done = pdf.with_name(pdf.stem + " [ZPRACOVANO]" + pdf.suffix)
+                pdf.rename(done)
+                print(f"Vstupni soubor oznacen: {done.name}")
+            except Exception as e:
+                print(f"CHYBA: {e}")