airflow_odbc_sql_execute_query_operator_example_dag.py

1import os
2from datetime import datetime
3
4from airflow import DAG
5from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator
6
7# The following environment variables are expected to be set for the connection
8ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
9DAG_ID = "example_odbc"
10
11with DAG(
12    dag_id=DAG_ID,
13    schedule=None,
14    start_date=datetime(2021, 1, 1),
15    catchup=False,
16    tags=["example"],
17) as dag:
18
19    # [START howto_operator_odbc]
20    # To use the ODBC operator, you must first configure an ODBC Connection in Airflow.
21    # The connection should specify the ODBC driver and other necessary parameters.
22    odbc_task = SQLExecuteQueryOperator(
23        task_id="odbc_task",
24        conn_id="odbc_default",
25        sql="SELECT 1",
26    )
27    # [END howto_operator_odbc]
28
29    odbc_task