refactoring and implementation

scripts/main.py

@@ -1,11 +1,12 @@
 import argparse
 from classes.Generate import Generate
 from classes.Cleanup import Cleanup
+from classes.Decrypt import Decrypt
 from getpass import getpass
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--mode',type=str, dest='mode',required=True,choices=['cleanup','generate'])
+    parser.add_argument('--mode',type=str, dest='mode',required=True,choices=['cleanup','generate','decrypt'])
     parser.add_argument('--amount',type=int, dest='amount_of_secret_holders',required=False,choices=range(1,9))
     parser.add_argument('--quota', type=int, dest='decryption_quota', choices=range(1,101),required=False)
     parser.add_argument('--master-password',type=str, dest='master_password',required=False)
@@ -18,7 +19,20 @@ if __name__ == '__main__':
             cleanup.deleteAll()
             exit()
         cleanup.cleanupForUser(args.user)
+        exit()
         
+    if args.mode == 'decrypt':
+        decrypt = Decrypt()
+        if args.user is None: 
+            print("Please type in the user number:")
+            decrypt.setUser(int(input()))
+        else:
+            decrypt.setUser(args.user)
+        print("Please enter the master password:")
+        user_password = getpass()
+        exit()
+    
+    
     if args.mode == 'generate':
         if args.master_password is None:
             print("Please enter the master password:")