import pymongo

client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))


def delete_data(param1, param2, param3):
    """
    根据查询条件删除一条文档
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
    Returns:
        res: True 执行完成
    """
    collection = client[param1][param2]
    collection.delete_one(param3)
    return True


def find_data(param1, param2, param3):
    """
    根据查询条件查询符合条件的所有数据
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
    Returns:
        res: list 符合条件的数据
    """
    collection = client[param1][param2]
    res = list(collection.find(param3, {'_id': False}))
    return res


def update_user_info_by_cid(cid, data):
    """
    更新用户信息
    Parameters:
        cid: 企业ID
        data: 更新数据
    Returns:
        info: 信息
    """
    collection = client['用户']['用户信息']
    res = collection.update_one({"企业ID": cid}, {"$set": data})
    info = res.raw_result['updatedExisting']
    return info


def find_user_info_by_cid(cid):
    """
    更新用户信息
    Parameters:
        cid: 企业ID
    Returns:
        info: 查询结果
    """
    collection = client['用户']['用户信息']
    data = collection.find({"企业ID": cid}, {'_id': False})
    record = list(data)
    return record