update 更新流程

Esg/EsgModel.py

@@ -5,7 +5,7 @@ from Esg.db import update_data
 from Esg.scripts.environmental import environmental_rating
 from Esg.scripts.governance import governance_rating
 from Esg.scripts.social import social_rating
-from common.scripts import read_json_file
+from common.scripts import read_json_file, update_process_status
 
 
 class EsgModel:
@@ -90,6 +90,7 @@ class EsgModel:
         """
         rating_result = self.rating_result
         update_data('ESG评价数据', '评价结果', {"企业ID": self.cid, "评价年度": self.period}, rating_result)
+        update_process_status(self.rid, 'ESG评价流程', {"评价流程.评价结果": 1})
 
     @staticmethod
     def score_rank(param):

Rating/RatingModel.py

@@ -1,5 +1,4 @@
 import copy
-import json
 import time
 
 from Rating.db import update_data
@@ -8,7 +7,7 @@ from Rating.scripts.operating_score.Information_technology import infomation_tec
 from Rating.scripts.operating_score.Manufacturing import manufacturing
 from Rating.scripts.risk_score import *
 from Rating.scripts.Indicators import *
-from common.scripts import read_json_file, sub_dict
+from common.scripts import read_json_file, sub_dict, update_process_status
 
 
 class RatingModel:
@@ -219,6 +218,7 @@ class RatingModel:
         """
         rating_result = self.rating_result
         update_data('综信评价数据', '评价结果', {"企业ID": self.cid}, rating_result)
+        update_process_status(self.eid, '综合评价流程', {"评价流程.评价结果": 1})
 
     def save_indicators(self):
         """
@@ -229,3 +229,4 @@ class RatingModel:
             item['评价ID'] = self.eid
             item['企业名称'] = self.company
             update_data('综信评价数据', '财指结果', {"企业ID": self.cid, "年报期": item['年报期']}, item)
+        update_process_status(self.eid, '综合评价流程', {"评价流程.财指结果": 1})

Report/ReportImpl.py

@@ -8,6 +8,7 @@ from DBHelper.MongoHelper import MongoHelper
 from Report.PdfReport import ReportGenerator
 from Report.ReportObj import ReportData
 from Report.scripts.path_tool import get_gen_report_path
+from common.scripts import update_process_status
 
 
 class HandleReportData:
@@ -941,6 +942,7 @@ class GenerateReport(ReportData):
                 {'评价ID': self.rid},
                 data
             )
+            update_process_status(self.rid, '综合评价流程', {"评价流程.报告数据": 1})
             fid = self.generate_pdf(self.name, data, '综信报告')
             return fid
 
@@ -996,6 +998,7 @@ class GenerateReport(ReportData):
                 {'评价ID': self.rid},
                 report_content
             )
+            update_process_status(self.rid, 'ESG评价流程', {"评价流程.报告数据": 1})
             fid = self.generate_pdf(self.name, report_content, 'ESG报告')
             return fid
 

common/scripts.py

@@ -2,6 +2,7 @@ import os
 import json
 import functools
 
+import requests
 from flask import request
 
 from setting import APP_NAME, API_SECRET
@@ -61,3 +62,20 @@ def sub_dict(param1, param2):
         子集
     """
     return dict((key, value) for key, value in param1.items() if key in param2)
+
+
+def update_process_status(rid, types, update):
+    """
+    调用更新流程接口
+    Parameters:
+        rid 评价id
+        types 综合评价流程/ESG评价流程
+        update dict 需要更新的数据
+    Returns:
+        result str 更新结果
+    """
+    url = "http://api.fecribd.com/etl_tfse/company/update_process_data"
+    token = "dmfd7FshT!5Wng9^gcCcQV7T6FBnVgl4"
+    headers = {'token': token, 'Content-Type': 'application/json; charset=UTF-8'}
+    data = json.dumps({"rid": rid, "types": types, "renew_data": update})
+    requests.post(url, headers=headers, data=data)