from flask import Blueprint, request
from common.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']
    skip = request.json['skip']
    limit = request.json['limit']
    data = company.search_company(skip, limit)
    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


@company_route.route('/edit', methods=['POST'])
@api_secret
def company_edit():
    """
    企业信息编辑
    Parameters:
        -
    Returns:
        result: 编辑结果
    """
    company = Company()
    company.cid = request.json['cid']
    data = request.json['data']
    result = company.company_edit(data)
    status = 200 if result else 210
    info = "企业信息编辑成功" if result else "企业信息编辑失败"
    return {"info": info}, status