mirror of
https://github.com/home-assistant/core.git
synced 2026-07-15 10:34:19 +01:00
Tom Matheussen
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
"""Satel Integra client."""
|
|
|
|
from collections.abc import Callable
|
|
|
|
from satel_integra.satel_integra import AsyncSatel
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_HOST, CONF_PORT, EVENT_HOMEASSISTANT_STOP
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
|
|
from .const import (
|
|
CONF_OUTPUT_NUMBER,
|
|
CONF_PARTITION_NUMBER,
|
|
CONF_SWITCHABLE_OUTPUT_NUMBER,
|
|
CONF_ZONE_NUMBER,
|
|
SUBENTRY_TYPE_OUTPUT,
|
|
SUBENTRY_TYPE_PARTITION,
|
|
SUBENTRY_TYPE_SWITCHABLE_OUTPUT,
|
|
SUBENTRY_TYPE_ZONE,
|
|
)
|
|
|
|
|
|
class SatelClient:
|
|
"""Client to connect to Satel Integra."""
|
|
|
|
controller: AsyncSatel
|
|
|
|
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
|
|
"""Initialize the client wrapper."""
|
|
self.hass = hass
|
|
self.config_entry = entry
|
|
|
|
host = entry.data[CONF_HOST]
|
|
port = entry.data[CONF_PORT]
|
|
|
|
# Make sure we initialize the Satel controller with the configured entries to monitor
|
|
partitions = [
|
|
subentry.data[CONF_PARTITION_NUMBER]
|
|
for subentry in entry.subentries.values()
|
|
if subentry.subentry_type == SUBENTRY_TYPE_PARTITION
|
|
]
|
|
|
|
zones = [
|
|
subentry.data[CONF_ZONE_NUMBER]
|
|
for subentry in entry.subentries.values()
|
|
if subentry.subentry_type == SUBENTRY_TYPE_ZONE
|
|
]
|
|
|
|
outputs = [
|
|
subentry.data[CONF_OUTPUT_NUMBER]
|
|
for subentry in entry.subentries.values()
|
|
if subentry.subentry_type == SUBENTRY_TYPE_OUTPUT
|
|
]
|
|
|
|
switchable_outputs = [
|
|
subentry.data[CONF_SWITCHABLE_OUTPUT_NUMBER]
|
|
for subentry in entry.subentries.values()
|
|
if subentry.subentry_type == SUBENTRY_TYPE_SWITCHABLE_OUTPUT
|
|
]
|
|
|
|
monitored_outputs = outputs + switchable_outputs
|
|
|
|
self.controller = AsyncSatel(
|
|
host, port, hass.loop, zones, monitored_outputs, partitions
|
|
)
|
|
|
|
entry.async_on_unload(
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, self.close)
|
|
)
|
|
|
|
async def async_connect(
|
|
self,
|
|
zones_update_callback: Callable[[dict[str, dict[int, int]]], None],
|
|
outputs_update_callback: Callable[[dict[str, dict[int, int]]], None],
|
|
partitions_update_callback: Callable[[], None],
|
|
) -> None:
|
|
"""Start controller connection."""
|
|
result = await self.controller.connect()
|
|
if not result:
|
|
raise ConfigEntryNotReady("Controller failed to connect")
|
|
|
|
self.config_entry.async_create_background_task(
|
|
self.hass,
|
|
self.controller.keep_alive(),
|
|
f"satel_integra.{self.config_entry.entry_id}.keep_alive",
|
|
eager_start=False,
|
|
)
|
|
|
|
self.config_entry.async_create_background_task(
|
|
self.hass,
|
|
self.controller.monitor_status(
|
|
partitions_update_callback,
|
|
zones_update_callback,
|
|
outputs_update_callback,
|
|
),
|
|
f"satel_integra.{self.config_entry.entry_id}.monitor_status",
|
|
eager_start=False,
|
|
)
|
|
|
|
@callback
|
|
def close(self, *args, **kwargs) -> None:
|
|
"""Close the connection."""
|
|
|
|
self.controller.close()
|