From a864749812881024ac2c37ab853aa6a489547aa2 Mon Sep 17 00:00:00 2001 From: HeshamTB Date: Sun, 6 Sep 2020 08:51:54 +0300 Subject: [PATCH] key-overwrite: prompt overwrite choice. --- rsa.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/rsa.py b/rsa.py index 320560f..d534067 100755 --- a/rsa.py +++ b/rsa.py @@ -31,9 +31,14 @@ def main(): if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen keyFileName = sys.argv[3] if keyExist(keyFileName): - ##ask - print("key already exists") - sys.exit(0) + choice = input("overwrite key %s (y/n)" % keyFileName) + if choice == "y": + key = generateKeys(keyFileName, int(sys.argv[2])) + elif choice == "n": + sys.exit(0) + else: + print("unrecognized choice!") + sys.exit(1) else: key = generateKeys(keyFileName, int(sys.argv[2])) print("e: ", key[E])