diff --git a/common/database_helpers.py b/common/database_helpers.py index 10bb8183..3d820d07 100644 --- a/common/database_helpers.py +++ b/common/database_helpers.py @@ -86,7 +86,20 @@ def execute_query(self): record.MOD_ID = "user" # These will need changing self.session.add(record) self.commit_changes() - + + +class UpdateQuery(Query): + + def __init__(self, table, row, new_values): + super().__init__(table) + self.row = row + self.new_values = new_values + + def execute_query(self): + log.info(f" Updating row in {self.table}") + self.row.update_from_dict(self.new_values) + self.commit_changes() + def insert_row_into_table(row): """