增加财务指标得分计算接口

This commit is contained in:
王思川 2021-12-20 17:50:28 +08:00
parent 7afc65de84
commit 283e2f6a0d
2 changed files with 32 additions and 2 deletions

View File

@ -14,7 +14,7 @@ class RatingModel:
综合信用评价模型
"""
def __init__(self, param):
def __init__(self):
"""
模型初始化方法
Parameters
@ -30,6 +30,21 @@ class RatingModel:
financial_indicators list 财务指标
"""
# Parameters
self.eid = None
self.cid = None
self.company = None
self.industry = None
self.period = None
self.questionnaire = None
self.financial_statement = None
self.risk_data = None
self.financial_indicators = None
self.operating_score = None
self.financial_score = None
self.risk_score = None
self.rating_result = None
def prepare_params(self, param):
self.eid = param['评价ID']
self.cid = param['企业ID']
self.company = param['企业名称']

View File

@ -11,9 +11,24 @@ rating_route = Blueprint('rating', __name__)
def general_model():
input_data = request.json['input_data']
try:
model = RatingModel(input_data)
model = RatingModel()
model.prepare_params(input_data)
model.save_indicators()
model.save_result()
return {"info": "模型执行成功", "result": model.rating_result['信用等级']}, 200
except Exception:
return {"info": "模型执行失败"}, 210
@rating_route.route('/financial_score', methods=['POST'])
@verify_token
def financial_score():
input_data = request.json['input_data']
try:
model = RatingModel()
model.industry = input_data['所属行业']
model.financial_indicators = input_data['财务指标']
result = model.calculate_financial_elements_score()
return {"info": "模型执行成功", "result": result}, 200
except Exception:
return {"info": "模型执行失败"}, 210