Error if import directory doesn't exist

pwncollege · Oct 21, 2024 · 11e4f23 · 11e4f23
1 parent 664525a
commit 11e4f23
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pwnshop/__main__.py b/pwnshop/__main__.py
@@ -171,7 +171,11 @@ def main():
     pwn.context.log_level = "ERROR"
 
     if getattr(args, "import", None):
-        imports = glob.glob(getattr(args, "import"))
+        import_dir = getattr(args, "import")
+        if not os.path.isdir(import_dir):
+            print(f"Error, {import_dir=} does not exist")
+            return -1
+        imports = glob.glob(import_dir)
         for i in imports:
             i = i.rstrip("/")
             sys.path.append(os.path.realpath(os.path.dirname(i)))