From 64993faf71899a75afff608eb8981399e1d3f039 Mon Sep 17 00:00:00 2001
From: airflow <airflow@example.com>
Date: Wed, 24 Sep 2025 03:55:33 +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=20clickhouse=5Fdag=5Fnew.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 clickhouse_dag_new.py | 32 +++++++++++++++++++++-----------
 1 file changed, 21 insertions(+), 11 deletions(-)

diff --git a/clickhouse_dag_new.py b/clickhouse_dag_new.py
index 10b1f36..817bf82 100644
--- a/clickhouse_dag_new.py
+++ b/clickhouse_dag_new.py
@@ -1,16 +1,26 @@
 from airflow import DAG
-from airflow.providers.postgres.operators.postgres import PostgresOperator
-from datetime import datetime
+from airflow_clickhouse_plugin.operators.clickhouse import ClickHouseOperator
+from airflow.utils.dates import days_ago
 
+# Аргументы DAG по умолчанию
+default_args = {
+    'owner': 'airflow',
+    'depends_on_past': False,
+    'start_date': days_ago(1),
+}
+
+# Создаем DAG
 with DAG(
-    dag_id="clickhouse_list_tables_1",
-    start_date=datetime(2025, 1, 1),
-    schedule_interval=None,
-    catchup=False,
+    dag_id='test_clickhouse_connection',
+    default_args=default_args,
+    schedule_interval=None,  # Запуск только вручную
+    tags=['test'],
 ) as dag:
 
-    list_tables = PostgresOperator(
-        task_id="list_tables",
-        postgres_conn_id="clickhouse_db",
-        sql="SELECT 1;"
-    )
+    # Задача с использованием ClickHouseOperator
+    test_query = ClickHouseOperator(
+        task_id='run_simple_query',
+        # Простой запрос для проверки соединения
+        sql='SELECT 1 AS test_value',
+        # clickhouse_conn_id='clickhouse_default' можно не указывать, т.к. это значение по умолчанию
+    )
\ No newline at end of file