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()