from airflow import DAG
from airflow.operators.python import PythonOperator
from datetime import datetime

# Импортируем hook из установленного пакета airflow-providers-clickhouse
from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook

def test_conn():
    hook = ClickHouseHook(
        clickhouse_conn_id="my_clickhouse_provider"  # ваш Connection ID в Airflow
    )
    hook.execute("SELECT 1")  # у этого пакета метод называется execute

with DAG(
    dag_id="test_clickhouse_dag",
    start_date=datetime(2024, 1, 1),
    schedule=None,
    catchup=False,
) as dag:
    PythonOperator(
        task_id="check_clickhouse",
        python_callable=test_conn,
    )