From 6a031cce7f68b653ec63881c68cd15dbeff23ed3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=AD=E6=A3=AE?= <pengsen@fecr.com.cn>
Date: Fri, 24 Dec 2021 16:38:39 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E5=AD=98=E5=82=A8=E6=8A=A5=E5=91=8A?=
 =?UTF-8?q?=E6=91=98=E8=A6=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 company/analysis/scripts.py | 69 +++++++++++++++++++++++++++++++++++++
 company/db.py               | 15 ++++++++
 rating/routes.py            | 10 ++++--
 3 files changed, 92 insertions(+), 2 deletions(-)
 create mode 100644 company/analysis/scripts.py

diff --git a/company/analysis/scripts.py b/company/analysis/scripts.py
new file mode 100644
index 0000000..11ac4a3
--- /dev/null
+++ b/company/analysis/scripts.py
@@ -0,0 +1,69 @@
+from company.db import find_data_in_tfse, conserve_data_to_tfse
+
+
+def credit_analysis(param):
+    """
+    存储综合评价分析中信用分析数据
+    Parameters:
+        param: 企业ID
+    Returns:
+        result: 保存信用分析数据结果
+    """
+    # Parameters 查询text_model数据
+    data = find_data_in_tfse('评价', '报告数据', {"企业ID": param})[0]
+
+    # Returns
+    result = dict()
+
+    # 经营分析
+    def business_analysis():
+        describe = data['报告内容'][1]['章节内容'][0]['小节内容'][1]['段落']
+        return describe
+
+    # 财务分析
+    def financial_analysis():
+        describe = list()
+        describe.append(data['报告内容'][3]['章节内容'][0]['小节内容'][1]['段落'])
+        describe.append(data['报告内容'][3]['章节内容'][1]['小节内容'][2]['段落'])
+        describe = ''.join(describe)
+        return describe
+
+    # 风险分析
+    def risk_analysis():
+        risk_01 = data['报告内容'][4]['章节内容'][0]['小节内容'][0]['段落']
+        list_01 = risk_01.split('，')
+        risk_02 = data['报告内容'][4]['章节内容'][1]['小节内容'][0]['段落']
+        list_02 = risk_02.split('，')
+        risk_03 = data['报告内容'][4]['章节内容'][2]['小节内容'][0]['段落']
+        list_03 = risk_03.split('，')
+        describe = list()
+        describe.append(list_01[::-1][0])
+        describe.append(list_02[::-1][0])
+        describe.append(list_03[::-1][0])
+        describe = ','.join(describe)
+        res = describe.replace('。', '', 2)
+        return res
+
+    # 评价意见
+    def evaluation_comments():
+        describe = list()
+        eva_01 = data['报告内容'][5]['章节内容'][0]['小节内容'][0]['段落']
+        eva_02 = data['报告内容'][5]['章节内容'][0]['小节内容'][1]['段落']
+        eva_03 = data['报告内容'][5]['章节内容'][0]['小节内容'][2]['段落']
+        describe.append(eva_01)
+        describe.append(eva_02)
+        describe.append(eva_03)
+        describe = ''.join(describe)
+        return describe
+
+    result['信用分析.经营分析'] = business_analysis()
+    result['信用分析.财务分析'] = financial_analysis()
+    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
diff --git a/company/db.py b/company/db.py
index f6d267a..7683f95 100644
--- a/company/db.py
+++ b/company/db.py
@@ -49,3 +49,18 @@ def insert_data_to_tfse(param1, param2, param3):
     collection = client_tfse[param1][param2]
     collection.insert_one(param3)
 
+
+def conserve_data_to_tfse(param1, param2, param3, param4):
+    """
+    保存数据
+    Parameters:
+        param1: string 数据库
+        param2: string 数据集
+        param3: string cid
+        param4: object 需要保存的数据
+    Returns:
+        -
+    """
+    collection = client_tfse[param1][param2]
+    collection.update_one(param3, {"$set": param4}, upsert=True)
+
diff --git a/rating/routes.py b/rating/routes.py
index d8971e6..91bd3ad 100644
--- a/rating/routes.py
+++ b/rating/routes.py
@@ -1,5 +1,6 @@
 from flask import Blueprint, request
 from common.scripts import verify_token
+from company.analysis.scripts import credit_analysis
 from rating.ReportEtl import CleanModel
 from rating.scripts.risk_count import *
 
@@ -37,6 +38,11 @@ def report_data():
         -
     """
     info = request.json
+
+    # 准备需要清洗的数据
     db_data = get_tfse_data(info)
-    result = CleanModel(info, db_data).storage_data()
-    return {"data": result}, 200
+
+    # 调用清洗数据模型
+    data = CleanModel(info, db_data).storage_data()
+
+    return {"data": data}, 200