Midea/midea_ac_lan/midea_entity.py

64 lines
1.9 KiB
Python
Raw Normal View History

2023-09-22 05:45:40 +00:00
import logging
from homeassistant.helpers.entity import Entity
from .const import DOMAIN
from .midea_devices import MIDEA_DEVICES
_LOGGER = logging.getLogger(__name__)
class MideaEntity(Entity):
def __init__(self, device, entity_key: str):
self._device = device
self._device.register_update(self.update_state)
self._config = MIDEA_DEVICES[self._device.device_type]["entities"][entity_key]
self._entity_key = entity_key
self._unique_id = f"{DOMAIN}.{self._device.device_id}_{entity_key}"
self.entity_id = self._unique_id
self._device_name = self._device.name
@property
def device(self):
return self._device
@property
def device_info(self):
return {
"manufacturer": "Midea",
"model": f"{MIDEA_DEVICES[self._device.device_type]['name']} "
f"{self._device.model}",
"identifiers": {(DOMAIN, self._device.device_id)},
"name": self._device_name
}
@property
def unique_id(self):
return self._unique_id
@property
def should_poll(self):
return False
@property
def state(self):
return self._device.get_attribute(self._entity_key)
@property
def name(self):
return f"{self._device_name} {self._config.get('name')}" if "name" in self._config \
else self._device_name
@property
def available(self):
return self._device.available
@property
def icon(self):
return self._config.get("icon")
def update_state(self, status):
if self._entity_key in status or "available" in status:
try:
self.schedule_update_ha_state()
except Exception as e:
_LOGGER.debug(f"Entity {self.entity_id} update_state {repr(e)}, status = {status}")