From 6ab6d46ab4b1611ed72bdbce85cf9324ce69b305 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 7 Aug 2023 07:47:01 +0300
Subject: Collect db-connections function in single module.

---
 gn_auth/auth/db/mariadb.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 gn_auth/auth/db/mariadb.py

(limited to 'gn_auth/auth/db/mariadb.py')

diff --git a/gn_auth/auth/db/mariadb.py b/gn_auth/auth/db/mariadb.py
new file mode 100644
index 0000000..a934fd9
--- /dev/null
+++ b/gn_auth/auth/db/mariadb.py
@@ -0,0 +1,26 @@
+"""Connections to MariaDB"""
+import traceback
+import contextlib
+from typing import Iterator
+
+import MySQLdb as mdb
+
+from .protocols import DbConnection
+
+@contextlib.contextmanager
+def database_connection(sql_uri) -> Iterator[DbConnection]:
+    """Connect to MySQL database."""
+    host, user, passwd, db_name, port = parse_db_url(sql_uri)
+    connection = mdb.connect(db=db_name,
+                             user=user,
+                             passwd=passwd or '',
+                             host=host,
+                             port=port or 3306)
+    try:
+        yield connection
+    except Exception as _exc: # TODO: Make the Exception class less general
+        logging.debug(traceback.format_exc())
+        connection.rollback()
+    finally:
+        connection.commit()
+        connection.close()
-- 
cgit v1.2.3