63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
"""Roborock Models."""
|
|
|
|
from dataclasses import dataclass
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from roborock.containers import HomeDataDevice, HomeDataProduct, NetworkInfo
|
|
from roborock.roborock_typing import DeviceProp
|
|
from vacuum_map_parser_base.map_data import MapData
|
|
|
|
|
|
@dataclass
|
|
class RoborockHassDeviceInfo:
|
|
"""A model to describe roborock devices."""
|
|
|
|
device: HomeDataDevice
|
|
network_info: NetworkInfo
|
|
product: HomeDataProduct
|
|
props: DeviceProp
|
|
|
|
def as_dict(self) -> dict[str, dict[str, Any]]:
|
|
"""Turn RoborockHassDeviceInfo into a dictionary."""
|
|
return {
|
|
"device": self.device.as_dict(),
|
|
"network_info": self.network_info.as_dict(),
|
|
"product": self.product.as_dict(),
|
|
"props": self.props.as_dict(),
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class RoborockA01HassDeviceInfo:
|
|
"""A model to describe A01 roborock devices."""
|
|
|
|
device: HomeDataDevice
|
|
product: HomeDataProduct
|
|
|
|
def as_dict(self) -> dict[str, dict[str, Any]]:
|
|
"""Turn RoborockA01HassDeviceInfo into a dictionary."""
|
|
return {
|
|
"device": self.device.as_dict(),
|
|
"product": self.product.as_dict(),
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class RoborockMapInfo:
|
|
"""A model to describe all information about a map we may want."""
|
|
|
|
flag: int
|
|
name: str
|
|
rooms: dict[int, str]
|
|
image: bytes | None
|
|
last_updated: datetime
|
|
map_data: MapData | None
|
|
|
|
@property
|
|
def current_room(self) -> str | None:
|
|
"""Get the currently active room for this map if any."""
|
|
if self.map_data is None or self.map_data.vacuum_room is None:
|
|
return None
|
|
return self.rooms.get(self.map_data.vacuum_room)
|