revert accidental changes

This commit is contained in:
Daniel Roth 2026-06-09 12:24:12 +00:00
parent 31c2e07a02
commit ade6891ff0

View file

@ -32,15 +32,11 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
self._session = session self._session = session
def get_plan_by_uploaded_file_id(self, uploaded_file_id: int) -> Optional[Plan]: def get_plan_by_uploaded_file_id(self, uploaded_file_id: int) -> Optional[Plan]:
plan_row = ( plan_row = self._session.execute( # pyright: ignore[reportDeprecated]
self._session.execute( # pyright: ignore[reportDeprecated] select(MagicPlanPlanModel).where(
select(MagicPlanPlanModel).where( col(MagicPlanPlanModel.uploaded_file_id) == uploaded_file_id
col(MagicPlanPlanModel.uploaded_file_id) == uploaded_file_id
)
) )
.scalars() ).scalars().one_or_none()
.one_or_none()
)
if plan_row is None: if plan_row is None:
return None return None
@ -51,9 +47,7 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
select(MagicPlanFloorModel).where( select(MagicPlanFloorModel).where(
col(MagicPlanFloorModel.magic_plan_plan_id) == plan_id col(MagicPlanFloorModel.magic_plan_plan_id) == plan_id
) )
) ).scalars().all()
.scalars()
.all()
) )
floor_ids = [cast(int, f.id) for f in floor_rows] floor_ids = [cast(int, f.id) for f in floor_rows]
@ -62,9 +56,7 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
select(MagicPlanRoomModel).where( select(MagicPlanRoomModel).where(
col(MagicPlanRoomModel.magic_plan_floor_id).in_(floor_ids) col(MagicPlanRoomModel.magic_plan_floor_id).in_(floor_ids)
) )
) ).scalars().all()
.scalars()
.all()
) )
room_ids = [cast(int, r.id) for r in room_rows] room_ids = [cast(int, r.id) for r in room_rows]
@ -73,9 +65,7 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
select(MagicPlanWindowModel).where( select(MagicPlanWindowModel).where(
col(MagicPlanWindowModel.magic_plan_room_id).in_(room_ids) col(MagicPlanWindowModel.magic_plan_room_id).in_(room_ids)
) )
) ).scalars().all()
.scalars()
.all()
) )
window_ids = [cast(int, w.id) for w in window_rows] window_ids = [cast(int, w.id) for w in window_rows]
@ -84,9 +74,7 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
select(MagicPlanDoorModel).where( select(MagicPlanDoorModel).where(
col(MagicPlanDoorModel.magic_plan_room_id).in_(room_ids) col(MagicPlanDoorModel.magic_plan_room_id).in_(room_ids)
) )
) ).scalars().all()
.scalars()
.all()
) )
door_ids = [cast(int, d.id) for d in door_rows] door_ids = [cast(int, d.id) for d in door_rows]
@ -98,9 +86,7 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
window_ids window_ids
) )
) )
) ).scalars().all()
.scalars()
.all()
} }
door_vent_by_door_id = { door_vent_by_door_id = {
cast(int, dv.magic_plan_door_id): dv cast(int, dv.magic_plan_door_id): dv
@ -108,9 +94,7 @@ class MagicPlanPostgresRepository(MagicPlanRepository):
select(MagicPlanDoorVentilationModel).where( select(MagicPlanDoorVentilationModel).where(
col(MagicPlanDoorVentilationModel.magic_plan_door_id).in_(door_ids) col(MagicPlanDoorVentilationModel.magic_plan_door_id).in_(door_ids)
) )
) ).scalars().all()
.scalars()
.all()
} }
windows_by_room_id: dict[int, list[Window]] = {} windows_by_room_id: dict[int, list[Window]] = {}