# coding=utf-8
import time

import rsa

public_key = """-----BEGIN PUBLIC KEY-----
MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCWJ0LZ/UMwouUq89szyUYk3BEB
kIAjgkjbzuRq9lBae7pCyBxcoVzwLm7Zm8ZorMutwaTJvVVTXFFEt9UphHkXMAOF
nqWcvN7Mv9aneJVRkk9h8ZvfDtnrEj6XAnmYRILnB4T0Bcek4dYvhjym77cDd6xG
6GPZ+/JE+3rLC58xPwIDAQAB
-----END PUBLIC KEY-----
"""


def rsaEncrypt(str):
    # 生成公钥、私钥
    pubkey = rsa.PublicKey.load_pkcs1_openssl_pem(public_key)
    # 明文编码格式
    content = str.encode('utf-8')
    # 公钥加密
    crypto = rsa.encrypt(content, pubkey)
    with open("Tengami", "wb+") as f:
        f.write(crypto)


if __name__ == '__main__':
    mac = None
    try:
        with open("MAC", "r+", encoding="UTF-8") as f:
            mac = f.read()
        if mac:
            rsaEncrypt(mac)
            print "生成授权文件成功！"
            time.sleep(2)
        else:
            print "生成授权文件失败！\n没有找到MAC文件！"
            time.sleep(2)
    except Exception as e:
        print "生成授权文件失败！\n没有找到MAC文件！"
        time.sleep(2)