implemented json file safe and dynamic master password

This commit is contained in:
2022-12-09 14:52:57 +01:00
parent ee4177a01a
commit f5e290df65
4 changed files with 39 additions and 9 deletions

View File

@@ -1,12 +1,14 @@
import argparse
from classes.Generate import Generate
from classes.Cleanup import Cleanup
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('--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)
args = parser.parse_args()
if args.mode == 'cleanup':
@@ -15,6 +17,11 @@ if __name__ == '__main__':
exit()
if args.mode == 'generate':
generate = Generate(args.amount_of_secret_holders, args.decryption_quota)
generate.execute()
if args.master_password is None:
print("Please enter the master password:")
master_password = getpass()
else:
master_password = args.master_password
generate = Generate(args.amount_of_secret_holders, args.decryption_quota,master_password)
generate.generate()
exit()