from airflow import DAG
from airflow.operators.python import PythonOperator
from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook
from datetime import datetime

def test_clickhouse():
    hook = ClickHouseHook(clickhouse_conn_id="my_clickhouse_provider")
    # Простейший запрос
    result = hook.get_records("SELECT 1")
    print("ClickHouse result:", result)

with DAG(
    dag_id="test_clickhouse_dag",
    start_date=datetime(2024, 1, 1),
    schedule=None,
    catchup=False,
    tags=["example"],
) as dag:
    test_task = PythonOperator(
        task_id="check_clickhouse",
        python_callable=test_clickhouse
    )