From 1533e62e7046915c4b925f8e383309c3c4fde9e9 Mon Sep 17 00:00:00 2001
From: airflow <airflow@example.com>
Date: Wed, 24 Sep 2025 06:50:12 +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 | 32 ++++++++++++++++----------------
 1 file changed, 16 insertions(+), 16 deletions(-)

diff --git a/test_clickhouse_dag.py b/test_clickhouse_dag.py
index 4fceccf..538247e 100644
--- a/test_clickhouse_dag.py
+++ b/test_clickhouse_dag.py
@@ -1,23 +1,23 @@
 from airflow import DAG
-from airflow.operators.python import PythonOperator
-from datetime import datetime
+from airflow.providers.clickhouse.operators.clickhouse import ClickHouseOperator
+from airflow.utils.dates import days_ago
 
-# Импортируем 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
+default_args = {
+    "owner": "airflow",
+}
 
 with DAG(
-    dag_id="test_clickhouse_dag",
-    start_date=datetime(2024, 1, 1),
-    schedule=None,
+    dag_id="test_clickhouse_select",
+    default_args=default_args,
+    description="Simple SELECT 1 in ClickHouse",
+    start_date=days_ago(1),
+    schedule_interval=None,      # запуск только вручную
     catchup=False,
+    tags=["clickhouse"],
 ) as dag:
-    PythonOperator(
-        task_id="check_clickhouse",
-        python_callable=test_conn,
+
+    select_1 = ClickHouseOperator(
+        task_id="select_one",
+        clickhouse_conn_id="my_clickhouse_provider",  # ваш Connection ID
+        sql="SELECT 1",
     )