#34: Add _get_table_to_filter method

ral-facilities · Aug 22, 2019 · ac2b923 · ac2b923
1 parent c8e85c4
commit ac2b923
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/common/database_helpers.py b/common/database_helpers.py
@@ -133,6 +133,12 @@ def apply_filter(self, query):
         query.base_query = query.base_query.filter(getattr(query.table, self.field) == self.value)
 
 
+    def _get_table_to_filter(self, query):
+        if type(query) is ISISInvestigationsQuery:
+            return INVESTIGATION
+        else:
+            return query.table
+
 class OrderFilter(QueryFilter):
     precedence = 1