# -*- coding: utf-8 -*-
# @Time    : 2021/9/05 13:35
# @Author  : jiafeng.li
# @File    : Hash.py
'''
各种加密方法
'''
from hashlib import sha1
from hashlib import md5
import binascii
from Crypto.Hash import SHA256
from Crypto.Cipher import DES
from Crypto.Cipher import AES

def my_md5(msg):
    '''
    md5 加密算法
    :param msg:需加密的字符串
    :return:返回的结果
    '''
    hl = md5()
    hl.update(msg.encode('utf-8'))
    return hl.hexdigest()

def my_sha1(msg):
    '''
    sha1 加密算法
    :param msg: 需加密的字符串
    :return: 返回的结果
    '''
    sh = sha1()
    sh.update(msg.encode('utf-8'))
    return sh.hexdigest()

def my_sha256(msg):
    '''
    sha256 加密算法
    :param msg: 需加密的字符串
    :return: 返回的结果
    '''
    sh = SHA256.new()
    sh.update(msg.encode('utf-8'))
    return sh.hexdigest()

def my_des(msg, key):
    """
    DES 算法加密
    :param msg: 需加密的字符串,长度必须为8的倍数，不足添加'='
    :param key: 8个字符
    :return: 加密后的字符
    """
    de = DES.new(key, DES.MODE_ECB)
    mss = msg + (8 - (len(msg) % 8)) * '='
    text = de.encrypt(mss.encode())
    return binascii.b2a_hex(text).decode()


def my_aes_encrypt(msg, key, vi):
    """
    AES 算法的加密
    :param msg: 需加密的字符串
    :param key: 必须为16，24，32位
    :param vi: 必须为16位
    :return: 加密后的字符
    """
    obj = AES.new(key, AES.MODE_CBC, vi)
    txt = obj.encrypt(msg.encode())
    return binascii.b2a_hex(txt).decode()


def my_aes_decrypt(msg, key, vi):
    """
    AES 算法的解密
    :param msg: 需解密的字符串
    :param key: 必须为16，24，32位
    :param vi: 必须为16位
    :return: 加密后的字符
    """
    msg = binascii.a2b_hex(msg)
    obj = AES.new(key, AES.MODE_CBC, vi)
    return obj.decrypt(msg).decode()