import pymongo

MG_USER = "root"
MG_PWD = "sromitdTW569kC#M"
MG_HOST = "116.63.130.34"
MG_PORT = "27018"
CLIENT = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(MG_USER, MG_PWD, MG_HOST, MG_PORT))


def FIND_DATA_IN_COMPANY(param1, param2):
    """
    从企业数据库中查询数据
    Parameters:
        param1 str 数据表
        param2 dict 查询条件
    Returns:
        record: 查询结果
    """
    collection = CLIENT['企业'][param1]
    data = collection.find(param2, {'_id': False})
    record = list(data)
    return record


def FIND_INDUSTRY_DATA_IN_RATING_INPUT(cid):
    """
    从评价数据库中查询数据
    Parameters:
        cid str 企业ID
    Returns:
        record: 查询结果
    """
    collection = CLIENT['评价']['综合评价填报']
    data = collection.find({"企业ID": cid}, {'_id': False, "行业选择": 1, "评价年度": 1})
    record = list(data)
    return record


def FIND_DATA_IN_MACRO_INDUSTRY_ANALYSIS(industry):
    """
    从宏观行业分析数据库中查询行业宏观分析信息
    Parameters:
        industry: str 行业
    Returns:
        record: 查询结果
    """
    collection = CLIENT['宏观']['行业宏观分析']
    data = collection.find({"行业": industry}, {'_id': False})
    record = list(data)
    if record:
        return record[0]
    else:
        return False