chore: Refactor main function to use directory path instead of zip path

hy20191108 · Jun 28, 2024 · 8c6da9a · 8c6da9a
1 parent 4a1624e
commit 8c6da9a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/idir2pdf/idir2pdf.py b/src/idir2pdf/idir2pdf.py
@@ -183,13 +183,13 @@ def convert(zip_any, pdfpath=None, progress=True) -> bytes:
 
 
 def main() -> None:
-    for zippath in tqdm.tqdm(sys.argv[1:], desc="process"):
-        tqdm.tqdm.write(zippath)
-        pdfpath = os.path.splitext(zippath)[0] + ".pdf"
+    for dirpath in tqdm.tqdm(sys.argv[1:], desc="process"):
+        tqdm.tqdm.write(dirpath)
+        pdfpath = dirpath + ".pdf"
         if os.path.isfile(pdfpath):
             continue
         try:
-            convert(zippath, pdfpath, progress=True)
+            convert(dirpath, pdfpath, progress=True)
         except KeyboardInterrupt:
             exit()