import base64
from Crypto.Cipher import AES
from binascii import b2a_hex,a2b_hex
 
 
class PrpCrypt(object):
  def __init__(self,key,iv):
    self.key=key.encode('utf-8')
    self.mode=AES.MODE_CBC
    self.iv=iv.encode('utf-8')
 
  def pad_byte(self, b):
    '''
    1 先計算所傳入bytes類型文本與16的余數
    2 在將此余數轉成bytes 當然用0補位也可以
    3 已知了 余數 那么就用余數*被轉成的余數，就得到了需要補全的bytes
    4 拼接原有文本和補位
    :param b: bytes類型的文本
    :return: 返回補全后的bytes文本
    '''
    bytes_num_to_pad = AES.block_size - (len(b) % AES.block_size)
    # python3 中默認unicode轉碼
    # 實際上byte_to_pad 就已經 將 數字轉成了unicode 對應的字符 即使你的入參正好是16的倍數，那么bytes也是把列表整體的轉碼也是有值的
    # 后邊解密的匿名函數 拿到最后一個數字后，就知道應該截取的長度，在反著切片就行了
    # 這樣保證了數據的完整性
    byte_to_pad = bytes([bytes_num_to_pad])
    padding = byte_to_pad * bytes_num_to_pad
    padded = b + padding
    return padded
 
  def encrypt(self,text):
    '''
    1 先生成aes實例
    2 對傳入的text轉成bytes
    3 對傳入的text補全
    4 調用encrypt 加密 得到密文
    5 先將密文轉16進制，在將16進制用base64轉碼，然后在將得到的base64解碼
    其實在步驟4 就已經完成了aes加密，我所在的公司加密比較復雜 ，需要的可以直接返回步驟4的值
 
    :param text:
    :return:
    '''
    cryptor = AES.new(self.key,self.mode,self.iv)
    text = text.encode('utf-8')
    text = self.pad_byte(text)
    self.ciphertext = cryptor.encrypt(text)
    cryptbase64 = base64.b64encode(b2a_hex(self.ciphertext)).decode('utf8')
    return cryptbase64
 
 
  def decrypt(self,text):
    '''
    解密和加密的順序是相反的
    1 定義匿名函數，去掉補位
    2 base64解碼
    3 生成aes實例
    4 16進制轉2進制
    5 使用decrypt解碼 得到補全的bytes類型明文
 
    :param text:
    :return: 解密且去掉補位的明文
    '''
    unpad = lambda s: s[:-ord(s[len(s) - 1:])]
    base64Str = base64.b64decode(text.encode('utf8'))
    cryptor = AES.new(self.key, self.mode, self.iv)
    aesStr = cryptor.decrypt(a2b_hex(base64Str))
    aesStr = str(unpad(aesStr), encoding='utf8')
    return aesStr
if __name__ == '__main__':
  data='sadfsasdgsdfgdsf123'
  pc=PrpCrypt('daajanbaqa654321','1234567890123456')
  redata = pc.encrypt(data)
  print(redata)
  result=pc.decrypt(redata)
  print(result)