mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
fix typing problems in mapper
This commit is contained in:
parent
079f7e3397
commit
b57453b8db
1 changed files with 5 additions and 4 deletions
|
|
@ -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))
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue