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

36 lines
779 B
Python

"""
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)