tfse-app-api-v0.2/esg/esg_db.py

92 lines
2.0 KiB
Python

import pymongo
client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
def find_rating_records(param):
"""
查询评价记录
Parameters:
param: 查询条件
Returns:
record: 查询结果
"""
collection = client['评价']['评价记录']
data = collection.find(param, {'_id': False})
record = list(data)
return record
def insert_rating_records(data):
"""
插入一条新评价记录
Parameters:
data: 新评价记录
Returns:
-
"""
collection = client['评价']['评价记录']
collection.insert_one(data)
def find_rating_inputs(param):
"""
查询填报信息记录
Parameters:
param: 查询条件
Returns:
records: 查询结果
"""
collection = client['评价']['ESG评价填报']
data = collection.find(param, {'_id': False})
record = list(data)
return record
def insert_general_input(data):
"""
插入一条新综合填报
Parameters:
data: 新综合填报数据
Returns:
-
"""
collection = client['评价']['ESG评价填报']
collection.insert_one(data)
def update_general_input(rid, data):
"""
更新综合填报数据
Parameters:
rid: 评价ID
data: 综合填报数据
Returns:
res: desc
"""
collection = client['评价']['ESG评价填报']
collection.update_one({"评价ID": rid}, {"$set": data})
def find_risk_data(rid):
"""
查找评价所需的风险数据
"""
collection = client['评价']['ESG风险数据']
data = collection.find({"评价ID": rid}, {'_id': False})
record = list(data)
return record
def update_rating_records(find, data):
"""
更新评价记录
Parameters:
find 查询条件
data 更新数据
Returns:
-
"""
collection = client['评价']['评价记录']
collection.update_one(find, {"$set": data})