airflow_dag_odbc_hook_sql_query_execution.py

1from datetime import datetime
2from airflow import DAG
3from airflow.decorators import task
4from airflow.providers.odbc.hooks.odbc import OdbcHook
5
6with DAG(
7    dag_id='example_odbc_query',
8    schedule_interval=None,
9    start_date=datetime(2023, 1, 1),
10    catchup=False,
11    tags=['example'],
12) as dag:
13
14    @task(task_id="execute_odbc_query")
15    def execute_query():
16        """
17        Example task that uses OdbcHook to execute a query.
18        The 'odbc_conn_id' should be configured in Airflow Connections.
19        """
20        odbc_hook = OdbcHook(odbc_conn_id='odbc_default')
21        
22        # SQL to execute
23        sql = "SELECT 1"
24        
25        # Execute the query and fetch results
26        results = odbc_hook.get_records(sql)
27        
28        for row in results:
29            print(f"Result: {row}")
30            
31        return results
32
33    execute_query()