"""MySQL Client for ows-product-digital-marketing testing.."""
from os import getenv

from pymysql import connect


class MySQLClient:
    """MySQL Client for ows-product-digital-marketing testing."""

    def __init__(self):
        """Initialize MySQLClient class."""
        self.db_host = getenv('DB_CLIENT_HOST')
        self.db_user = getenv('DB_CLIENT_USER')
        self.db_password = getenv('DB_CLIENT_PASS')
        self.db_database = getenv('DB_CLIENT_DB')

    def execute_query(self, query_string, parameters):
        """Execute a query."""
        connection = connect(host=self.db_host, user=self.db_user,
                             password=self.db_password, db=self.db_database)

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

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

        return result