import pymongo
from bson import ObjectId
from gridfs import GridFS

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


def find_records(param1, param2, param3):
    """
    查询数据库相关记录
    Parameters:
        param1: str 数据库名称
        param2: str 数据集名称
        param3: obj 查询条件
    Returns:
        result: bool 执行结果
    """
    collection = client[param1][param2]
    res = list(collection.find(param3, {'_id': False}))
    return res


def insert_data(param1, param2, param3):
    """
    插入一条数据
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
    Returns:
        result: bool 执行结果
    """
    collection = client[param1][param2]
    collection.insert_one(param3)
    return True


def update_data(param1, param2, param3, param4):
    """
    更新一条数据
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
        param4: obj 更新数据
    Returns:
        result: bool 更新结果
    """
    collection = client[param1][param2]
    res = collection.update_one(param3, {"$set": param4}, upsert=True)
    info = res.raw_result['ok']
    return info