tfse-etl-api-v0.2/CompanyData/CompanyDataRoutes.py

90 lines
2.1 KiB
Python

from flask import Blueprint, request
from common.utils import verify_token
from CompanyData.job_scripts import *
company_route = Blueprint('company', __name__)
@company_route.route('/init_company_data', methods=['POST'])
@verify_token
def basic_info():
"""
初始化企业数据
"""
req = request.json
cid = req['cid']
company_name = req['company_name']
# 拉取企业数据
info1 = drag_company_data_request(company_name)
if info1 != "企业数据拉取成功":
return {"info": "数据拉取失败"}, 210
# 清洗基本信息数据
basic_info_etl(cid, company_name)
return {"info": "企业基本信息初始化成功"}
@company_route.route('/company_index_etl', methods=['POST'])
@verify_token
def general_rating():
"""
1.准备综合评价信息数据
2.准备财务要素信息
3.准备风险要素信息
4.准备指标明细数据 (模型计算过程中已完成)
Parameters:
rid str 评价ID
cid str 企业ID
Returns:
-
"""
rid = request.json['rid']
cc_rating_etl(rid)
financial_analysis(rid)
risk_analysis_etl(rid)
return {"info": "数据准备完成"}, 200
@company_route.route('/esg_index_etl', methods=['POST'])
@verify_token
def esg_rating():
"""
1.ESG打分结果数据
Parameters:
rid str 评价ID
cid str 企业ID
Returns:
-
"""
rid = request.json['rid']
esg_rating_etl(rid)
return {"info": "数据准备完成"}, 200
@company_route.route('/update_process_data', methods=['POST'])
@verify_token
def update_process():
"""
更新流程数据
Parameters:
rid str 评价ID
data dict 需要更新的数据
Returns:
-
"""
rid = request.json['rid']
renew_data = request.json['renew_data']
types = request.json['types']
db = "综信评价数据" if types == "综合评价流程" else "ESG评价数据"
info = update_process_scripts(db, rid, renew_data)
status = 200 if info else 210
result = '更新成功' if info else '更新失败'
return {"info": result}, status