import logging

from mysql.connector import Error, MySQLConnection

from ..utils import CoreConfig
from ..utils import logger as cl


class MySqlClient:
    log = cl.Logger(logging.DEBUG)

    def __init__(
        self,
        host=CoreConfig.MYSQL_DB_HOST,
        database=CoreConfig.MYSQL_DB_NAME,
        user=CoreConfig.MYSQL_DB_USER,
        password=CoreConfig.MYSQL_DB_PASS,
        port=CoreConfig.MYSQL_DB_PORT,
    ):
        self.host = host
        self.database = database
        self.user = user
        self.password = password
        self.port = port

    def connect(self):
        try:
            self.log.info("Connecting to MySQL database...")
            conn = MySQLConnection(
                host=self.host, database=self.database, user=self.user, port=self.port, password=self.password
            )

            if conn.is_connected():
                self.log.info("connection established.")
                return conn

            self.log.info("connection failed.")
        except Error as error:
            self.log.info(error)