mirror of https://github.com/home-assistant/core
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
"""Support for IHC switches."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from ihcsdk.ihccontroller import IHCController
|
|
|
|
from homeassistant.components.switch import SwitchEntity
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from .const import CONF_OFF_ID, CONF_ON_ID, DOMAIN, IHC_CONTROLLER
|
|
from .entity import IHCEntity
|
|
from .util import async_pulse, async_set_bool
|
|
|
|
|
|
def setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the IHC switch platform."""
|
|
if discovery_info is None:
|
|
return
|
|
devices = []
|
|
for name, device in discovery_info.items():
|
|
ihc_id = device["ihc_id"]
|
|
product_cfg = device["product_cfg"]
|
|
product = device["product"]
|
|
# Find controller that corresponds with device id
|
|
controller_id = device["ctrl_id"]
|
|
ihc_controller: IHCController = hass.data[DOMAIN][controller_id][IHC_CONTROLLER]
|
|
ihc_off_id = product_cfg.get(CONF_OFF_ID)
|
|
ihc_on_id = product_cfg.get(CONF_ON_ID)
|
|
|
|
switch = IHCSwitch(
|
|
ihc_controller, controller_id, name, ihc_id, ihc_off_id, ihc_on_id, product
|
|
)
|
|
devices.append(switch)
|
|
add_entities(devices)
|
|
|
|
|
|
class IHCSwitch(IHCEntity, SwitchEntity):
|
|
"""Representation of an IHC switch."""
|
|
|
|
def __init__(
|
|
self,
|
|
ihc_controller: IHCController,
|
|
controller_id: str,
|
|
name: str,
|
|
ihc_id: int,
|
|
ihc_off_id: int,
|
|
ihc_on_id: int,
|
|
product=None,
|
|
) -> None:
|
|
"""Initialize the IHC switch."""
|
|
super().__init__(ihc_controller, controller_id, name, ihc_id, product)
|
|
self._ihc_off_id = ihc_off_id
|
|
self._ihc_on_id = ihc_on_id
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn the switch on."""
|
|
if self._ihc_on_id:
|
|
await async_pulse(self.hass, self.ihc_controller, self._ihc_on_id)
|
|
else:
|
|
await async_set_bool(self.hass, self.ihc_controller, self.ihc_id, True)
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn the device off."""
|
|
if self._ihc_off_id:
|
|
await async_pulse(self.hass, self.ihc_controller, self._ihc_off_id)
|
|
else:
|
|
await async_set_bool(self.hass, self.ihc_controller, self.ihc_id, False)
|
|
|
|
def on_ihc_change(self, ihc_id, value):
|
|
"""Handle IHC resource change."""
|
|
self._attr_is_on = value
|
|
self.schedule_update_ha_state()
|