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