rename repo

custom_components/midea_auto_codec/switch.py

@@ -0,0 +1,34 @@
+from homeassistant.components.switch import SwitchEntity
+from homeassistant.const import (
+    Platform,
+    CONF_DEVICE_ID,
+    CONF_DEVICE,
+    CONF_ENTITIES,
+)
+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")
+    rationale = hass.data[DOMAIN][DEVICES][device_id].get("rationale")
+    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, rationale, entity_key, config))
+    async_add_entities(devs)
+
+
+class MideaSwitchEntity(MideaBinaryBaseEntity, SwitchEntity):
+
+    def turn_on(self):
+        self._set_status_on_off(self._entity_key, True)
+
+    def turn_off(self):
+        self._set_status_on_off(self._entity_key, False)