# -*- coding: utf-8 -*-
from Crypto.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
from Crypto.PublicKey import RSA
import base64
import json


class rsa_encrypt():
    def rsa(self, data, key):
        max_len = 117
        # data 数据转换格式
        data = json.dumps(data)

        # 导入读取到的公钥(要先转一下格式)
        rsakey = RSA.importKey(base64.b64decode(key))

        # 生成加密对象
        cipher = Cipher_pkcs1_v1_5.new(rsakey)

        # 通过生成的对象加密message明文，注意，在python3中加密的数据必须是bytes类型的数据，不能是str类型的数据
        res = []
        data_utf = bytes(data, 'utf-8')

        # 分段加密
        for i in range(0, len(data_utf), max_len):
            res.append(cipher.encrypt(data_utf[i:i + max_len]))

        # 返回拼接好的base64编码加密段
        return bytes.decode(base64.b64encode(b''.join(res)))