1 files changed, 68 insertions, 0 deletions
diff --git a/gn3/api/correlation.py b/gn3/api/correlation.py
new file mode 100644
index 0000000..4e3e07e
--- /dev/null
+++ b/gn3/api/correlation.py
@@ -0,0 +1,68 @@
+"""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
+
+
+correlation = Blueprint("correlation", __name__)
+
+
+
+# xtodo implement neat db setup
+@correlation.before_request
+def connect_db():
+    """add connection to db method"""
+    print("@app.before_request connect_db")
+    db_connection = getattr(g, '_database', None)
+    if db_connection is None:
+        print("Get new database connector")
+        g.db = g._database = create_engine(SQL_URI, encoding="latin1")
+
+    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
+
+
+    
+    try:
+        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