From 44395576bc509471c7804ba28f37a9865b3eb2f5 Mon Sep 17 00:00:00 2001
From: airflow <airflow@example.com>
Date: Wed, 24 Sep 2025 06:45:53 +0400
Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=B1=D0=BD=D0=BE=D0=B2=D0=B8=D1=82?=
 =?UTF-8?q?=D1=8C=20test=5Fclickhouse=5Fdag.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 test_clickhouse_dag.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/test_clickhouse_dag.py b/test_clickhouse_dag.py
index 7852421..4fceccf 100644
--- a/test_clickhouse_dag.py
+++ b/test_clickhouse_dag.py
@@ -1,11 +1,15 @@
 from airflow import DAG
-from airflow.providers.clickhouse.hooks.clickhouse import ClickHouseHook
 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")
-    hook.run("SELECT 1")
+    hook = ClickHouseHook(
+        clickhouse_conn_id="my_clickhouse_provider"  # ваш Connection ID в Airflow
+    )
+    hook.execute("SELECT 1")  # у этого пакета метод называется execute
 
 with DAG(
     dag_id="test_clickhouse_dag",
@@ -13,4 +17,7 @@ with DAG(
     schedule=None,
     catchup=False,
 ) as dag:
-    PythonOperator(task_id="check_clickhouse", python_callable=test_conn)
+    PythonOperator(
+        task_id="check_clickhouse",
+        python_callable=test_conn,
+    )