Handle generic exception occurring while using the connection

Rollback any transaction if any exception occurs before the connection
is closed.

1 files changed, 4 insertions, 0 deletions

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()