update 信用分析数据接口

company/analysis/scripts.py

@@ -1,16 +1,17 @@
 from company.db import find_data_in_tfse, conserve_data_to_tfse
 
 
-def credit_analysis(param):
+def credit_analysis(param1, param2):
     """
     存储综合评价分析中信用分析数据
     Parameters:
-        param: 企业ID
+        param1: 企业ID
+        param2: 评价ID
     Returns:
         result: 保存信用分析数据结果
     """
     # Parameters 查询text_model数据
-    data = find_data_in_tfse('评价', '报告数据', {"企业ID": param})[0]
+    data = find_data_in_tfse('评价', '报告数据', {"企业ID": param1, "评价ID": param2})[0]
 
     # Returns
     result = dict()
@@ -22,10 +23,13 @@ def credit_analysis(param):
 
     # 财务分析
     def financial_analysis():
+        if data['行业选择'][0] == '制造业':
             describe = list()
             describe.append(data['报告内容'][3]['章节内容'][0]['小节内容'][1]['段落'])
             describe.append(data['报告内容'][3]['章节内容'][1]['小节内容'][2]['段落'])
             describe = ''.join(describe)
+        else:
+            describe = data['报告内容'][3]['章节内容'][0]['小节内容'][2]['段落']
         return describe
 
     # 风险分析
@@ -61,9 +65,4 @@ def credit_analysis(param):
     result['信用分析.风险分析'] = risk_analysis()
     result['信用分析.评价意见'] = evaluation_comments()
 
-    # 保存数据
-    conserve_data_to_tfse('企业', '综合评价分析', {'企业ID': param}, result)
-
-
-if __name__ == '__main__':
-    credit_analysis('mKJreZRw')
+    return result

company/routes.py

@@ -1,6 +1,7 @@
 from flask import Blueprint, request
 
 from common.scripts import verify_token
+from company.analysis.scripts import credit_analysis
 from company.scripts import *
 
 company_route = Blueprint('company', __name__)
@@ -53,3 +54,22 @@ def general_rating():
     risk_analysis_etl(rid)
 
     return {"info": "数据准备完成"}
+
+
+@company_route.route('/company_credit_etl', methods=['POST'])
+@verify_token
+def credit_etl():
+    """
+    清洗综合信用分析所需要的数据
+    Parameters:
+        rid str 评价ID
+        cid str 企业ID
+    Returns:
+        清洗好的信用分析数据
+    """
+    # Parameters
+    rid = request.json['rid']
+    cid = request.json['cid']
+    data = credit_analysis(cid, rid)
+
+    return {"data": data}, 200