mirror of
https://github.com/home-assistant/core.git
synced 2026-06-03 22:24:02 +01:00
epenet
d766aae436
Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: frenck <195327+frenck@users.noreply.github.com>
103 lines
3.2 KiB
Python
103 lines
3.2 KiB
Python
"""Support for bypassing Risco alarm zones."""
|
|
|
|
from typing import Any
|
|
|
|
from pyrisco.common import Zone
|
|
|
|
from homeassistant.components.switch import SwitchEntity
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .coordinator import RiscoDataUpdateCoordinator
|
|
from .entity import RiscoCloudZoneEntity, RiscoLocalZoneEntity
|
|
from .models import RiscoConfigEntry
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: RiscoConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up the Risco switch."""
|
|
risco_data = config_entry.runtime_data
|
|
if local_data := risco_data.local_data:
|
|
async_add_entities(
|
|
RiscoLocalSwitch(local_data.system.id, zone_id, zone)
|
|
for zone_id, zone in local_data.system.zones.items()
|
|
)
|
|
elif cloud_data := risco_data.cloud_data:
|
|
coordinator = cloud_data.coordinator
|
|
async_add_entities(
|
|
RiscoCloudSwitch(coordinator, zone_id, zone)
|
|
for zone_id, zone in coordinator.data.zones.items()
|
|
)
|
|
|
|
|
|
class RiscoCloudSwitch(RiscoCloudZoneEntity, SwitchEntity):
|
|
"""Representation of a bypass switch for a Risco cloud zone."""
|
|
|
|
_attr_entity_category = EntityCategory.CONFIG
|
|
_attr_translation_key = "bypassed"
|
|
|
|
def __init__(
|
|
self, coordinator: RiscoDataUpdateCoordinator, zone_id: int, zone: Zone
|
|
) -> None:
|
|
"""Init the zone."""
|
|
super().__init__(
|
|
coordinator=coordinator,
|
|
suffix="_bypassed",
|
|
zone_id=zone_id,
|
|
zone=zone,
|
|
)
|
|
|
|
@property
|
|
def is_on(self) -> bool | None:
|
|
"""Return true if the zone is bypassed."""
|
|
return self._zone.bypassed
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn the entity on."""
|
|
await self._bypass(True)
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn the entity off."""
|
|
await self._bypass(False)
|
|
|
|
async def _bypass(self, bypass: bool) -> None:
|
|
alarm = await self._risco.bypass_zone(self._zone_id, bypass)
|
|
self._zone = alarm.zones[self._zone_id]
|
|
self.async_write_ha_state()
|
|
|
|
|
|
class RiscoLocalSwitch(RiscoLocalZoneEntity, SwitchEntity):
|
|
"""Representation of a bypass switch for a Risco local zone."""
|
|
|
|
_attr_entity_category = EntityCategory.CONFIG
|
|
_attr_translation_key = "bypassed"
|
|
|
|
def __init__(self, system_id: str, zone_id: int, zone: Zone) -> None:
|
|
"""Init the zone."""
|
|
super().__init__(
|
|
system_id=system_id,
|
|
suffix="_bypassed",
|
|
zone_id=zone_id,
|
|
zone=zone,
|
|
)
|
|
|
|
@property
|
|
def is_on(self) -> bool | None:
|
|
"""Return true if the zone is bypassed."""
|
|
return self._zone.bypassed
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn the entity on."""
|
|
await self._bypass(True)
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn the entity off."""
|
|
await self._bypass(False)
|
|
|
|
async def _bypass(self, bypass: bool) -> None:
|
|
await self._zone.bypass(bypass)
|