Check if keys dir exists. if not make it before saving to prevent exception.

2020-04-17 16:34:03 +03:00 · 2020-04-17 16:34:03 +03:00 · 654cb52ca4
commit 654cb52ca4
parent 936a6eebb6
1 changed files with 2 additions and 0 deletions
--- a/rsa.py
+++ b/rsa.py
@ -134,6 +134,8 @@ def readKeyFile(keyName):
    

 def saveKeyFile(key, fileName):
+    if not os.path.isdir(keysFolder):
+        os.makedirs(keysFolder)
    with open(keysFolder+fileName, "w") as keyFile:
        keyFile.write("{0}\n{1}\n{2}\n".format(hex(key[0]), hex(key[1]), hex(key[2])))