from flask import Blueprint, request
from common.security.APIAuth import api_secret
from company.Company import Company

company_route = Blueprint('company', __name__)


@company_route.route('/search', methods=['POST'])
@api_secret
def search_company():
    """
    企业搜索
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    company = Company()
    company.query = request.json['query']
    data = company.search_company()
    return {"info": '查询成功', "result": data}, 200


@company_route.route('/index', methods=['POST'])
@api_secret
def company_index():
    """
    企业主页
    Parameters:
        -
    Returns:
        info: 返回信息
        result: 返回数据
    """
    company = Company()
    company.cid = request.json['cid']
    data = company.company_index()
    return {"info": '企业信息查询成功', "result": data}, 200