fun with crypto lib in python

rsa.py

from Crypto.PublicKey import RSA
from Crypto import Random
import ast


def newkeys(keysize):
    random_generator = Random.new().read
    key = RSA.generate(keysize, random_generator)
    private, public = key, key.publickey()
    return public, private


def encrypt(message, pub_key):
    return pub_key.encrypt(message, 32)


def decrypt(message, private_key):
    return private_key.decrypt(ast.literal_eval(str(message)))