40 lines
862 B
Python
40 lines
862 B
Python
"""
|
|
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)
|