import pymongo

MG_USER = "root"
MG_PWD = "RYIHrqml#LSW6#!*"
MG_HOST = "116.63.130.34"
MG_PORT = "27020"
CLIENT = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(MG_USER, MG_PWD, MG_HOST, MG_PORT))


def INSERT_DATA(param1, param2, param3):
    """
    插入一条数据
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 插入数据
    Returns:
        True 执行成功
        False 执行失败
    """
    collection = CLIENT[param1][param2]
    collection.insert_one(param3)
    return True


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


def UPDATE_DATA(param1, param2, param3, param4):
    """
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
        param4: obj 插入数据
    Returns:
        True 执行完成
    """
    collection = CLIENT[param1][param2]
    collection.update_one(param3, param4)
    return True


def UPSERT_DATA(param1, param2, param3, param4):
    """
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
        param4: obj 插入数据
    Returns:
        True 执行完成
    """
    collection = CLIENT[param1][param2]
    collection.update_one(param3, {"$set": param4}, upsert=True)
    return True


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


def FIND_DATA_PAGE(param1, param2, param3, param4, param5):
    """
    根据查询条件查询符合条件的所有数据
    Args:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
        param4: skip 页码
        param5: limit 每页条数
    Returns:
        records: list 符合条件的数据
    """
    collection = CLIENT[param1][param2]
    param4 = (param4 - 1) * param5
    count = collection.find(param3).count()
    records = list(collection.find(param3, {'_id': False}).skip(param4).limit(param5))
    result = {
        "total": count,
        "data": records
    }
    return result


def UPDATE_INFO(param1, param2, param3, param4):
    """
    更新信息
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
        param4: obj 插入数据
    Returns:
        record: 编辑结果
    """
    collection = CLIENT[param1][param2]
    res = collection.update_one(param3, {"$set": param4})
    info = res.raw_result['updatedExisting']
    return info