adding file not found check

vinissou · Oct 3, 2024 · 206cbaf · 206cbaf
1 parent ced53b0
commit 206cbaf
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pdf_page_size_fixer.py b/pdf_page_size_fixer.py
@@ -26,9 +26,12 @@ def main():
     if args.options:
         messages.options()
 
-    src = fitz.Document(args.file)
-    doc = fitz.Document()
-    out = src.name
+    if args.file:
+        src = fitz.Document(args.file)
+        doc = fitz.Document()
+        out = src.name
+    else:
+        raise FileNotFoundError("\n\n \033[31m Input file not provided \033[0m \n")
 
     print("\n FILE:", out)