Apache Airflow Common SQL Provider

from airflow import DAG
from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator
from datetime import datetime

with DAG(
    'sql_example_dag',
    start_date=datetime(2023, 1, 1),
    schedule_interval=None,
    catchup=False,
    tags=['sql', 'example']
) as dag:
    # Ensure 'my_database_conn' is configured in Airflow Connections
    execute_query_task = SQLExecuteQueryOperator(
        task_id='run_simple_sql_query',
        conn_id='my_database_conn',
        sql='SELECT 1;'
    )

    # Example of running a SQL query from a file
    # For this to work, you would typically place 'my_query.sql' in your DAGs folder's 'include' directory,
    # or set the template_searchpath in the DAG definition.
    # For demonstration, we'll use a placeholder string.
    execute_templated_query = SQLExecuteQueryOperator(
        task_id='run_templated_sql_query',
        conn_id='my_database_conn',
        sql="""SELECT * FROM users WHERE registration_date < '{{ ds }}';""",
        parameters={'table_name': 'users'}
    )