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