add sqltap profiling agsi

store/neurostore/core.py

@@ -11,11 +11,23 @@
 from connexion.resolver import MethodResolver
 from flask_caching import Cache
 import sqltap.wsgi
+import sqltap
 
 from .or_json import ORJSONDecoder, ORJSONEncoder
 from .database import init_db
 
 
+class SQLTapMiddleware:
+    def __init__(self, app):
+        self.app = app
+
+    async def __call__(self, scope, receive, send):
+        profiler = sqltap.start()
+        await self.app(scope, receive, send)
+        statistics = profiler.collect()
+        sqltap.report(statistics, "report.txt", report_format="text")
+
+
 connexion_app = connexion.FlaskApp(__name__, specification_dir="openapi/")
 
 app = connexion_app.app
@@ -45,6 +57,11 @@
     allow_headers=["*"],
 )
 
+# add sqltap
+connexion_app.add_middleware(
+    SQLTapMiddleware,
+)
+
 connexion_app.add_api(
     openapi_file,
     base_path="/api",

store/neurostore/models/data.py

@@ -150,7 +150,8 @@ class BaseStudy(BaseMixin, db.Model):
     user = relationship("User", backref=backref("base_studies"))
     # retrieve versions of same study
     versions = relationship(
-        "Study", backref=backref("base_study"))
+        "Study", backref=backref("base_study")
+    )
 
     def update_has_images_and_points(self):
         # Calculate has_images and has_coordinates for the BaseStudy

store/neurostore/tests/api/test_studysets.py

@@ -23,7 +23,7 @@ def test_post_and_get_studysets(auth_client, ingest_neurosynth, session):
         == post_resp.json()
     )
 
-@add_event_listeners
+# @add_event_listeners
 def test_add_many_studies_to_studyset(auth_client, ingest_neurosynth, session):
     existing_studies = Study.query.all()
     existing_study_ids = [s.id for s in existing_studies]