保存模型计算结果
This commit is contained in:
parent
d558ca283e
commit
bf45bfdfef
|
@ -1,7 +1,7 @@
|
||||||
import copy
|
import copy
|
||||||
import time
|
import time
|
||||||
|
|
||||||
from Rating.db import conserve_data
|
from Rating.db import update_data
|
||||||
from Rating.scripts.financial_score import financial_score
|
from Rating.scripts.financial_score import financial_score
|
||||||
from Rating.scripts.operating_score.Manufacturing import manufacturing
|
from Rating.scripts.operating_score.Manufacturing import manufacturing
|
||||||
from Rating.scripts.risk_score import *
|
from Rating.scripts.risk_score import *
|
||||||
|
@ -187,18 +187,20 @@ class RatingModel:
|
||||||
"""
|
"""
|
||||||
保存评价结果
|
保存评价结果
|
||||||
"""
|
"""
|
||||||
# rating_result = self.rating_result
|
rating_result = self.rating_result
|
||||||
# conserve_data(db='评价', col='评价结果', find={"企业ID": self.cid}, insert_data=rating_result)
|
update_data('评价', '评价结果', {"企业ID": self.cid}, rating_result)
|
||||||
|
|
||||||
def save_indicators(self):
|
def save_indicators(self):
|
||||||
"""
|
"""
|
||||||
保存指标值
|
保存指标值
|
||||||
"""
|
"""
|
||||||
# conserve_data(db='企业', col='指标明细', find={"企业名称": self.company, "年报期": ele['年报期']}, insert_data=ele)
|
for item in self.financial_indicators:
|
||||||
|
item['企业ID'] = self.cid
|
||||||
|
item['企业名称'] = self.company
|
||||||
|
update_data('企业', '指标明细', {"企业名称": self.company, "年报期": item['年报期']}, item)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
import json
|
data = read_json_file(file_path('/static/test_data/demo01.json'))
|
||||||
with open("../static/test_data/demo01.json", "r", encoding="utf-8") as f:
|
model = RatingModel(data)
|
||||||
data = json.load(f)
|
model.save_result()
|
||||||
res = RatingModel(data)
|
|
||||||
|
|
18
Rating/db.py
18
Rating/db.py
|
@ -11,23 +11,25 @@ def find_threshold(param):
|
||||||
"""
|
"""
|
||||||
根据行业查询阈值
|
根据行业查询阈值
|
||||||
Parameters:
|
Parameters:
|
||||||
param:
|
param: str 二级行业
|
||||||
Returns:
|
Returns:
|
||||||
threshold
|
data 查询结果
|
||||||
"""
|
"""
|
||||||
collection = client['模型']['财务指标阈值']
|
collection = client['模型']['财务指标阈值']
|
||||||
data = collection.find({"二级行业": param}, {'_id': False})
|
data = collection.find({"二级行业": param}, {'_id': False})
|
||||||
return list(data)
|
return list(data)
|
||||||
|
|
||||||
|
|
||||||
def conserve_data(**kwargs):
|
def update_data(db, table, find, data):
|
||||||
"""
|
"""
|
||||||
保存数据
|
保存数据
|
||||||
Parameters:
|
Parameters:
|
||||||
kwargs: 查询条件
|
db: str 数据库
|
||||||
|
table: str 表
|
||||||
|
find: dict 查询条件
|
||||||
|
data: dict 更新数据
|
||||||
Returns:
|
Returns:
|
||||||
保存结果
|
-
|
||||||
"""
|
"""
|
||||||
collection = client[kwargs['db']][kwargs['col']]
|
collection = client[db][table]
|
||||||
collection.update_one(kwargs['find'], {"$set": kwargs['insert_data']}, upsert=True)
|
collection.update_one(find, {"$set": data}, upsert=True)
|
||||||
return '保存成功'
|
|
||||||
|
|
Loading…
Reference in New Issue