1 files changed, 46 insertions, 0 deletions
diff --git a/uploader/db_utils.py b/uploader/db_utils.py
new file mode 100644
index 0000000..ef26398
--- /dev/null
+++ b/uploader/db_utils.py
@@ -0,0 +1,46 @@
+"""module contains all db related stuff"""
+import logging
+import traceback
+import contextlib
+from urllib.parse import urlparse
+from typing import Any, Tuple, Optional, Iterator, Callable
+
+import MySQLdb as mdb
+from redis import Redis
+from flask import current_app as app
+
+def parse_db_url(db_url) -> Tuple:
+    """
+    Parse SQL_URI configuration variable.
+    """
+    parsed_db = urlparse(db_url)
+    return (parsed_db.hostname, parsed_db.username,
+            parsed_db.password, parsed_db.path[1:], parsed_db.port)
+
+
+@contextlib.contextmanager
+def database_connection(db_url: Optional[str] = None) -> Iterator[mdb.Connection]:
+    """function to create db connector"""
+    host, user, passwd, db_name, db_port = parse_db_url(
+        db_url or app.config["SQL_URI"])
+    connection = mdb.connect(
+        host, user, passwd, db_name, port=(db_port or 3306))
+    try:
+        yield connection
+        connection.commit()
+    except mdb.Error as _mdb_err:
+        logging.error(traceback.format_exc())
+        connection.rollback()
+    finally:
+        connection.close()
+
+def with_db_connection(func: Callable[[mdb.Connection], Any]) -> Any:
+    """Call `func` with a MySQDdb database connection."""
+    with database_connection(app.config["SQL_URI"]) as conn:
+        return func(conn)
+
+def with_redis_connection(func: Callable[[Redis], Any]) -> Any:
+    """Call `func` with a redis connection."""
+    redisuri = app.config["REDIS_URL"]
+    with Redis.from_url(redisuri, decode_responses=True) as rconn:
+        return func(rconn)