添加接口密钥

common.py

@@ -4,11 +4,10 @@
 import functools
 import json
 import requests
+from flask import request
 
 from conserve_methods import conserve_data
-
+from setting import API_SECRET
-token = "32737167-cb63-4ce9-9397-d66169488f51"
-headers = {'Authorization': token}
 
 
 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

setting.py

@@ -15,3 +15,7 @@ DB_TYFX = '天眼风险'
 DB_JYZK = '经营状况'
 DB_ZSCQ = '知识产权'
 DB_GSFZ = '公司发展'
+
+
+# API接口密钥
+API_SECRET = "uzdq51N4!I0%HY4sCaQ!aeCSIDIVIdAM"