Naprawdę zmagałem się z tym przez ostatnie 90 minut, oto bardziej deklaratywny sposób naśladowania dla nowicjuszy:
from airflow.hooks.mysql_hook import MySqlHook
def fetch_records():
request = "SELECT * FROM your_table"
mysql_hook = MySqlHook(mysql_conn_id = 'the_connection_name_sourced_from_the_ui', schema = 'specific_db')
connection = mysql_hook.get_conn()
cursor = connection.cursor()
cursor.execute(request)
sources = cursor.fetchall()
print(sources)
...your DAG() as dag: code
task = PythonOperator(
task_id = 'fetch_records',
python_callable = fetch_records
)
Spowoduje to zwrócenie do dziennika zawartości zapytania do bazy danych.
Mam nadzieję, że przyda się to komuś innemu.