"""MySQL Client for ows-product-digital-marketing testing.."""

from typing import Any

from pymysql import connect

from tests.integration import config


class MySQLClient:
    """MySQL Client for ows-video testing."""

    def __init__(self) -> None:
        """Initialize MySQLClient class."""
        self.db_host = config.AR_DB_HOST
        self.db_user = config.AR_DB_USER
        self.db_password = config.AR_DB_PASSWORD
        self.db_database = config.AR_DB_NAME

    def establish_connection(self) -> Any:
        """Establish art_relations connection."""
        return connect(
            host=self.db_host,
            user=self.db_user,
            password=self.db_password,
            db=self.db_database,
        )

    def execute_query(self, query_string: str, *parameters: object) -> Any:
        """Execute a query."""
        connection = self.establish_connection()
        cursor = connection.cursor()

        try:
            cursor.execute(query_string, parameters)
            result = cursor.fetchall()

        finally:
            cursor.close()
            connection.commit()
            connection.close()

        return result