key-overwrite: prompt overwrite choice.
This commit is contained in:
parent
825d8eb647
commit
a864749812
11
rsa.py
11
rsa.py
@ -31,9 +31,14 @@ def main():
|
|||||||
if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen <keysize> <keyname>
|
if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen <keysize> <keyname>
|
||||||
keyFileName = sys.argv[3]
|
keyFileName = sys.argv[3]
|
||||||
if keyExist(keyFileName):
|
if keyExist(keyFileName):
|
||||||
##ask
|
choice = input("overwrite key %s (y/n)" % keyFileName)
|
||||||
print("key already exists")
|
if choice == "y":
|
||||||
sys.exit(0)
|
key = generateKeys(keyFileName, int(sys.argv[2]))
|
||||||
|
elif choice == "n":
|
||||||
|
sys.exit(0)
|
||||||
|
else:
|
||||||
|
print("unrecognized choice!")
|
||||||
|
sys.exit(1)
|
||||||
else:
|
else:
|
||||||
key = generateKeys(keyFileName, int(sys.argv[2]))
|
key = generateKeys(keyFileName, int(sys.argv[2]))
|
||||||
print("e: ", key[E])
|
print("e: ", key[E])
|
||||||
|
Loading…
Reference in New Issue
Block a user