tfse-admin-api-v0.2/TestCenter/tc_routes.py

from flask import Blueprint, request

from TestCenter.tc_obj import TestObj
from common.APIAuth import api_secret
from user.user_auth import verify_token

test_route = Blueprint('test', __name__)


@test_route.route('/enterprise_certification', methods=['POST'])
@api_secret
def enterprise_certification():
    """测试用企业认证"""
    test_obj = TestObj()
    test_obj.company = request.json['company']
    test_obj.code = request.json['code']
    test_obj.legal_person = request.json['legal_person']
    test_obj.cid = request.json['cid']
    return {"info": "调整中"}, 200


@test_route.route('/search_evaluate_record', methods=['POST'])
@api_secret
def search_evaluate_record_route():
    """
    根据评价类型查询评价数据库
    Parameters:
        -
    Returns:
        -
    """
    test_obj = TestObj()
    test_obj.types = request.json['types']
    data = test_obj.search_evaluate_record()
    status = 200 if data else 210
    return {"data": data}, status


@test_route.route('/delete_rid_cascade', methods=['POST'])
@api_secret
def delete_rid_cascade_route():
    """
    级联删除与评价ID有关的所有数据
    Parameters:
        -
    Returns:
        -
    """
    test_obj = TestObj()
    test_obj.rid = request.json['rid']
    test_obj.record = request.json['record']
    test_obj.data = request.json['data']
    info = test_obj.delete_rid_cascade()
    status = 200 if info == '删除成功' or info == '状态更改成功' else 210
    return {"info": info}, status


@test_route.route('/delete_cid_cascade', methods=['POST'])
@api_secret
def delete_uid_cascade_route():
    """
    级联删除与uid有关的所有数据
    Parameters:
        -
    Returns:
        -
    """
    test_obj = TestObj()
    test_obj.uid = request.json['cid']
    test_obj.data = request.json['data']
    info = test_obj.delete_cid_cascade()
    status = 200 if info == '企业数据删除成功' else 210
    return {"info": "删除成功"}, status


@test_route.route('/generate_rid_information', methods=['POST'])
@api_secret
def generate_rid_information_route():
    """
    根据rid生成企业主页信息
    Parameters:
        -
    Returns:
        -
    """
    test_obj = TestObj()
    test_obj.rid = request.json['rid']
    test_obj.types = request.json['types']
    info = test_obj.generate_rid_information()
    status = 200 if info == '执行完成' else 210
    return {"info": info}, status


@test_route.route('/change_verified', methods=['GET'])
@api_secret
def change_verified():
    """
    企业认证状态修改
    Parameters:
        -
    Returns:
        info: 修改结果结果
    """
    test_obj = TestObj()
    test_obj.cid = request.args.get('cid')
    info = test_obj.change_verified()
    status = 200 if info else 210
    res = '认证状态修改成功' if info else '认证状态修改失败'
    return {"info": res}, status


@test_route.route('/new_company', methods=['POST'])
@verify_token
def new_company_route(**kwargs):
    """
    新增一家测试企业
    """
    test_obj = TestObj()
    test_obj.email = request.json['email']
    test_obj.company = request.json['company']
    test_obj.code = request.json['code']
    test_obj.legal_person = request.json['legal_person']
    info = test_obj.new_test_company()
    return {"info": info}, 200