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

50 lines
1.2 KiB
Python

from flask import Blueprint, request
from common.scripts import verify_token
from rating.ReportEtl import CleanModel
from rating.scripts.common import get_tfse_data
from rating.scripts.risk_count import *
rating_route = Blueprint('rating', __name__)
@rating_route.route('/risk_data', methods=['POST'])
@verify_token
def risk_data():
"""
风险数据清洗
Parameters:
company_name 企业名称
company_id 企业ID
evaluation_id 评价ID
Returns:
res: desc
"""
company_name = request.json['company_name']
company_id = request.json['company_id']
evaluation_id = request.json['evaluation_id']
risk_data_etl(company_name, company_id, evaluation_id)
return {"info": "执行成功"}, 200
@rating_route.route('/report_data', methods=['POST'])
@verify_token
def report_data():
"""
保存报告数据
Parameters:
rid str 评价ID
cid str 企业ID
company str 企业名称
Returns:
-
"""
info = request.json
# 准备需要清洗的数据
db_data = get_tfse_data(info)
# 调用清洗数据模型
data = CleanModel(info, db_data).storage_data()
return {"data": data}, 200