"""
Mongo_tfse
"""

import pymongo

DB_HOST = "116.63.130.34"
DB_PASS = "UTlC9cCoglD1cI1*"
DB_USER = "root"
DB_PORT = "27021"
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(DB_USER, DB_PASS, DB_HOST, DB_PORT))


def find_threshold(param):
    """
    根据行业查询阈值
    Parameters:
        param: str 二级行业
    Returns:
        data 查询结果
    """
    collection = client['模型端']['综信评价财指阈值']
    data = collection.find({"二级行业": param}, {'_id': False})
    return list(data)


def update_data(db, table, find, data):
    """
    保存数据
    Parameters:
        db: str 数据库
        table: str 表
        find: dict 查询条件
        data: dict 更新数据
    Returns:
        -
    """
    collection = client[db][table]
    collection.update_one(find, {"$set": data}, upsert=True)