from DBHelper.MongoHelper import MongoHelper
from Manage.company.company_db import *
from Manage.company.company_obj import SearchCompany


class SearchCompanyImpl(SearchCompany):
    """"""
    db = MongoHelper("tfse_v0.21")

    def conditions_search(self):
        """"""


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

        for item in evaluate_info:
            item['报告'] = '/file/get_company_report?file_id={}'.format(item['报告fid'])
            item['证书'] = '/file/get_certification?file_id={}'.format(item['证书fid'])
            item.pop('报告fid')
            item.pop('证书fid')
    return result


def company_edit_script(cid, data):
    """
    编辑企业信息
    Parameters:
        cid: 企业ID
        data: 编辑后数据
    Returns:
        result: 编辑数据
    """
    result = update_company_info(cid, data)
    return result