refactor: specify arg type

elixir-cloud-aai · Jun 16, 2024 · e9b44e0 · e9b44e0
1 parent 2831ad0
commit e9b44e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/middleware/decrypt.py b/middleware/decrypt.py
@@ -75,7 +75,7 @@ def move_files(file_paths: list[Path], output_path: Path) -> None:
 
 if __name__ == "__main__":
     parser = ArgumentParser()
-    parser.add_argument("file_paths", nargs='+')
+    parser.add_argument("file_paths", nargs='+', type=Path)
     parser.add_argument(
         "--output-dir",
         required=True,
@@ -84,7 +84,7 @@ def move_files(file_paths: list[Path], output_path: Path) -> None:
 
     args = parser.parse_args()
     output_dir = Path(args.output_dir)
-    paths = [Path(path) for path in args.file_paths]
+    paths = args.file_paths
     keys = get_private_keys(paths)
     decrypt_files(paths, keys, output_dir)
     move_files(paths, output_dir)