tfse-etl-api-v0.2/company/routes.py

74 lines
1.7 KiB
Python

from flask import Blueprint, request
from common.scripts import verify_token
from company.scripts import *
company_route = Blueprint('company', __name__)
@company_route.route('/init_company_data', methods=['POST'])
@verify_token
def basic_info():
"""
初始化企业数据
包括 天眼查数据库、企业基本信息
Parameters:
cid 企业ID
company_name 企业名称
Returns:
res: desc
"""
cid = json.loads(request.data)['cid']
company_name = json.loads(request.data)['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']
general_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