From 0410f6e90b80d4aa8427ea3c398fe70e1977043c Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Wed, 13 Jul 2022 01:19:57 +0300
Subject: Add missing db_utils module

---
 qc_app/db_utils.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 qc_app/db_utils.py

(limited to 'qc_app')

diff --git a/qc_app/db_utils.py b/qc_app/db_utils.py
new file mode 100644
index 0000000..95b0057
--- /dev/null
+++ b/qc_app/db_utils.py
@@ -0,0 +1,20 @@
+"""module contains all db related stuff"""
+from typing import Tuple
+
+from urllib.parse import urlparse
+import MySQLdb as mdb
+from flask import current_app as app
+
+def parse_db_url() -> Tuple:
+    """
+    Parse SQL_URI configuration variable.
+    """
+    parsed_db = urlparse(app.config["SQL_URI"])
+    return (parsed_db.hostname, parsed_db.username,
+            parsed_db.password, parsed_db.path[1:])
+
+
+def database_connection() -> mdb.Connection:
+    """function to create db connector"""
+    host, user, passwd, db_name = parse_db_url()
+    return mdb.connect(host, user, passwd, db_name)
-- 
cgit v1.2.3