import pymongo
from bson import ObjectId
from gridfs import GridFS

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 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 find_admin_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 insert_admin_data(param1, param2, param3):
    """
    插入数据（admin）
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
    Returns:
        res: list 符合条件的数据
    """
    collection = client[param1][param2]
    collection.insert_one(param3)
    return True


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 update_data(param1, param2, param3, param4):
    """
    更新用户信息
    Parameters:
        param1: str 数据库
        param2: str 数据集
        param3: obj 查询条件
        param4: obj 更新数据
    Returns:
        info: 更新结果
    """
    collection = client[param1][param2]
    res = collection.update_one(param3, {"$set": param4})
    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


def delete_pdf(param1, param2, param3):
    """
    根据id删除报告/证书
    Parameters:
        param1: 数据库
        param2: 数据集
        param3: 文件fid
    Returns:
        record: 删除结果
    """
    database = param1
    collection = param2
    fs = GridFS(client[database], collection)
    fs.delete(ObjectId(param3))