tfse-admin-api-v0.2/Test/TestRoute.py

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