sio-2425/delivery1/client/bin/rep_subject_credentials

#!/bin/python3
import os
import sys
import logging
import argparse

from lib import key_pair

logging.basicConfig(format='%(levelname)s\t- %(message)s')
logger = logging.getLogger()
logger.setLevel(logging.INFO)

# Generate a key pair for a subject
# password - file for public key, file for private key
def generateKeyPair(args):

    parser = argparse.ArgumentParser()

    parser.add_argument('password', nargs='?', default=None)
    parser.add_argument('pubfile', nargs='?', default=None)
    parser.add_argument('privfile', nargs='?', default=None)

    args = parser.parse_args()

    if len(args) != 3:
        logger.error("Need password and file to store keys")
        sys.exit(-1)


    #Generate the key pair
    key_pair.generate_key_pair(args.pubfile, args.privfile, 2048, args.password)

    #Save key files in environment
    os.environ['PRIV_KEY'] = args.privfile
    os.environ['PUB_KEY'] = args.pubfile

    return 0

if __name__ == '__main__':
    generateKeyPair(sys.argv[1:])