import os from Crypto.PublicKey import RSA from consts import PRIVATE_KEY_NAME, PUBLIC_KEY_NAME def create_keys(private_key_path, public_key_path): keys = RSA.generate(bits=4096) with open(private_key_path, "wb") as private_fp: private_fp.write(keys.export_key('PEM')) with open(public_key_path, "wb") as public_fp: public_fp.write(keys.publickey().export_key('PEM')) def main(): output_dir = os.path.dirname(os.path.abspath(__file__)) private_key_path = os.path.join(output_dir, PRIVATE_KEY_NAME) public_key_path = os.path.join(output_dir, PUBLIC_KEY_NAME) create_keys(private_key_path, public_key_path) if __name__ == '__main__': main()