working build, smoothed out edges.

main.py

@@ -2,6 +2,7 @@ import sys
 import getpass
 import secrets
 import os
+import os.path
 import random
 from cryptography.fernet import Fernet
 from base64 import urlsafe_b64encode as b64e, urlsafe_b64decode as b64d
@@ -138,6 +139,8 @@ class Password:
 
 class State:
     def __init__(self, filepath, password):
+        if not os.path.isfile(filepath):
+            raise FileNotFoundError
         self.filepath = filepath
         try:
             file = open(filepath, 'r')
@@ -156,7 +159,6 @@ class State:
 
         tail = lines[1]
         s = tail.split(":")
-        print(s)
         self.rsa_key = RsaKey(eval(s[0]),eval(s[1]))
 
         self.passwords = {}
@@ -293,8 +295,11 @@ def main():
         filepath = sys.argv[1]
 
     if len(sys.argv) < 3:
+        if os.path.isfile(filepath):
             print(f"No password supplied. Usage: {sys.argv[0]} <database path> <password>")
             return
+        else:
+            password = "???"
     else:
         password = sys.argv[2]
 
@@ -331,11 +336,10 @@ def main():
             keysha = password_encrypt("super secret hihi >_>".encode(), password)
             file.write(keysha)
             file.write('\n'.encode())
-            print(repr(key))
             file.write(f"{key}".encode())
             file.write('\n'.encode())
             file.close()
-            return main()
+            return 
         else:
             return
     except Exception as e: