fix typing problems in mapper

This commit is contained in:
Daniel Roth 2026-05-06 08:24:57 +00:00
parent 079f7e3397
commit b57453b8db

View file

@ -1,3 +1,4 @@
import datatypes.magicplan.api.response as api
from datatypes.magicplan.api.response import MagicPlan
from datatypes.magicplan.domain.models import Plan, Floor, Room, Window, Door
@ -10,7 +11,7 @@ def map_plan(mp: MagicPlan) -> Plan:
)
def _map_floor(f) -> Floor:
def _map_floor(f: api.Floor) -> Floor:
return Floor(
level=f.level,
name=f.name,
@ -18,7 +19,7 @@ def _map_floor(f) -> Floor:
)
def _map_room(r) -> Room:
def _map_room(r: api.Room) -> Room:
width, length = _parse_dimensions(r.dimensions)
return Room(
name=r.name,
@ -39,7 +40,7 @@ def _parse_dimensions(dimensions: str | None) -> tuple[float, float]:
return width, length
def _map_window(wi) -> Window:
def _map_window(wi: api.WallItem) -> Window:
return Window(
width=round(wi.size.x, 2),
height=round(wi.size.z, 2),
@ -48,5 +49,5 @@ def _map_window(wi) -> Window:
)
def _map_door(wi) -> Door:
def _map_door(wi: api.WallItem) -> Door:
return Door(width=round(wi.size.x, 2))