添加接口密钥
This commit is contained in:
parent
9e0353f586
commit
1c57799bba
22
common.py
22
common.py
|
@ -4,11 +4,10 @@
|
|||
import functools
|
||||
import json
|
||||
import requests
|
||||
from flask import request
|
||||
|
||||
from conserve_methods import conserve_data
|
||||
|
||||
token = "32737167-cb63-4ce9-9397-d66169488f51"
|
||||
headers = {'Authorization': token}
|
||||
from setting import API_SECRET
|
||||
|
||||
|
||||
def check_status_code(func):
|
||||
|
@ -74,6 +73,7 @@ def general_request(url, company, db, col):
|
|||
Returns:
|
||||
'保存成功'
|
||||
"""
|
||||
headers = {'Authorization': "32737167-cb63-4ce9-9397-d66169488f51"}
|
||||
# 根据传入url进行请求
|
||||
res = json.loads(requests.get(url, headers=headers).text)
|
||||
# 判断请求返回结果中的error_code
|
||||
|
@ -92,3 +92,19 @@ def general_request(url, company, db, col):
|
|||
return '保存成功'
|
||||
else:
|
||||
return status
|
||||
|
||||
|
||||
def verify_token(func):
|
||||
"""
|
||||
校验token
|
||||
"""
|
||||
@functools.wraps(func)
|
||||
def internal(*args, **kwargs):
|
||||
try:
|
||||
token = request.headers.get('token')
|
||||
if token != API_SECRET:
|
||||
return {"info": "接口密钥错误"}, 401
|
||||
except Exception:
|
||||
return {"info": "请求异常"}, 401
|
||||
return func(*args, **kwargs)
|
||||
return internal
|
||||
|
|
|
@ -15,3 +15,7 @@ DB_TYFX = '天眼风险'
|
|||
DB_JYZK = '经营状况'
|
||||
DB_ZSCQ = '知识产权'
|
||||
DB_GSFZ = '公司发展'
|
||||
|
||||
|
||||
# API接口密钥
|
||||
API_SECRET = "uzdq51N4!I0%HY4sCaQ!aeCSIDIVIdAM"
|
||||
|
|
Loading…
Reference in New Issue