import pymongo

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


def find_rating_records(param):
    """
    查询评价记录
    Parameters:
        param: 查询条件
    Returns:
        record: 查询结果
    """
    collection = client['评价']['评价记录']
    data = collection.find(param, {'_id': False})
    record = list(data)
    return record


def find_process_records(param1, param2):
    """
    查询流程记录
    Parameters:
        param1: 数据集名称
        param2: 查询条件
    Returns:
        record: 查询结果
    """
    collection = client['评价'][param1]
    data = collection.find(param2, {'_id': False})
    record = list(data)
    return record


def insert_process_records(data):
    """
    插入一条新流程记录
    Parameters:
        data: 流程数据
    Returns:
        -
    """
    collection = client['评价']['综合评价流程']
    collection.insert_one(data)


def insert_rating_records(data):
    """
    插入一条新评价记录
    Parameters:
        data: 新评价记录
    Returns:
        -
    """
    collection = client['评价']['评价记录']
    collection.insert_one(data)


def find_rating_inputs(param):
    """
    查询填报信息记录
    Parameters:
        param: 查询条件
    Returns:
        records: 查询结果
    """
    collection = client['评价']['综合评价填报']
    data = collection.find(param, {'_id': False})
    record = list(data)
    return record


def insert_general_input(data):
    """
    插入一条新综合填报
    Parameters:
        data: 新综合填报数据
    Returns:
        -
    """
    collection = client['评价']['综合评价填报']
    collection.insert_one(data)


def update_general_input(rid, data):
    """
    更新综合填报数据
    Parameters:
        rid: 评价ID
        data: 综合填报数据
    Returns:
        res: desc
    """
    collection = client['评价']['综合评价填报']
    collection.update_one({"评价ID": rid}, {"$set": data})


def find_risk_data(rid):
    """
    查找评价所需的风险数据
    """
    collection = client['评价']['风险数据']
    data = collection.find({"评价ID": rid}, {'_id': False})
    record = list(data)
    return record


def update_rating_records(find, data):
    """
    更新评价记录
    Parameters:
        find 查询条件
        data 更新数据
    Returns:
        -
    """
    collection = client['评价']['评价记录']
    collection.update_one(find, {"$set": data})