Create IncludeFilter

ral-facilities · Jul 29, 2019 · 7e855ce · 7e855ce
1 parent a0b18a8
commit 7e855ce
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/common/database_helpers.py b/common/database_helpers.py
@@ -153,6 +153,14 @@ def apply_filter(self, query):
         query.is_limited = True
 
 
+class IncludeFilter(QueryFilter):
+    def __init__(self, included_filters):
+        self.included_filters = included_filters
+
+    def apply_filter(self, query):
+        query.include_related_entities = True
+
+
 def insert_row_into_table(row):
     """
     Insert the given row into its table