Files
midea-meiju-codec/custom_components/midea_meiju_codec/switch.py
unknown 10d127c940 v0.0.3
2023-09-09 00:14:41 +08:00

36 lines
1.1 KiB
Python

from homeassistant.components.switch import SwitchEntity
from homeassistant.const import (
Platform,
CONF_DEVICE_ID,
CONF_DEVICE,
CONF_ENTITIES,
STATE_ON,
STATE_OFF
)
from .const import (
DOMAIN,
DEVICES
)
from .midea_entities import MideaBinaryBaseEntity
async def async_setup_entry(hass, config_entry, async_add_entities):
device_id = config_entry.data.get(CONF_DEVICE_ID)
device = hass.data[DOMAIN][DEVICES][device_id].get(CONF_DEVICE)
manufacturer = hass.data[DOMAIN][DEVICES][device_id].get("manufacturer")
entities = hass.data[DOMAIN][DEVICES][device_id].get(CONF_ENTITIES).get(Platform.SWITCH)
devs = []
if entities is not None:
for entity_key, config in entities.items():
devs.append(MideaSwitchEntity(device, manufacturer, entity_key, config))
async_add_entities(devs)
class MideaSwitchEntity(MideaBinaryBaseEntity, SwitchEntity):
def turn_on(self):
self._device.set_attribute(attribute=self._entity_key, value=self._binary_rationale[1])
def turn_off(self):
self._device.set_attribute(attribute=self._entity_key, value=self._binary_rationale[0])