import ast
from Crypto.Cipher import PKCS1_OAEP
from Crypto.PublicKey import RSA
with open('cert/referral', 'rb') as f: key_text = f.read()
privkey = RSA.importKey(key_text)
publickey = privkey.publickey()
encryptor = PKCS1_OAEP.new(publickey)
decryptor = PKCS1_OAEP.new(privkey)

def encrypt(msg):
    bit_msg = bytes(msg, encoding= 'utf-8')
    encrypted = encryptor.encrypt(bit_msg)
    return encrypted

def decrypt(encrypted):
    decrypted = decryptor.decrypt(ast.literal_eval(str(encrypted)))
    return decrypted.decode('ascii')


if __name__ == "__main__":
    msg = 'hi'
    en = encrypt(msg)
    de = decrypt(en)
    assert msg == de