Обновить 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.utils.dates import days_ago
+from airflow.operators.python import PythonOperator
+from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook
+from datetime import datetime
 
-default_args = {
-    "owner": "airflow",
-}
+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_select",
-    default_args=default_args,
-    description="Simple SELECT 1 in ClickHouse",
-    start_date=days_ago(1),
-    schedule_interval=None,      # запуск только вручную
+    dag_id="test_clickhouse_dag",
+    start_date=datetime(2024, 1, 1),
+    schedule=None,
     catchup=False,
-    tags=["clickhouse"],
+    tags=["example"],
 ) as dag:
-
-    select_1 = ClickHouseOperator(
-        task_id="select_one",
-        clickhouse_conn_id="my_clickhouse_provider",  # ваш Connection ID
-        sql="SELECT 1",
+    test_task = PythonOperator(
+        task_id="check_clickhouse",
+        python_callable=test_clickhouse
     )