"""Check the various connection used in the application"""
import sys
import traceback

import redis
import MySQLdb

from qc_app.db_utils import database_connection

def check_redis(uri: str):
    "Check the redis connection"
    try:
        with redis.Redis.from_url(uri) as rconn:
            rconn.ping()
    except redis.exceptions.ConnectionError as conn_err:
        print(conn_err, file=sys.stderr)
        print(traceback.format_exc(), file=sys.stderr)
        sys.exit(1)

def check_db(uri: str):
    "Check the mysql connection"
    try:
        with database_connection(uri) as dbconn: # pylint: disable=[unused-variable]
            pass
    except MySQLdb.OperationalError as op_err:
        print(op_err, file=sys.stderr)
        print(traceback.format_exc(), file=sys.stderr)
        sys.exit(1)