From 171ad980715586114c49bc26969c15cd049f635f Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Wed, 14 Aug 2019 08:11:29 +0100 Subject: [PATCH] #18: Update get_row_by_id --- common/database_helpers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/common/database_helpers.py b/common/database_helpers.py index 2bff8ad1..7e623aa6 100644 --- a/common/database_helpers.py +++ b/common/database_helpers.py @@ -131,7 +131,7 @@ def __init__(self, field, value, operation): def apply_filter(self, query): if self.operation == "eq": - query.base_query = query.base_query.filter(getattr(query.table, self.field) == self.value) + query.base_query = query.base_query.filter(getattr(query.table, self.field) == self.value) elif self.operation == "like": query.base_query = query.base_query.filter(getattr(query.table, self.field).like(f"%{self.value}%")) else: @@ -241,6 +241,7 @@ def apply_filters(self, query): def insert_row_into_table(table, row): """ Insert the given row into its table + :param table: The table to be inserted to :param row: The row to be inserted """ create_query = CreateQuery(table, row) @@ -268,7 +269,7 @@ def get_row_by_id(table, id): read_query = ReadQuery(table) try: log.info(f" Querying {table.__tablename__} for record with ID: {id}") - where_filter = WhereFilter("ID", id) + where_filter = WhereFilter("ID", id, "eq") where_filter.apply_filter(read_query) return read_query.get_single_result() finally: