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

Updates generate_key to accept output directory. It stores the keys in the current dir by default.

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-01-23 20:45:45 +01:00
parent ebea93c103
commit a84d753c4a
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
apps/generate_key.py
View File

@@ -30,14 +30,21 @@ def save_pk(pk, filename):
if __name__ == "__main__": if __name__ == "__main__":
name = "pisa" name = "pisa"
output_dir = "."
opts, _ = getopt(argv[1:], "n:", ["name"]) opts, _ = getopt(argv[1:], "n:d:", ["name", "dir"])
for opt, arg in opts: for opt, arg in opts:
if opt in ["-n", "--name"]: if opt in ["-n", "--name"]:
name = arg name = arg
SK_FILE_NAME = "../{}_sk.der".format(name) if opt in ["-d", "--dir"]:
PK_FILE_NAME = "../{}_pk.der".format(name) output_dir = arg
if output_dir.endswith("/"):
output_dir = output_dir[:-1]
SK_FILE_NAME = "{}/{}_sk.der".format(output_dir, name)
PK_FILE_NAME = "{}/{}_pk.der".format(output_dir, 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)) print('A key with name "{}" already exists. Aborting.'.format(SK_FILE_NAME))