#34: Add ISISInvestigationsCountQuery

ral-facilities · Aug 22, 2019 · 31677c3 · 31677c3
1 parent 455bfdd
commit 31677c3
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/common/database_helpers.py b/common/database_helpers.py
@@ -440,3 +440,12 @@ def get_investigations_for_user(user_id, filters):
 
     finally:
         query.session.close()
+
+class ISISInvestigationsCountQuery(CountQuery):
+    """
+    The query class used for /users/<:id>/investigations/count
+    """
+
+    def __init__(self, user_id):
+        super().__init__(INVESTIGATIONUSER)
+        self.base_query = self.base_query.join(INVESTIGATION).filter(INVESTIGATIONUSER.USER_ID == user_id)