From af62c1490acf191c32ec4316ce9a2f9b2c0cc530 Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Mon, 29 Jul 2019 08:21:36 +0100 Subject: [PATCH] Create DeleteQuery --- common/database_helpers.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/common/database_helpers.py b/common/database_helpers.py index 3d820d07..f956065c 100644 --- a/common/database_helpers.py +++ b/common/database_helpers.py @@ -101,6 +101,18 @@ def execute_query(self): self.commit_changes() +class DeleteQuery(Query): + + def __init__(self, table, row): + super().__init__(table) + self.row = row + + def execute_query(self): + log.info(f" Deleting row {self.row} from {self.table.__tablename__}") + self.session.delete(self.row) + self.commit_changes() + + def insert_row_into_table(row): """ Insert the given row into its table