aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2026-02-23 15:34:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

Adds name parameter to generate_keys so the name can be chosen from terminal

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-01-20 16:21:15 +01:00
parent 4eb55eaacc
commit 396c56e5db
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
apps/generate_key.py
View File

@@ -1,5 +1,6 @@
import os.path import os.path
from sys import exit from getopt import getopt
from sys import argv, exit
from cryptography.hazmat.backends import default_backend from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives import serialization
@@ -9,9 +10,6 @@ from cryptography.hazmat.primitives.asymmetric import ec
# Simple tool to generate an ECDSA private key using the secp256k1 curve and save private and public keys # Simple tool to generate an ECDSA private key using the secp256k1 curve and save private and public keys
# as 'pisa_sk.der' 'and pisa_pk.der', respectively. # as 'pisa_sk.der' 'and pisa_pk.der', respectively.
SK_FILE_NAME = "../pisa_sk.der"
PK_FILE_NAME = "../pisa_pk.der"
def save_sk(sk, filename): def save_sk(sk, filename):
der = sk.private_bytes( der = sk.private_bytes(
@@ -31,6 +29,16 @@ def save_pk(pk, filename):
if __name__ == "__main__": if __name__ == "__main__":
name = "pisa"
opts, _ = getopt(argv[1:], "n:", ["name"])
for opt, arg in opts:
if opt in ["-n", "--name"]:
name = arg
SK_FILE_NAME = "../{}_sk.der".format(name)
PK_FILE_NAME = "../{}_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)) print('A key with name "{}" already exists. Aborting.'.format(SK_FILE_NAME))
exit(1) exit(1)