python对称加密实现_Python实现AES加密(对称加密算法)

Python实现AES加密(对称加密算法)

发布时间：2018-11-11 21:20,

浏览次数：1017

, 标签：

Python

AES

from Cryptodome.Cipher import AES from binascii import b2a_hex,a2b_hex from

Cryptodome import Random class AesEncryption(object): def __init__(self, key,

mode=AES.MODE_CFB): self.key = self.check_key(key) #

密钥key长度必须为16,24或者32bytes的长度 self.mode = mode self.iv =

Random.new().read(AES.block_size) def check_key(self, key):

'检测key的长度是否为16,24或者32bytes的长度' try: if isinstance(key, bytes): assert len(key)

in [16, 24, 32] return key elif isinstance(key, str): assert len(key.encode())

in [16, 24, 32] return key.encode() else: raise

Exception(f'密钥必须为str或bytes,不能为{type(key)}') except AssertionError:

print('输入的长度不正确') def check_data(self,data): '检测加密的数据类型' if isinstance(data,

str): data = data.encode() elif isinstance(data, bytes): pass else: raise

Exception(f'加密的数据必须为str或bytes,不能为{type(data)}') return data def encrypt(self,

data): ' 加密函数 ' data = self.check_data(data) cryptor = AES.new(self.key,

self.mode,self.iv) return b2a_hex(cryptor.encrypt(data)).decode() def

decrypt(self,data): ' 解密函数 ' data = self.check_data(data) cryptor =

AES.new(self.key, self.mode,self.iv) return

cryptor.decrypt(a2b_hex(data)).decode() if __name__ == '__main__': key =

input('请输入key:') data = '你真帅' aes = AesEncryption(key) e = aes.encrypt(data) #

调用加密函数 d = aes.decrypt(e) # 调用解密函数 print(e) print(d)