added encrypt and decrypt testing

This commit is contained in:
crp3844
2022-09-07 11:04:47 -04:00
parent dab4eaa585
commit 75dfd0c758

View File

@@ -294,6 +294,12 @@ if __name__ == "__main__":
action="store_true",
dest="decrypt",
)
parser.add_option(
"-e",
"--encrypt",
action="store_true",
dest="encrypt",
)
parser.add_option(
"-m",
"--message",
@@ -320,8 +326,21 @@ if __name__ == "__main__":
client.open()
#print(encrypt(client, "test"))
if opts.decrypt:
print(decrypt(client, opts.message))
if opts.encrypt:
while True:
passphrase = getpass("Please enter your pool/dataset unlock passphrase to encrypt: ")
passphrase2 = getpass("Please enter the passphrase again: ")
if passphrase == passphrase2:
break
else:
print("The passphrases do not match, please try again.")
encrypted_passphrase = encrypt(client, passphrase)
print("Your encrypted passphrase: {}".format(encrypted_passphrase))
sys.exit(0)
elif opts.decrypt:
passphrase = input("Please enter the encryted passphrase to decrypt: ")
decrypted_passphrase = decrypt(client, passphrase)
print("Your decrypted passphrase: {}".format(decrypted_passphrase))
sys.exit(0)
@@ -366,6 +385,7 @@ if __name__ == "__main__":
elif user_input.casefold() == "n":
account_details = ask_for_account_details()
config = configparser.ConfigParser()
config.read(secrets_ini_file)
config.add_section(account_details["encrypted_account_email_address"])
for key in account_details.keys():
if not key == "encrypted_account_email_address":
@@ -376,6 +396,7 @@ if __name__ == "__main__":
elif user_input.casefold() == "d":
config = configparser.ConfigParser()
config.read(secrets_ini_file)
account_section_to_delete = select_account(accounts, "delete")
print("Are you sure you wish to delete {} account? ".format(account_section_to_delete))
confirmation = input("y/n> ")