"""Subclass of existing airflow mysql hook to connect to art_relations."""

from airflow.providers.mysql.hooks.mysql import MySqlHook

from lib.config import AR_MYSQL_CONN_NAME


class ArtRelationsMySqlHook(MySqlHook):
    """MySQL hook for connecting to art_relations."""

    def __init__(self, *args, **kwargs):
        """Override MySqlHook.__init__ to set default values."""
        kwargs.setdefault('mysql_conn_id', AR_MYSQL_CONN_NAME)

        super().__init__(*args, **kwargs)