55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
"""Entity classes for the Airzone Cloud integration."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from abc import ABC, abstractmethod
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from aioairzone_cloud.const import AZD_NAME, AZD_SYSTEM_ID, AZD_ZONES
|
||
|
|
|
||
|
|
from homeassistant.config_entries import ConfigEntry
|
||
|
|
from homeassistant.helpers.entity import DeviceInfo
|
||
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||
|
|
|
||
|
|
from .const import DOMAIN, MANUFACTURER
|
||
|
|
from .coordinator import AirzoneUpdateCoordinator
|
||
|
|
|
||
|
|
|
||
|
|
class AirzoneEntity(CoordinatorEntity[AirzoneUpdateCoordinator], ABC):
|
||
|
|
"""Define an Airzone Cloud entity."""
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def get_airzone_value(self, key: str) -> Any:
|
||
|
|
"""Return Airzone Cloud entity value by key."""
|
||
|
|
|
||
|
|
|
||
|
|
class AirzoneZoneEntity(AirzoneEntity):
|
||
|
|
"""Define an Airzone Cloud Zone entity."""
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
coordinator: AirzoneUpdateCoordinator,
|
||
|
|
entry: ConfigEntry,
|
||
|
|
zone_id: str,
|
||
|
|
zone_data: dict[str, Any],
|
||
|
|
) -> None:
|
||
|
|
"""Initialize."""
|
||
|
|
super().__init__(coordinator)
|
||
|
|
|
||
|
|
self.system_id = zone_data[AZD_SYSTEM_ID]
|
||
|
|
self.zone_id = zone_id
|
||
|
|
|
||
|
|
self._attr_device_info = DeviceInfo(
|
||
|
|
identifiers={(DOMAIN, f"{entry.unique_id}_{zone_id}")},
|
||
|
|
manufacturer=MANUFACTURER,
|
||
|
|
name=zone_data[AZD_NAME],
|
||
|
|
via_device=(DOMAIN, f"{entry.unique_id}_{self.system_id}"),
|
||
|
|
)
|
||
|
|
|
||
|
|
def get_airzone_value(self, key: str) -> Any:
|
||
|
|
"""Return zone value by key."""
|
||
|
|
value = None
|
||
|
|
if zone := self.coordinator.data[AZD_ZONES].get(self.zone_id):
|
||
|
|
if key in zone:
|
||
|
|
value = zone[key]
|
||
|
|
return value
|