update 存储报告摘要

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')

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)
+

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