from Airflow.operators.sql_operator import SQLGetHook from airflow.models import DAG from airflow.operators.python_operator import PythonOperator
dag = DAG(dag_id='example_sql', default_args=default_args, schedule_interval=None)
def get_data_from_mysql(): sql = "SELECT * FROM my_table" hook = SQLGetHook(mysql_conn_id='mysql_conn') results = hook.get_records(sql)
for result in results:
print(result)
get_data = PythonOperator( task_id='get_data', python_callable=get_data_from_mysql, dag=dag )
from airflow.operators.mssql_operator import MsSqlOperator from airflow.models import DAG
dag = DAG(dag_id='example_mssql', default_args=default_args, schedule_interval=None)
get_data = MsSqlOperator( task_id='get_data', sql="SELECT * FROM my_table", mssql_conn_id='mssql_conn', database='my_db', dag=dag )
get_data