From d1933573d6fa934fbc7d259aa006e30e58885017 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=AD=E6=A3=AE?= <pengsen@fecr.com.cn>
Date: Fri, 31 Dec 2021 15:23:15 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E4=BF=A1=E7=94=A8=E5=88=86=E6=9E=90?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 company/analysis/scripts.py | 25 ++++++++++++-------------
 company/routes.py           | 20 ++++++++++++++++++++
 2 files changed, 32 insertions(+), 13 deletions(-)

diff --git a/company/analysis/scripts.py b/company/analysis/scripts.py
index 11ac4a3..fb084a4 100644
--- a/company/analysis/scripts.py
+++ b/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():
-        describe = list()
-        describe.append(data['报告内容'][3]['章节内容'][0]['小节内容'][1]['段落'])
-        describe.append(data['报告内容'][3]['章节内容'][1]['小节内容'][2]['段落'])
-        describe = ''.join(describe)
+        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')
\ No newline at end of file
+    return result
diff --git a/company/routes.py b/company/routes.py
index f787841..6663cc6 100644
--- a/company/routes.py
+++ b/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