import pymongo

clients = {
    "tfse": pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018)),
    "tfse_admin": pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'RYIHrqml#LSW6#!*', '116.63.130.34', 27020)),
}


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


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


def insert_data(param1, param2, param3, param4):
    """
    插入一条数据
    Parameters:
        param1: str 服务器
        param2: str 数据库
        param3: str 数据集
        param4: obj 插入数据
    Returns:
        True 执行成功
        False 执行失败
    """
    try:
        collection = clients[param1][param2][param3]
        collection.insert_one(param4)
        return True
    except Exception:
        return False