Skip to content

Commit f737769

Browse files
authored
Fab provider SQLA1 -> SQLA2: Fix api_fastapi/core_api/routes/public/assets.py (#57218)
1 parent 7e7f2a2 commit f737769

File tree

1 file changed

+7
-6
lines changed
  • airflow-core/src/airflow/api_fastapi/core_api/routes/public

1 file changed

+7
-6
lines changed

airflow-core/src/airflow/api_fastapi/core_api/routes/public/assets.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,11 @@
1818
from __future__ import annotations
1919

2020
from datetime import datetime
21-
from typing import TYPE_CHECKING, Annotated
21+
from typing import TYPE_CHECKING, Annotated, cast
2222

2323
from fastapi import Depends, HTTPException, status
2424
from sqlalchemy import and_, delete, func, select
25+
from sqlalchemy.engine import CursorResult
2526
from sqlalchemy.orm import joinedload, subqueryload
2627

2728
from airflow._shared.timezones import timezone
@@ -245,7 +246,7 @@ def get_asset_aliases(
245246
)
246247

247248
return AssetAliasCollectionResponse(
248-
asset_aliases=session.scalars(asset_aliases_select),
249+
asset_aliases=list(session.scalars(asset_aliases_select)),
249250
total_entries=total_entries,
250251
)
251252

@@ -336,7 +337,7 @@ def get_asset_events(
336337
assets_events = session.scalars(assets_event_select)
337338

338339
return AssetEventCollectionResponse(
339-
asset_events=assets_events,
340+
asset_events=list(assets_events),
340341
total_entries=total_entries,
341342
)
342343

@@ -611,7 +612,7 @@ def delete_asset_queued_events(
611612
asset_id=asset_id, before=before, permitted_dag_ids=readable_dags_filter.value
612613
)
613614
delete_stmt = delete(AssetDagRunQueue).where(*where_clause).execution_options(synchronize_session="fetch")
614-
result = session.execute(delete_stmt)
615+
result = cast("CursorResult", session.execute(delete_stmt))
615616
if result.rowcount == 0:
616617
raise HTTPException(
617618
status.HTTP_404_NOT_FOUND,
@@ -645,7 +646,7 @@ def delete_dag_asset_queued_events(
645646
)
646647

647648
delete_statement = delete(AssetDagRunQueue).where(*where_clause)
648-
result = session.execute(delete_statement)
649+
result = cast("CursorResult", session.execute(delete_statement))
649650

650651
if result.rowcount == 0:
651652
raise HTTPException(status.HTTP_404_NOT_FOUND, f"Queue event with dag_id: `{dag_id}` was not found")
@@ -680,7 +681,7 @@ def delete_dag_asset_queued_event(
680681
delete_statement = (
681682
delete(AssetDagRunQueue).where(*where_clause).execution_options(synchronize_session="fetch")
682683
)
683-
result = session.execute(delete_statement)
684+
result = cast("CursorResult", session.execute(delete_statement))
684685
if result.rowcount == 0:
685686
raise HTTPException(
686687
status.HTTP_404_NOT_FOUND,

0 commit comments

Comments
 (0)