aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 06:04:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Creates data_dir when generating keys if it does not exist

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-03-23 11:16:49 +01:00
parent 01ad0d4cdf
commit 09f32b09aa
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
generate_keys.py
View File

@@ -1,4 +1,5 @@
import os.path import os.path
from pathlib import Path
from getopt import getopt from getopt import getopt
from sys import argv, exit from sys import argv, exit
@@ -48,15 +49,14 @@ if __name__ == "__main__":
if opt in ["-d", "--dir"]: if opt in ["-d", "--dir"]:
output_dir = arg output_dir = arg
if output_dir.endswith("/"): # Create the output folder it it does not exist (and all the parents if they don't either)
output_dir = output_dir[:-1] Path(output_dir).mkdir(parents=True, exist_ok=True)
SK_FILE_NAME = "{}/{}_sk.der".format(output_dir, name) SK_FILE_NAME = os.path.join(output_dir, "{}_sk.der".format(name))
PK_FILE_NAME = "{}/{}_pk.der".format(output_dir, name) PK_FILE_NAME = os.path.join(output_dir, "{}_pk.der".format(name))
if os.path.exists(SK_FILE_NAME): if os.path.exists(SK_FILE_NAME):
print('A key with name "{}" already exists. Aborting.'.format(SK_FILE_NAME)) exit('A key with name "{}" already exists. Aborting.'.format(SK_FILE_NAME))
exit(1)
sk = ec.generate_private_key(ec.SECP256K1, default_backend()) sk = ec.generate_private_key(ec.SECP256K1, default_backend())
pk = sk.public_key() pk = sk.public_key()