forked from HomeAssistant/midea-meiju-codec
96 lines
3.3 KiB
Python
96 lines
3.3 KiB
Python
from enum import IntEnum
|
|
from homeassistant.helpers.entity import Entity
|
|
from homeassistant.const import (
|
|
STATE_ON,
|
|
STATE_OFF
|
|
)
|
|
from .const import DOMAIN
|
|
from .core.logger import MideaLogger
|
|
|
|
|
|
class Rationale(IntEnum):
|
|
EQUALLY = 0
|
|
GREATER = 1
|
|
LESS = 2
|
|
|
|
class MideaEntity(Entity):
|
|
def __init__(self, device, manufacturer: str | None, entity_key: str, config: dict):
|
|
self._device = device
|
|
self._device.register_update(self.update_state)
|
|
self._entity_key = entity_key
|
|
self._config = config
|
|
self._device_name = self._device.device_name
|
|
self._attr_native_unit_of_measurement = self._config.get("unit")
|
|
self._attr_device_class = self._config.get("device_class")
|
|
self._attr_state_class = self._config.get("state_class")
|
|
self._attr_unit_of_measurement = self._config.get("unit")
|
|
self._attr_icon = self._config.get("icon")
|
|
self._attr_unique_id = f"{DOMAIN}.{self._device.device_id}_{self._entity_key}"
|
|
MideaLogger.debug(self._attr_unique_id)
|
|
self._attr_device_info = {
|
|
"manufacturer": "Midea" if manufacturer is None else manufacturer,
|
|
"model": f"{self._device.model}",
|
|
"identifiers": {(DOMAIN, self._device.device_id)},
|
|
"name": self._device_name
|
|
}
|
|
name = self._config.get("name")
|
|
if name is None:
|
|
name = self._entity_key.replace("_", " ").title()
|
|
self._attr_name = f"{self._device_name} {name}"
|
|
self.entity_id = self._attr_unique_id
|
|
|
|
@property
|
|
def should_poll(self):
|
|
return False
|
|
|
|
@property
|
|
def state(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def available(self):
|
|
return self._device.connected
|
|
|
|
def get_mode(self, key_of_modes, rationale: Rationale = Rationale.EQUALLY):
|
|
for mode, status in key_of_modes.items():
|
|
match = True
|
|
for attr, value in status.items():
|
|
state_value = self._device.get_attribute(attr)
|
|
if state_value is None:
|
|
match = False
|
|
break
|
|
if rationale is Rationale.EQUALLY and state_value != value:
|
|
match = False
|
|
break
|
|
if rationale is Rationale.GREATER and state_value < value:
|
|
match = False
|
|
break
|
|
if rationale is Rationale.LESS and state_value > value:
|
|
match = False
|
|
break
|
|
if match:
|
|
return mode
|
|
return None
|
|
|
|
def update_state(self, status):
|
|
if self._entity_key in status or "connected" in status:
|
|
|
|
try:
|
|
self.schedule_update_ha_state()
|
|
except Exception as e:
|
|
pass
|
|
|
|
|
|
class MideaBinaryBaseEntity(MideaEntity):
|
|
def __init__(self, device, manufacturer: str | None, entity_key: str, config: dict):
|
|
super().__init__(device, manufacturer, entity_key, config)
|
|
binary_rationale = config.get("binary_rationale")
|
|
self._binary_rationale = binary_rationale if binary_rationale is not None else ["off", "on"]
|
|
|
|
@property
|
|
def state(self):
|
|
return STATE_ON if self.is_on else STATE_OFF
|
|
|
|
@property
|
|
def is_on(self):
|
|
return self._device.get_attribute(self._entity_key) == self._binary_rationale[1] |