evol 2

utils/db/connect_db

@@ -0,0 +1,22 @@
+import logging
+import mysql.connector
+
+logger = logging.getLogger(__name__)
+
+def conn_db(cfg):
+    """Establishes a connection to the MySQL database.
+
+    Args:
+        cfg: The configuration object containing database connection details.
+
+    Returns:
+        A MySQL database connection object.
+    """
+    try:
+        conn = mysql.connector.connect(user=cfg.dbuser, password=cfg.dbpass, host=cfg.dbhost, port=cfg.dbport)
+        conn.autocommit = True
+        return conn
+    except mysql.connector.Error as e:
+        logger.error(f'{e}')
+        exit(e.errno)
+    return None