import pymongo

client_tfse = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
client_tyc = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'gP%40DwMSVd5Sh6EiH', '116.63.130.34', 27019))


def find_data_in_tyc(param1, param2, param3):
    """
    根据企业ID查询企业信息
    Parameters:
        param1: string 数据库 db
        param2: string 数据集 collection
        param3: dict   查询条件 例如 {"企业名称": "xxx有限公司"}
    Returns:
        record: 查询结果
    """
    collection = client_tyc[param1][param2]
    data = collection.find(param3, {'_id': False})
    record = list(data)
    return record


def find_data_in_tfse(param1, param2, param3):
    """
    根据企业ID查询企业信息
    Parameters:
        param1: string 数据库 db
        param2: string 数据集 collection
        param3: dict   查询条件 例如 {"企业名称": "xxx有限公司"}
    Returns:
        record: 查询结果
    """
    collection = client_tfse[param1][param2]
    data = collection.find(param3, {'_id': False})
    record = list(data)
    return record


def insert_data_to_tfse(param1, param2, param3):
    """
    插入数据
    Parameters:
        param1: string 数据库
        param2: string 数据集
        param3: dict 数据
    Returns:
        -
    """
    collection = client_tfse[param1][param2]
    collection.insert_one(param3)


def conserve_data_to_tfse(param1, param2, param3, param4):
    """
    保存数据
    Parameters:
        param1: string 数据库
        param2: string 数据集
        param3: string cid
        param4: object 需要保存的数据
    Returns:
        -
    """
    collection = client_tfse[param1][param2]
    collection.update_one(param3, {"$set": param4}, upsert=True)