from flask import Blueprint, request

from Manage.Company.CompanyImpl import SearchCompanyBodyImpl
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_company(**kwargs):
    """企业搜索"""
    try:

        req = request.json

        search_body_impl = SearchCompanyBodyImpl()
        search_body = SearchCompanyBodyImpl().SearchBody()
        sort_body = SearchCompanyBodyImpl().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('/index', methods=['POST'])
@verify_token
def company_index(**kwargs):
    """
    企业主页
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    return {"info": '调整中'}, 200


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

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