Python 加密字串生成相关工具库

#!/usr/bin/python3
# -*- coding:utf-8 -*-
"""
@author: JHC
@file: util_encrtpt.py
@time: 2023/5/28 14:17
@desc:
"""
import base64
import hashlib
import uuid
import random


class EncryptProcess(object):
    """
    加密
    """

    def make_md5(self, file: str = None, data: str = None, salt: bytes = b""):
        """
        生成md5
        :param file:
        :param data:
        :param salt:
        :return:
        """
        if file:
            hash = hashlib.md5(salt)
            with open(file, 'rb') as fp:
                hash.update(fp.read())
            return hash.hexdigest()
        if data:
            md5_machine = hashlib.md5(salt)
            md5_machine.update(data.strip().encode('utf-8'))
            return md5_machine.hexdigest()

    def encode_base64(self, date: bytes, retype=None):
        """
        base64 编码
        :param date:
        :param retype: 默认返回字符串类型，还支持返回bytes
        :return:
        """
        if not retype:
            return base64.b64encode(date).decode("utf-8")
        else:
            return base64.b64encode(date)

    def decode_base64(self, date) -> str:
        """
        base64 解码
        :param date:bytes/str都支持
        :return:
        """
        return base64.urlsafe_b64decode(date).decode("utf-8")

    def decode_byte_str(self, date: bytes) -> str:
        """
        bytes转str
        """
        return date.decode("utf-8")

    def decode_unicode(self, date) -> str:
        """
        unicode转中文，支持bytes和str类型输入
        """
        if isinstance(date, str):
            return date.encode().decode("unicode_escape")
        elif isinstance(date, bytes):
            return date.decode("unicode_escape")
        else:
            raise ValueError('输入类型错误{}{}'.format(date, type(date)))

    def make_uuid(self, status: int = 0):
        """
        生成uuid
        :param status:
        :return:
        """
        if status == 0:
            return str(uuid.uuid4())
        elif status == 1:
            return str(uuid.uuid4()).replace("-", "_")

    def get_random_str(self, secret_len=4):
        """
        获取指定长度的随机字符串列表
        :param secret_len:
        :return:
        """
        result = [str(i) for i in range(secret_len)]
        random_char_list = ["#", "*", "_", "-", ".", "$", "@"]
        for i in range(secret_len):
            random_num = str(random.randint(0, 9))
            random_low_alpha = chr(random.randint(97, 122))
            random_up_alpha = chr(random.randint(97, 122)).upper()
            random_char = random.choice(random_char_list)
            result[i] = random.choice(
                [random_num, random_low_alpha, random_up_alpha, random_char])
        return result