164 lines
5.4 KiB
Python
164 lines
5.4 KiB
Python
from flask import Blueprint, request
|
|
|
|
from Test.TestImpl import CompanyAuthImpl, CompanyIndexImpl, DeleteCompanyDataImpl, GenerateFillInData
|
|
from Utils.ErrorUtil import ReturnConditionCheckFailed, APIReturnError
|
|
from Utils.RouteUtil import RouteUtil
|
|
from user.user_auth import verify_token
|
|
|
|
test_route = Blueprint('test', __name__)
|
|
|
|
|
|
@test_route.route('/enterprise_certification', methods=['POST'])
|
|
@verify_token
|
|
def enterprise_certification(**kwargs):
|
|
"""测试用企业认证"""
|
|
try:
|
|
json_ = request.json
|
|
RouteUtil.require_params_check(json_, ['cid', "company", "code", "legal_person"])
|
|
|
|
company_auth = CompanyAuthImpl()
|
|
company_auth.cid = json_['cid']
|
|
verify = company_auth.VerifyInfo()
|
|
verify.name = json_['company']
|
|
verify.code = json_['code']
|
|
verify.legal_person = json_['legal_person']
|
|
company_auth.verify_info = verify
|
|
info = company_auth.enterprise_certification()
|
|
return info
|
|
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|
|
|
|
|
|
@test_route.route('/change_verified', methods=['GET'])
|
|
@verify_token
|
|
def change_verified(**kwargs):
|
|
"""企业认证状态修改"""
|
|
try:
|
|
args = request.args
|
|
RouteUtil.require_params_check(args, ['cid'])
|
|
|
|
company_auth = CompanyAuthImpl()
|
|
company_auth.cid = args['cid']
|
|
company_auth.change_verified()
|
|
return {"info": "执行成功"}, 200
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|
|
|
|
|
|
@test_route.route('/generate_rid_information', methods=['POST'])
|
|
@verify_token
|
|
def generate_information(**kwargs):
|
|
"""根据rid生成企业主页信息"""
|
|
try:
|
|
json_ = request.json
|
|
RouteUtil.require_params_check(json_, ["rid"])
|
|
|
|
company_index = CompanyIndexImpl()
|
|
company_index.rid = json_['rid']
|
|
info = company_index.generate_homepage()
|
|
return info
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|
|
|
|
|
|
@test_route.route('/new_company', methods=['POST'])
|
|
@verify_token
|
|
def new_company_route(**kwargs):
|
|
"""新增一家测试企业"""
|
|
try:
|
|
json_ = request.json
|
|
RouteUtil.require_params_check(json_, ["email", "company", "code", "legal_person"])
|
|
|
|
company_auth = CompanyAuthImpl()
|
|
company_auth.email = json_['email']
|
|
company_auth.name = json_['company']
|
|
res = company_auth.company_register()
|
|
if type(res) is str:
|
|
company_auth.cid = res
|
|
verify = company_auth.VerifyInfo()
|
|
verify.name = json_['company']
|
|
verify.code = json_['code']
|
|
verify.legal_person = json_['legal_person']
|
|
company_auth.verify_info = verify
|
|
info = company_auth.enterprise_certification()
|
|
elif res is False:
|
|
info = '注册失败'
|
|
else:
|
|
info = res
|
|
return info
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|
|
|
|
|
|
@test_route.route('/delete_rid_cascade', methods=['POST'])
|
|
@verify_token
|
|
def delete_rid_cascade(**kwargs):
|
|
"""级联删除与评价ID有关的所有数据"""
|
|
try:
|
|
json_ = request.json
|
|
RouteUtil.require_params_check(json_, ["rid", "record", "data"])
|
|
|
|
delete_data = DeleteCompanyDataImpl()
|
|
delete_data.rid = json_['rid']
|
|
record = json_['record']
|
|
data = json_['data']
|
|
info = delete_data.form_rid_delete(record, data)
|
|
return {"info": info}, 200
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|
|
|
|
|
|
@test_route.route('/delete_cid_cascade', methods=['POST'])
|
|
@verify_token
|
|
def delete_cid_cascade(**kwargs):
|
|
"""级联删除与企业ID有关的所有数据"""
|
|
try:
|
|
json_ = request.json
|
|
RouteUtil.require_params_check(json_, ["cid", "data"])
|
|
|
|
delete_data = DeleteCompanyDataImpl()
|
|
delete_data.cid = json_['cid']
|
|
data = json_['data']
|
|
info = delete_data.form_cid_delete(data)
|
|
return {"info": info}, 200
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|
|
|
|
|
|
@test_route.route('/insert_esg_data', methods=['POST'])
|
|
@verify_token
|
|
def insert_esg_data(**kwargs):
|
|
"""自动化生成esg问卷填报数据"""
|
|
try:
|
|
json_ = request.json
|
|
RouteUtil.require_params_check(json_, ["rid"])
|
|
|
|
generate_data = GenerateFillInData()
|
|
generate_data.rid = json_['rid']
|
|
info = generate_data.insert_esg_data()
|
|
return {"info": info}, 200
|
|
except APIReturnError as e:
|
|
return {"info": e.error_info}, e.status_code
|
|
except ReturnConditionCheckFailed as e:
|
|
e.log_error()
|
|
return {"info": e.failed_info}, e.status_code
|