Обновить test_clickhouse_dag.py

test_clickhouse_dag.py

@@ -1,23 +1,22 @@
 from airflow import DAG
-from airflow.providers.clickhouse.operators.clickhouse import ClickHouseOperator
+from airflow.operators.python import PythonOperator
-from airflow.utils.dates import days_ago
+from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook
+from datetime import datetime
 
-default_args = {
+def test_clickhouse():
-    "owner": "airflow",
+    hook = ClickHouseHook(clickhouse_conn_id="my_clickhouse_provider")
-}
+    # Простейший запрос
+    result = hook.get_records("SELECT 1")
+    print("ClickHouse result:", result)
 
 with DAG(
-    dag_id="test_clickhouse_select",
+    dag_id="test_clickhouse_dag",
-    default_args=default_args,
+    start_date=datetime(2024, 1, 1),
-    description="Simple SELECT 1 in ClickHouse",
+    schedule=None,
-    start_date=days_ago(1),
-    schedule_interval=None,      # запуск только вручную
     catchup=False,
-    tags=["clickhouse"],
+    tags=["example"],
 ) as dag:
-
+    test_task = PythonOperator(
-    select_1 = ClickHouseOperator(
+        task_id="check_clickhouse",
-        task_id="select_one",
+        python_callable=test_clickhouse
-        clickhouse_conn_id="my_clickhouse_provider",  # ваш Connection ID
-        sql="SELECT 1",
     )