34 lines
741 B
Python
34 lines
741 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:
|
|
Returns:
|
|
threshold
|
|
"""
|
|
collection = client['模型']['财务指标阈值']
|
|
data = collection.find({"二级行业": param}, {'_id': False})
|
|
return list(data)
|
|
|
|
|
|
def conserve_data(**kwargs):
|
|
"""
|
|
保存数据
|
|
Parameters:
|
|
kwargs: 查询条件
|
|
Returns:
|
|
保存结果
|
|
"""
|
|
collection = client[kwargs['db']][kwargs['col']]
|
|
collection.update_one(kwargs['find'], {"$set": kwargs['insert_data']}, upsert=True)
|
|
return '保存成功'
|