116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
import hmac
|
||
import json
|
||
from datetime import datetime, date
|
||
from hashlib import md5, sha256
|
||
|
||
|
||
def get_sign(param_dict: dict, request_dict: dict, app_secret: str, sign_method: str) -> str:
|
||
if sign_method == "hmac-sha256":
|
||
return get_sign_with_hmac_sha256(param_dict, request_dict, app_secret)
|
||
elif sign_method == "md5":
|
||
return get_sign_with_md5(param_dict, request_dict, app_secret)
|
||
else:
|
||
raise Exception("Unsupported sign method: " + sign_method)
|
||
|
||
|
||
def get_sign_with_md5(param_dict: dict, request_dict: dict, app_secret: str) -> str:
|
||
"""
|
||
使用md5计算签名
|
||
:param param_dict: 公共参数字典
|
||
:param request_dict: api参数字典
|
||
:param app_secret: 密钥
|
||
:return:
|
||
"""
|
||
param_dict.update(request_dict)
|
||
keys = list(param_dict.keys())
|
||
keys.sort()
|
||
parameters = "%s%s%s" % (app_secret,
|
||
str().join(
|
||
'%s%s' % (k, param_dict[k]) for k in keys if not isinstance(param_dict[k], bytes)),
|
||
app_secret)
|
||
sign = md5(parameters.encode("utf-8")).hexdigest().upper()
|
||
return sign
|
||
|
||
|
||
def get_sign_with_hmac_sha256(param_dict: dict, request_dict: dict, app_secret: str):
|
||
"""
|
||
使用hmac sha256计算签名
|
||
:param param_dict: 公共参数字典
|
||
:param request_dict: api参数字典
|
||
:param app_secret: 密钥
|
||
:return:
|
||
"""
|
||
param_dict.update(request_dict)
|
||
keys = list(param_dict.keys())
|
||
keys.sort()
|
||
parameters = str().join('%s%s' % (k, param_dict[k]) for k in keys if not isinstance(param_dict[k], bytes))
|
||
sign = hmac.new(app_secret.encode("utf-8"), parameters.encode("utf-8"), digestmod=sha256).hexdigest().upper()
|
||
return str(sign)
|
||
|
||
|
||
def json_default(value):
|
||
if isinstance(value, (datetime, date)):
|
||
return value.strftime('%Y-%m-%d %H:%M:%S')
|
||
else:
|
||
return value.__dict__
|
||
|
||
|
||
class TopJsonEncoder(json.JSONEncoder):
|
||
def default(self, obj):
|
||
if isinstance(obj, (datetime, date)):
|
||
return obj.strftime('%Y-%m-%d %H:%M:%S')
|
||
else:
|
||
return json.JSONEncoder.default(self, obj)
|
||
|
||
|
||
def convert_basic(param):
|
||
"""
|
||
转换基本类型
|
||
:param param:
|
||
:return:
|
||
"""
|
||
if isinstance(param, (datetime, date)):
|
||
return param.strftime("%Y-%m-%d %H:%M:%S")
|
||
elif isinstance(param, bool):
|
||
return str(param).lower()
|
||
elif isinstance(param, bytes):
|
||
return param
|
||
else:
|
||
return str(param)
|
||
|
||
|
||
def convert_basic_list(param):
|
||
"""
|
||
转换字符串、数字类型列表，列表 -> 逗号分割字符串
|
||
:param param:
|
||
:return:
|
||
"""
|
||
if isinstance(param, (list, tuple, set)):
|
||
return ",".join(convert_basic(i) for i in param)
|
||
else:
|
||
return param
|
||
|
||
|
||
def convert_struct(param):
|
||
"""
|
||
转换dict 对象类型参数，转json字符串
|
||
:param param:
|
||
:return:
|
||
"""
|
||
if isinstance(param, str):
|
||
return param
|
||
else:
|
||
return json.dumps(param, cls=TopJsonEncoder, default=json_default, ensure_ascii=False)
|
||
|
||
|
||
def convert_struct_list(param):
|
||
"""
|
||
转换复杂类型列表
|
||
:param param:
|
||
:return:
|
||
"""
|
||
if isinstance(param, (set, list, tuple)):
|
||
return json.dumps(param, cls=TopJsonEncoder, default=json_default, ensure_ascii=False)
|
||
else:
|
||
return str(param)
|