#34: Use get investigations count in resource

ral-facilities · Aug 22, 2019 · ccdec3b · ccdec3b
1 parent 3261184
commit ccdec3b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/resources/table_endpoints/table_endpoints.py b/src/resources/table_endpoints/table_endpoints.py
@@ -1,6 +1,6 @@
 from flask_restful import Resource
 
-from common.database_helpers import get_investigations_for_user
+from common.database_helpers import get_investigations_for_user, get_investigations_for_user_count
 from common.helpers import requires_session_id, queries_records, get_filters_from_query_string
 
 
@@ -12,7 +12,10 @@ def get(self, id):
 
 
 class UsersInvestigationsCount(Resource):
-    pass
+    @requires_session_id
+    @queries_records
+    def get(self, id):
+        return get_investigations_for_user_count(id, get_filters_from_query_string()), 200
 
 
 class InstrumentsFacilityCycles(Resource):