From 8a7fbf1b63b2c4ca7aeab6a214be9c2940b22d04 Mon Sep 17 00:00:00 2001
From: P3ngSaM <61768364+P3ngSaM@users.noreply.github.com>
Date: Mon, 25 Apr 2022 13:46:19 +0800
Subject: [PATCH] =?UTF-8?q?update=20testforadmin=E5=8C=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 TestForAdmin/APIAuth.py  | 21 +++++++++++++++++++++
 TestForAdmin/__init__.py |  0
 TestForAdmin/route.py    | 23 +++++++++++++++++++++++
 app.py                   |  2 ++
 4 files changed, 46 insertions(+)
 create mode 100644 TestForAdmin/APIAuth.py
 create mode 100644 TestForAdmin/__init__.py
 create mode 100644 TestForAdmin/route.py

diff --git a/TestForAdmin/APIAuth.py b/TestForAdmin/APIAuth.py
new file mode 100644
index 0000000..2568c19
--- /dev/null
+++ b/TestForAdmin/APIAuth.py
@@ -0,0 +1,21 @@
+import functools
+
+from flask import request
+
+
+def api_secret(func):
+    """
+    校验接口请求密钥
+    """
+    secret = "EZgo9ykxrYuBMYnYmmKIh"  # 接口密钥
+
+    @functools.wraps(func)
+    def internal(*args, **kwargs):
+        try:
+            token = request.headers.get('secret')
+            if token != secret:
+                return {"info": "接口密钥错误"}, 401
+        except Exception:
+            return {"info": "请求异常"}, 401
+        return func(*args, **kwargs)
+    return internal
diff --git a/TestForAdmin/__init__.py b/TestForAdmin/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/TestForAdmin/route.py b/TestForAdmin/route.py
new file mode 100644
index 0000000..c23f7e3
--- /dev/null
+++ b/TestForAdmin/route.py
@@ -0,0 +1,23 @@
+from flask import Blueprint, request, Response
+
+from CompanyUser.CompanyUserImpl import CompanyVerifyThreeFactorsImpl
+from TestForAdmin.APIAuth import api_secret
+from Utils.ErrorUtil import ReturnConditionCheckFailed
+
+test_admin_route = Blueprint('test_admin', __name__)
+
+
+@test_admin_route.route('/company_verify', methods=['GET'])
+@api_secret
+def company_verify_route():
+    """企业认证"""
+    try:
+        company_verify = CompanyVerifyThreeFactorsImpl()
+        company_verify.cid = request.args['cid']
+        company_verify.company_verify()
+        return {"info": "认证成功"}, 200
+    except ReturnConditionCheckFailed as e:
+        e.log_error()
+        return {"info": e.failed_info}, e.status_code
+    except KeyError:
+        return {"info": "参数异常"}, 400
diff --git a/app.py b/app.py
index dcfc8de..876b436 100644
--- a/app.py
+++ b/app.py
@@ -7,6 +7,7 @@ from CompanyUser.CompanyUserRoute import company_user_route
 from file.file_routes import file_route
 from Rating.Credit.CreditRoutes import credit_route
 from Rating.Esg.EsgRoutes import esg_route
+from TestForAdmin.route import test_admin_route
 
 app = Flask(__name__)
 CORS(app, supports_credentials=True)
@@ -18,6 +19,7 @@ app.register_blueprint(company_user_route, url_prefix='/app/user')
 app.register_blueprint(credit_route, url_prefix='/app/input')
 app.register_blueprint(company_route, url_prefix='/app/company')
 app.register_blueprint(esg_route, url_prefix='/app/esg')
+app.register_blueprint(test_admin_route, url_prefix='/app/test')
 
 
 @app.route('/app/version')