使用python对密码进行加密

Posted 2021-05-21 _雪辉_

#!/usr/bin/python3
# -*- coding:utf-8 -*-

import base64
from Crypto.Cipher import AES


class use_aes:

    def __init__(self, key):
        if len(key) > 32:
            key = key[:32]
        self.key = self.to_16(key)

    def to_16(self, key):
        """
        转为16倍数的bytes数据
        :param key:
        :return:
        """
        key = bytes(key, encoding="utf8")
        while len(key) % 16 != 0:
            key += b'\\0'
        return key  
#初始化加密器
    def aes(self):
        return AES.new(self.key, AES.MODE_ECB) # 初始化加密器
#加密
    def encrypt(self, text):
        aes = self.aes()
        return str(base64.encodebytes(aes.encrypt(self.to_16(text))),
                   encoding='utf8').replace('\\n', '') 
#解密
    def decodebytes(self, text):
        aes = self.aes()
        return str(aes.decrypt(base64.decodebytes(bytes(
            text, encoding='utf-8'))).rstrip(b'\\0').decode("utf-8"))


if __name__ == '__main__':
    aes_pass = use_aes("xuhui970125")
    encrypt = aes_pass.encrypt('KhDkNtF2y2')
    decode = aes_pass.decodebytes('HDD47BEkDoCfp0auOV3DQQ==')
    print(encrypt)
    print(decode)