保存模型计算结果

Rating/RatingModel.py

@@ -1,7 +1,7 @@
 import copy
 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.operating_score.Manufacturing import manufacturing
 from Rating.scripts.risk_score import *
@@ -187,18 +187,20 @@ class RatingModel:
         """
         保存评价结果
         """
-        # rating_result = self.rating_result
-        # conserve_data(db='评价', col='评价结果', find={"企业ID": self.cid}, insert_data=rating_result)
+        rating_result = self.rating_result
+        update_data('评价', '评价结果', {"企业ID": self.cid}, rating_result)
 
     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__':
-    import json
-    with open("../static/test_data/demo01.json", "r", encoding="utf-8") as f:
-        data = json.load(f)
-        res = RatingModel(data)
+    data = read_json_file(file_path('/static/test_data/demo01.json'))
+    model = RatingModel(data)
+    model.save_result()

Rating/db.py

@@ -11,23 +11,25 @@ def find_threshold(param):
     """
     根据行业查询阈值
     Parameters:
-        param:
+        param: str 二级行业
     Returns:
-        threshold
+        data 查询结果
     """
     collection = client['模型']['财务指标阈值']
     data = collection.find({"二级行业": param}, {'_id': False})
     return list(data)
 
 
-def conserve_data(**kwargs):
+def update_data(db, table, find, data):
     """
     保存数据
     Parameters:
-        kwargs: 查询条件
+        db: str 数据库
+        table: str 表
+        find: dict 查询条件
+        data: dict 更新数据
     Returns:
-        保存结果
+        -
     """
-    collection = client[kwargs['db']][kwargs['col']]
-    collection.update_one(kwargs['find'], {"$set": kwargs['insert_data']}, upsert=True)
-    return '保存成功'
+    collection = client[db][table]
+    collection.update_one(find, {"$set": data}, upsert=True)