from company.db import *


def search_company_script(query, skip, limit):
    """
    查询企业信息
    Parameters:
        query: 查询条件（企业名称/企业id）
        skip: 默认0
        limit: 每页显示条数
    Returns:
        result: 返回数据
    """
    result = find_company_records(query, skip, limit)
    return result


def company_index_script(cid):
    """
    查询企业相关信息
    Parameters:
        cid: 企业ID
    Returns:
        result: 返回数据
    """
    user_info = find_user_info(cid)
    basic_info = find_company_basic_info(cid)
    evaluate_info = find_company_evaluate_info(cid)
    result = dict()
    if len(user_info) > 0:
        result['企业名称'] = user_info[0]['企业名称']
        result['企业ID'] = cid
        result['已认证'] = user_info[0]['已认证']
    if len(basic_info) > 0:
        result['基本信息'] = dict()
        result['基本信息']['工商信息'] = basic_info[0]['工商信息']
        result['基本信息']['股东信息'] = basic_info[0]['股东信息']
        result['基本信息']['主要成员'] = basic_info[0]['主要成员']
    if len(evaluate_info) > 0:
        result['评价记录'] = list()
        result['评价记录'] = evaluate_info
    return result