tfse-admin-api-v0.2/Manage/Company/CompanyRoutes.py

from flask import Blueprint, request

from Manage.Company.CompanyImpl import SearchCompanyApiImpl, CompanyIndexApiImpl
from Utils.ErrorUtil import ReturnConditionCheckFailed
from user.user_auth import verify_token

company_route = Blueprint('company', __name__)


@company_route.route('/search', methods=['POST'])
@verify_token
def search_route(**kwargs):
    """企业搜索"""
    try:

        req = request.json

        search_body_impl = SearchCompanyApiImpl()
        search_body = SearchCompanyApiImpl().SearchBody()
        sort_body = SearchCompanyApiImpl().SortBody()

        search = req['search'] if req.__contains__('search') else {}
        sort = req['sort'] if req.__contains__('sort') else {}

        if search.__contains__('企业ID'):
            search_body.cid = search['企业ID']

        if search.__contains__('企业名称'):
            search_body.name = search['企业名称']

        if search.__contains__('一级行业'):
            search_body.industry = search['一级行业']

        if search.__contains__('综信评价'):
            search_body.credit_level = search['综信评价']

        if search.__contains__('ESG评价'):
            search_body.esg_level = search['ESG评价']

        if sort != {}:
            sort_body.field = list(sort.keys())[0]
            sort_body.sort = list(sort.values())[0]

        search_body_impl.search = search_body
        search_body_impl.sort = sort_body
        search_body_impl.page_size = req['page_size']
        search_body_impl.page_no = req['page_no']

        result = search_body_impl.conditions_search()
        return {"info": '查询结果', "result": result}, 200
    except ReturnConditionCheckFailed as e:
        return {"info": e.failed_info}, e.status_code


@company_route.route('/head_info', methods=["GET"])
@verify_token
def head_info_route(**kwargs):
    """头部信息"""
    try:
        impl = CompanyIndexApiImpl()
        impl.head_info.cid = kwargs['cid']
        impl.get_head_info()
        result = impl.head_info.dict_to_show()
        return {"info": "查询结果", "result": result}, 200
    except ReturnConditionCheckFailed as e:
        e.log_error()
        return {"info": e.failed_info}, e.status_code


@company_route.route('/index', methods=['POST'])
@verify_token
def index_route(**kwargs):
    """企业主页"""
    return {"info": '调整中'}, 200


@company_route.route('/edit', methods=['POST'])
@verify_token
def edit_route(**kwargs):
    """
    企业信息编辑
    Parameters:
        -
    Returns:
        result: 编辑结果
    """

    return {"info": '调整中'}, 200