diff options
Diffstat (limited to 'gn3')
| -rw-r--r-- | gn3/api/correlation.py | 36 |
1 files changed, 6 insertions, 30 deletions
diff --git a/gn3/api/correlation.py b/gn3/api/correlation.py index 4e3e07e..217b7ce 100644 --- a/gn3/api/correlation.py +++ b/gn3/api/correlation.py @@ -1,23 +1,17 @@ """Endpoints for computing correlation""" -import pickle import time from flask import Blueprint from flask import jsonify from flask import request from flask import g -from flask import after_this_request -from default_settings import SQL_URI - -# import pymysql - from sqlalchemy import create_engine -from gn3.correlation.correlation_computations import compute_correlation +from default_settings import SQL_URI +from gn3.correlation.correlation_computations import compute_correlation correlation = Blueprint("correlation", __name__) - # xtodo implement neat db setup @correlation.before_request def connect_db(): @@ -31,38 +25,20 @@ def connect_db(): g.initial_time = time.time() -@correlation.after_request -def after_request_func(response): - final_time = time.time() - g.initial_time - print(f"This request for Correlation took {final_time} Seconds") - - g.initial_time = None - - return response - - - - @correlation.route("/corr_compute", methods=["POST"]) def corr_compute_page(): """api for doing correlation""" - # todo accepts both form and json data - correlation_input = request.json if correlation_input is None: - return jsonify({"error": str("Bad request")}),400 - + return jsonify({"error": str("Bad request")}), 400 - try: - corr_results = compute_correlation(correlation_input_data=correlation_input) + corr_results = compute_correlation( + correlation_input_data=correlation_input) - except Exception as error: # pylint: disable=broad-except return jsonify({"error": str(error)}) - return { - "correlation_results":corr_results - }
\ No newline at end of file + return {"correlation_results": corr_results} |