From ac2b92316477a9cfe2f93d88e12f92564dbe91df Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Thu, 22 Aug 2019 08:00:46 +0100 Subject: [PATCH] #34: Add _get_table_to_filter method --- common/database_helpers.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/common/database_helpers.py b/common/database_helpers.py index 99656fb3..5ec4fd91 100644 --- 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