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

109 lines
2.3 KiB
Python

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_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})
def insert_process_records(data):
"""
插入一条新流程记录
Parameters:
data: 流程数据
Returns:
-
"""
collection = client['ESG评价数据']['评价流程']
collection.insert_one(data)