from pyDes import *

import base64

class Des3(object):

    def __init__(self, key, iv):

        # 这里密钥key长度必须为16/24, ,偏移量ivs

        self.key = key

        self.mode = CBC

        self.iv = iv

    # 加密函数，如果text不是16的倍数【加密文本text必须为16的倍数！】，那就补足为16的倍数

    def encrypt(self, text):

        des3 = triple_des(self.key, self.mode, self.iv, pad=None, padmode=PAD_PKCS5)

        data = des3.encrypt(text)

        data = base64.b64encode(data)

        return data.decode('utf-8')

    # 解密后，去掉补足的空格用strip() 去掉

    def decrypt(self, data):

        des3 = triple_des(self.key, self.mode, self.iv, pad=None, padmode=PAD_PKCS5)

        data = base64.b64decode(data)

        text = des3.decrypt(data)

        return text.decode('hex')