diff --git a/test_clickhouse_dag.py b/test_clickhouse_dag.py index 538247e..7e5ba8a 100644 --- a/test_clickhouse_dag.py +++ b/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 )