nicer error handling for python vault

NitorCreations · Oct 30, 2024 · 5cf90be · 5cf90be
1 parent d918ad3
commit 5cf90be
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/python-pyo3/python/p_vault/vault.py b/python-pyo3/python/p_vault/vault.py
@@ -4,9 +4,15 @@
 
 
 def main():
-    # Override the script name in the arguments list so the Rust CLI works correctly
-    args = ["pvault"] + sys.argv[1:]
-    nitor_vault_rs.run(args)
+    try:
+        # Override the script name in the arguments list so the Rust CLI works correctly
+        args = ["pvault"] + sys.argv[1:]
+        nitor_vault_rs.run(args)
+    except KeyboardInterrupt:
+        print("\naborted")
+        exit(1)
+    except Exception as e:
+        print(f"Error: {e}")
 
 
 if __name__ == "__main__":