From 825d8eb647acdb0415ad9001f4c4899d085dbe9a Mon Sep 17 00:00:00 2001
From: HeshamTB <hishaminv@gmail.com>
Date: Sun, 6 Sep 2020 08:07:15 +0300
Subject: [PATCH] Block overwrite

---
 rsa.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/rsa.py b/rsa.py
index a46d885..320560f 100755
--- a/rsa.py
+++ b/rsa.py
@@ -30,7 +30,12 @@ def main():
     print("hesham-rsa version ", VERSION)
     if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen <keysize> <keyname>
         keyFileName = sys.argv[3]
-        key = generateKeys(keyFileName, int(sys.argv[2]))
+        if keyExist(keyFileName):
+            ##ask
+            print("key already exists")
+            sys.exit(0)
+        else:
+            key = generateKeys(keyFileName, int(sys.argv[2]))
         print("e: ", key[E])
         print("n: ", key[N])
         print("d: ", key[D])
@@ -351,5 +356,8 @@ def backTrack(x):
         print("\b",end="")
         length -= 1
 
+def keyExist(keyName):
+    exist = os.path.exists(keysFolder+keyName)
+    return exist
 if __name__ == "__main__":
     main()