002-python 加密解密

python 加密解密

1、base64 加解密

import base64

def encrypt(text):
    return base64.standard_b64encode(text.encode('utf-8'))

def decrypt(text):
    return str(base64.standard_b64decode(text),'utf-8')

if __name__ == '__main__':
    enstr = encrypt('ab123456cd')
    destr = decrypt(enstr)
    print(f'加密：{enstr}')
    print(f'解密：{destr}')

2、AES 加解密

安装包：pycryptodome
linux下安装：pycryptodome
windows下安装：pycryptodomex

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

def make_random_str(length=16):
    """
    生成一个指定长度的随机字符串，字符串由数字、大小写字母组成
    """
    return ''.join([random.choice(string.digits + string.ascii_letters) for i in range(length)])

class Prpcrypt():
    def __init__(self, key = 'default_key',length = 16):
        self.key = key
        self.length = length
        self.mode = AES.MODE_ECB
        self.cryptor = AES.new(self.pad_key(self.key).encode(), self.mode)

    # 加密函数，如果 pwd 不是 16 的倍数【加密文本 pwd 必须为16的倍数！】，那就补足为 16 的倍数
    def pad(self,pwd):
        # 加密内容需要长达 16 位字符，所以进行空格拼接
        while len(pwd) % self.length != 0:
            pwd += ' '
        return pwd

    # 这里密钥key 长度必须为16（AES-128）、24（AES-192）、或32（AES-256）Bytes 长度.目前AES-128足够用
    def pad_key(self,key):
        # 加密密钥需要长达 16 位字符，所以进行空格拼接
        while len(key) % self.length != 0:
            key += ' '
        return key

    # 加密函数
    def encrypt(self, pwd):
        # 加密的字符需要转换为bytes
        self.ciphertext = self.cryptor.encrypt(self.pad(pwd).encode())
        # 因为 AES 加密时候得到的字符串不一定是 ascii 字符集的，输出到终端或者保存时候可能存在问题
        # 所以这里统一把加密后的字符串转化为 16 进制字符串
        return b2a_hex(self.ciphertext)

    # 解密函数
    def decrypt(self, pwd):
        # 解密后，去掉补足的空格用strip() 去掉
        plain_text = self.cryptor.decrypt(a2b_hex(pwd)).decode()
        return plain_text.rstrip(' ')


if __name__ == '__main__':
    length = 16
    key = make_random_str(length=16)
    pc = Prpcrypt(key=key,length=length)
    e = pc.encrypt("mypwd")
    d = pc.decrypt(e)
    print(key)
    print(len(key))
    print(f'加密后：{e}')
    print(f'解密后：{d}')

    # pc = Prpcrypt(key='9i2RPKCJ2pfSOZU12',length=16)
    # d = pc.decrypt(b'0817a18e4e4913e99ed91114a7f325fd')
    # print(d)