From d5d0c4718a49c5a9d26a069c5d5964d58bb324f9 Mon Sep 17 00:00:00 2001 From: Frederick Muriuki Muriithi Date: Fri, 18 Aug 2023 14:50:01 +0300 Subject: Commit on success, rollback on error. --- gn3/db_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/gn3/db_utils.py b/gn3/db_utils.py index 2f65a33..5fd250d 100644 --- a/gn3/db_utils.py +++ b/gn3/db_utils.py @@ -35,7 +35,10 @@ def database_connection(sql_uri) -> Iterator[Connection]: port=port or 3306) try: yield connection + except mdb.Error as _mbde: + connection.rollback() finally: + connection.commit() connection.close() -- cgit v1.2.3