Create LimitFilter

ral-facilities · Jul 29, 2019 · a0b18a8 · a0b18a8
1 parent 67bf964
commit a0b18a8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/common/database_helpers.py b/common/database_helpers.py
@@ -144,6 +144,15 @@ def apply_filter(self, query):
             raise BadFilterError(f" Bad filter: {self.direction}")
 
 
+class LimitFilter(QueryFilter):
+    def __init__(self, limit_value):
+        self.limit_value = limit_value
+
+    def apply_filter(self, query):
+        query.base_query = query.base_query.limit(self.limit_value)
+        query.is_limited = True
+
+
 def insert_row_into_table(row):
     """
     Insert the given row into its table