rsa_key_generator.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os                  # for handling paths
import sys                 # for getting sys.argv
from Cryptodome.PublicKey import RSA
import getpass             # for securely entering a user's pass phrase to access the private key

# - GLOBAL SCOPE VARIABLES start -
full_path = os.path.dirname(os.path.realpath(sys.argv[0]))
# - GLOBAL SCOPE VARIABLES end -

# GENERATE NEW RSA PUBLIC-PRIVATE KEY PAIR:
passphrase_1 = getpass.getpass(prompt='Enter your private key passphrase (your input will be hidden):\n')
passphrase_2 = getpass.getpass(prompt='Confirm your private key passphrase (your input will be hidden):\n')
if passphrase_1 != passphrase_2:
    raise ValueError("Your inputs don't match!")
print('Inputs match, generating your keys...')
new_key = RSA.generate(4096)
export_format = 'PEM'
private_key = new_key.exportKey(pkcs=8, passphrase=passphrase_1)
public_key = new_key.publickey().exportKey(format=export_format)
with open(os.path.join(full_path, "private_key." + export_format.lower()), "wb") as f:
    f.write(private_key)
with open(os.path.join(full_path, "public_key." + export_format.lower()), "wb") as f:
    f.write(public_key)
print('Finished!')