From 202c009d15d360ac9b4dc73a916af70af8133106 Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Mon, 29 Jul 2019 08:25:35 +0100 Subject: [PATCH] Create WhereFilter --- common/database_helpers.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/common/database_helpers.py b/common/database_helpers.py index cd97199a..32db6ac0 100644 --- a/common/database_helpers.py +++ b/common/database_helpers.py @@ -83,7 +83,7 @@ def execute_query(self): record.CREATE_TIME = datetime.datetime.now() record.MOD_TIME = datetime.datetime.now() record.CREATE_ID = "user" - record.MOD_ID = "user" # These will need changing + record.MOD_ID = "user" # These will need changing self.session.add(record) self.commit_changes() @@ -118,6 +118,16 @@ class QueryFilter(ABC): def apply_filter(self, query): pass + +class WhereFilter(QueryFilter): + def __init__(self, field, value): + self.field = field + self.value = value + + def apply_filter(self, query): + query.base_query = query.base_query.filter(getattr(query.table, self.field) == self.value) + + def insert_row_into_table(row): """ Insert the given row into its table