模型打分接口

.gitlab-ci.yml

@@ -4,8 +4,8 @@ stages:
 job:
   stage: deploy
   script:
-#    - docker stop tfse_rating
-#    - docker rm tfse_rating
+    - docker stop tfse_rating
+    - docker rm tfse_rating
     - docker build -t tfse_rating .
     - docker run -d -p 51012:51012 --name tfse_rating -v /etc/timezone:/etc/timezone:ro -v /etc/localtime:/etc/localtime:ro tfse_rating
   only:

Rating/routes.py

@@ -0,0 +1,14 @@
+from flask import Blueprint, request, Response
+
+from Rating.RatingModel import RatingModel
+
+rating_route = Blueprint('rating', __name__)
+
+
+@rating_route.route('/general_model', methods=['POST'])
+def general_model():
+    input_data = request.json['input_data']
+    model = RatingModel(input_data)
+    model.save_indicators()
+    model.save_result()
+    return {"info": "模型执行成功"}, 200

Report/routes.py

@@ -13,7 +13,7 @@ report_route = Blueprint('result', __name__)
 
 @report_route.route('/gen_pdf', methods=['POST'])
 def gen_pdf():
-    with open("static/test_data/report_template.json", "r", encoding="utf-8") as f:
+    with open("../static/test_data/report_template.json", "r", encoding="utf-8") as f:
         data = json.load(f)
     req = request.json
     name = req['企业名称']

app.py

@@ -1,13 +1,17 @@
 from flask import Flask
 from flask_cors import *
 
-from routes import report_route
+from Rating.routes import rating_route
+from Report.routes import report_route
 
 app = Flask(__name__)
 app.config['JSON_SORT_KEYS'] = False
-app.register_blueprint(report_route, url_prefix='/tfse_rating/report')
 CORS(app, supports_credentials=True)
 
 
+app.register_blueprint(report_route, url_prefix='/tfse_rating/report')
+app.register_blueprint(rating_route, url_prefix='/tfse_rating/rating')
+
+
 if __name__ == '__main__':
     app.run()