#34: Add get investigation count function

common/database_helpers.py

@@ -441,6 +441,7 @@ def get_investigations_for_user(user_id, filters):
     finally:
         query.session.close()
 
+
 class ISISInvestigationsCountQuery(CountQuery):
     """
     The query class used for /users/<:id>/investigations/count
@@ -449,3 +450,15 @@ class ISISInvestigationsCountQuery(CountQuery):
     def __init__(self, user_id):
         super().__init__(INVESTIGATIONUSER)
         self.base_query = self.base_query.join(INVESTIGATION).filter(INVESTIGATIONUSER.USER_ID == user_id)
+
+
+def get_investigations_for_user_count(user_id, filters):
+    count_query = ISISInvestigationsCountQuery(user_id)
+    filter_handler = FilterOrderHandler()
+    for query_filter in filters:
+        if len(query_filter) == 0:
+            pass
+        else:
+            filter_handler.add_filter(QueryFilterFactory.get_query_filter(query_filter))
+    filter_handler.apply_filters(count_query)
+    return count_query.get_count()