Обновить postgre-localhost-demo.py

This commit is contained in:
2025-10-22 07:00:13 +04:00
parent 390d67866b
commit 1ada4a8fef

View File

@@ -28,9 +28,7 @@ def save_query_to_csv(**kwargs):
from airflow.providers.postgres.hooks.postgres import PostgresHook
pg_hook = PostgresHook(postgres_conn_id='pg-localhost-demo')
sql = "SELECT * FROM bookings.segments s
join bookings.flights f on s.flight_id = f.flight_id
limit 100"
sql = "SELECT * FROM bookings.segments s join bookings.flights f on s.flight_id = f.flight_id limit 100"
records = pg_hook.get_records(sql)
columns = [desc[0] for desc in pg_hook.get_conn().cursor().description]