From c0fb1e6001a0d059f71032279e6c6ba590d35b1a Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 19 May 2025 16:52:05 -0500
Subject: Handle generic exception occurring while using the connection

Rollback any transaction if any exception occurs before the connection
is closed.
---
 gn_libs/mysqldb.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'gn_libs/mysqldb.py')

diff --git a/gn_libs/mysqldb.py b/gn_libs/mysqldb.py
index eabd1e1..2fd0f00 100644
--- a/gn_libs/mysqldb.py
+++ b/gn_libs/mysqldb.py
@@ -132,6 +132,10 @@ def database_connection(sql_uri: str, logger: logging.Logger = _logger) -> Itera
         logger.error("DB error encountered", exc_info=True)
         connection.rollback()
         raise _mbde from None
+    except:
+        logger.error("General exception encountered", exc_info=True)
+        connection.rollback()
+        raise _exc from None
     finally:
         connection.close()
 
-- 
cgit v1.2.3