#coding=utf-8 
''''' 
加密的一方和解密的一方必須提前確定好key值 
''' 
from Crypto.Cipher import AES 
from binascii import b2a_hex, a2b_hex 
 
class MyCrypto(): 
  def __init__(self, key): 
    self.key_len = len(key) 
    if not self.key_len == 16 and not self.key_len == 24 and not self.key_len == 32: 
      raise Exception("length of key is wrong") 
    self.key = key 
    self.mode = AES.MODE_CBC  #這種模式更加安全 
 
  def encrypt(self, text): 
    ''''' 
      被加密的明文長度必須是key長度的整數(shù)倍,如果不夠,則用\0進行填充 
      轉(zhuǎn)成16進制字符串,是因為避免不可見的ascii在顯示的時候搗亂 
    ''' 
    cryptor = AES.new(self.key, self.mode, self.key) 
    count = len(text) 
    add = self.key_len - (count % self.key_len) 
    text = text + ('\0' * add) 
    self.ciphertext = cryptor.encrypt(text) 
    return b2a_hex(self.ciphertext) 
 
 
  def decrypt(self, text): 
    ''''' 
      解密后需注意,加密時有可能填充\0,因此要去掉右側(cè)的\0 
    ''' 
    cryptor = AES.new(self.key, self.mode, self.key) 
    plain_text = cryptor.decrypt(a2b_hex(text)) 
    return plain_text.rstrip('\0') 
 
 
if __name__ == '__main__': 
  mc = MyCrypto("kwsy_zds20160822") 
  e = mc.encrypt("張東升") 
  d = mc.decrypt(e) 
  print e,d