core/homeassistant/components/elkm1/switch.py

36 lines
1.0 KiB
Python

"""Support for control of ElkM1 outputs (relays)."""
from homeassistant.components.switch import SwitchDevice
from . import DOMAIN as ELK_DOMAIN, ElkEntity, create_elk_entities
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
"""Create the Elk-M1 switch platform."""
if discovery_info is None:
return
elk_datas = hass.data[ELK_DOMAIN]
entities = []
for elk_data in elk_datas.values():
elk = elk_data["elk"]
entities = create_elk_entities(
elk_data, elk.outputs, "output", ElkOutput, entities
)
async_add_entities(entities, True)
class ElkOutput(ElkEntity, SwitchDevice):
"""Elk output as switch."""
@property
def is_on(self) -> bool:
"""Get the current output status."""
return self._element.output_on
async def async_turn_on(self, **kwargs):
"""Turn on the output."""
self._element.turn_on(0)
async def async_turn_off(self, **kwargs):
"""Turn off the output."""
self._element.turn_off()