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