preload more attributes

neurostuff · Nov 7, 2023 · 94a8e8a · 94a8e8a
1 parent 007a84c
commit 94a8e8a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/store/neurostore/resources/data.py b/store/neurostore/resources/data.py
@@ -7,6 +7,7 @@
 import sqlalchemy.sql.expression as sae
 from sqlalchemy.orm import joinedload
 from sqlalchemy.sql import func
+from sqlalchemy import union
 
 
 from .utils import view_maker
@@ -227,8 +228,11 @@ def post(self):
                 BaseStudy.query.filter_by(**filter_params)
                 .options(
                     joinedload(BaseStudy.versions)
-                    .joinedload(Study.studyset_studies)
-                    .joinedload(StudysetStudy.studyset)
+                    .options(joinedload(Study.studyset_studies)
+                             .joinedload(StudysetStudy.studyset),
+                             joinedload(Study.user),
+                    ),
+                    joinedload(BaseStudy.user)
                 )
                 .one_or_none()
             )