Adding check for valid output path

FENGHUNZHILING · Apr 12, 2022 · 22f656b · 22f656b
1 parent 8eeba71
commit 22f656b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/video2x/video2x.py b/video2x/video2x.py
@@ -586,6 +586,11 @@ def main() -> int:
         if not args.input.is_file():
             logger.critical("Input path is not a file")
             return 1
+
+        # Output file won't exist yet, but the parent directory should.
+        if not args.output.parent.exists():
+            logger.critical(f"Output directory does not exist: {args.output}")
+            return 1
 
         # set logger level
         if os.environ.get("LOGURU_LEVEL") is None: