Merge branch 'ps' into 'master'

update 优化根据rid生成主页信息接口

See merge request root/tfse_admin!2

TestCenter/tc_impl.py

@@ -146,15 +146,21 @@ def generate_rid_information_script(rid, types):
             result type 返回数据
         """
         record = find_data('评价', '评价记录', {'评价ID': rid, '进行状态': '进行'})
-
         if not record:
-            return False, "没有正在进行的评价"
+            return "没有正在进行的评价"
 
-        inputs = find_data('评价', '综合信用填报', {'评价ID': rid})[0]
+        inputs = find_data('评价', '综合评价填报', {'评价ID': rid})
-        risks = {"风险数据": find_data('评价', '风险数据', {'评价ID': rid})[0]}
+        if not inputs:
-        risks["风险数据"].pop('企业ID')
+            return '未查询到此rid相关填报数据'
-        risks["风险数据"].pop('评价ID')
+
-        rating_inputs = {**inputs, **risks}
+        risks = find_data('评价', '风险数据', {'评价ID': rid})
+        if not risks:
+            return '未查询到此rid相关风险数据'
+
+        risk = {"风险数据": risks[0]}
+        risk["风险数据"].pop('企业ID')
+        risk["风险数据"].pop('评价ID')
+        rating_inputs = {**inputs[0], **risk}
 
         # 打分模型
         def general_model():
@@ -253,14 +259,21 @@ def generate_rid_information_script(rid, types):
         record = find_data('评价', '评价记录', {'评价ID': rid, '进行状态': '进行'})
 
         if not record:
-            return False, "没有正在进行的评价"
+            return "没有正在进行的评价"
 
-        inputs = find_data('评价', 'ESG评价填报', {'评价ID': rid})[0]
+        inputs = find_data('评价', 'ESG评价填报', {'评价ID': rid})
-        risks = {"风险数据": find_data('评价', 'ESG风险数据', {'评价ID': rid})[0]}
+        if not inputs:
-        risks["风险数据"].pop('企业ID')
+            return '未查询到此rid相关填报数据'
-        risks["风险数据"].pop('评价ID')
+
-        risks["风险数据"].pop('更新日期')
+        risks = find_data('评价', 'ESG风险数据', {'评价ID': rid})
-        rating_inputs = {**inputs, **risks}
+        if not risks:
+            return '未查询到此rid相关风险数据'
+
+        risk = {"风险数据": risks[0]}
+        risk["风险数据"].pop('企业ID')
+        risk["风险数据"].pop('评价ID')
+        risk["风险数据"].pop('更新日期')
+        rating_inputs = {**inputs[0], **risk}
 
         # 打分模型
         def general_model():
@@ -307,7 +320,8 @@ def generate_rid_information_script(rid, types):
             data = {
                 "company": rating_inputs['企业名称'],
                 "cid": rating_inputs['企业ID'],
-                "rid": rating_inputs['评价ID']
+                "rid": rating_inputs['评价ID'],
+                "type": 'esg'
             }
             res = requests.post(url, headers=headers, data=json.dumps(data))
             if res.status_code == 200:
@@ -334,16 +348,16 @@ def generate_rid_information_script(rid, types):
                 return False
 
         if not general_model():
-            return False, "打分失败"
+            return "打分失败"
 
         if not gen_report():
-            return False, "生成报告失败"
+            return "生成报告失败"
 
         if not gen_certified():
-            return False, "生成证书失败"
+            return "生成证书失败"
 
         if not company_index():
-            return False, "准备主页数据失败"
+            return "准备主页数据失败"
 
         update_data('评价', '评价记录', {'评价ID': rid}, {'进行状态': '完成'})
         return '执行完成'