tfse-model-api-v0.2/Rating/db.py

"""
Mongo_tfse
"""

import pymongo

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


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)