72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
#!/usr/bin/env python
|
||
# encoding: utf-8
|
||
"""
|
||
@author: kira
|
||
@contact: 262667641@qq.com
|
||
@file: rules.py
|
||
@time: 2023/3/14 16:21
|
||
@desc:自定义加密规则
|
||
"""
|
||
import json
|
||
import time
|
||
|
||
from natsort import natsorted
|
||
|
||
__all__ = ["md5_sign", "sha1_sign", "rsa_sign"]
|
||
|
||
from common.crypto.encryption_str import sha1_secret_str, md5
|
||
from common.crypto.encryption_rsa import Rsa
|
||
|
||
|
||
def md5_sign(post_data: dict):
|
||
"""
|
||
MD5 数据加签
|
||
Args:
|
||
**data:需要加钱的数据DD
|
||
|
||
Returns:
|
||
|
||
"""
|
||
print(f"开始执行md5_sign加密,待加密参数={post_data}")
|
||
sorted_list = []
|
||
for key, value in post_data.items():
|
||
try:
|
||
sorted_params = str(key) + str(value)
|
||
sorted_list.append(sorted_params)
|
||
except Exception:
|
||
raise
|
||
sort = natsorted(sorted_list) # 列表自然排序
|
||
argument = "加签所需要的密钥"
|
||
keystore = argument + ("".join(sort)) # 生成加带密钥的新字符串
|
||
sign_value = md5(keystore)
|
||
return {**post_data, **{"sign": sign_value}}
|
||
|
||
|
||
def sha1_sign(post_data: dict):
|
||
"""sha1 加密"""
|
||
print(f"开始执行sha1_sign加密,待加密参数={post_data}")
|
||
timestamp = int(round(time.time() * 1000)) # 毫秒级时间戳
|
||
argument = {"secretKey": "xxxx", "timestamp": timestamp} # 加密加盐参数
|
||
res = {**post_data, **argument}
|
||
sorted_list = []
|
||
for key, value in res.items():
|
||
try:
|
||
value = json.dumps(value, ensure_ascii=False) if isinstance(value, list) or isinstance(value,
|
||
dict) else str(
|
||
value)
|
||
sorted_params = str(key) + "=" + value
|
||
sorted_list.append(sorted_params)
|
||
except Exception as e:
|
||
raise e
|
||
sort = natsorted(sorted_list) # 列表自然排序 返回[(key,value),(key,value)]
|
||
splicing_str = "&".join(sort) # 排序结果 key=value&key=value 拼接
|
||
encrypted_str = sha1_secret_str(splicing_str) # sha1加密
|
||
sign = {"signature": encrypted_str}
|
||
return {**res, **sign} # request 中使用json入参，则传dict
|
||
|
||
|
||
def rsa_sign(post_data):
|
||
"""rsa 加密"""
|
||
rs = Rsa(post_data)
|
||
return rs.rsa_encrypt()
|