update testforadmin包

This commit is contained in:
P3ngSaM 2022-04-25 13:46:19 +08:00
parent 86d77ec036
commit 8a7fbf1b63
4 changed files with 46 additions and 0 deletions

21
TestForAdmin/APIAuth.py Normal file
View File

@ -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

0
TestForAdmin/__init__.py Normal file
View File

23
TestForAdmin/route.py Normal file
View File

@ -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

2
app.py
View File

@ -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')