156 lines
5.6 KiB
Python
156 lines
5.6 KiB
Python
"""Sternwarte Home Assistant integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import Platform
|
|
from functools import partial
|
|
|
|
from homeassistant.core import HomeAssistant, ServiceCall
|
|
|
|
from .const import (
|
|
DOMAIN,
|
|
SERVICE_CLEAR_CLOSE_SCHEDULE,
|
|
SERVICE_CLEAR_OPEN_SCHEDULE,
|
|
SERVICE_CLEAR_SCHEDULE,
|
|
SERVICE_CLOSE_ROOF,
|
|
SERVICE_OPEN_ROOF,
|
|
SERVICE_POWER_OFF,
|
|
SERVICE_POWER_ON,
|
|
SERVICE_SET_NAUTICAL_SCHEDULE,
|
|
SERVICE_SET_NAUTICAL_OPEN_SCHEDULE,
|
|
SERVICE_SET_NAUTICAL_CLOSE_SCHEDULE,
|
|
SERVICE_STOP_ROOF,
|
|
)
|
|
from .coordinator import SternwarteDataUpdateCoordinator
|
|
|
|
PLATFORMS: list[Platform] = [
|
|
Platform.BINARY_SENSOR,
|
|
Platform.BUTTON,
|
|
Platform.COVER,
|
|
Platform.DATETIME,
|
|
Platform.SENSOR,
|
|
Platform.SWITCH,
|
|
]
|
|
|
|
|
|
async def _async_call_on_all(
|
|
hass: HomeAssistant,
|
|
entry_action: str,
|
|
) -> None:
|
|
"""Run an action on all Sternwarte coordinators."""
|
|
for coordinator in hass.data.get(DOMAIN, {}).values():
|
|
await getattr(coordinator, entry_action)()
|
|
|
|
|
|
async def _async_handle_open_roof(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_open_roof")
|
|
|
|
|
|
async def _async_handle_close_roof(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_close_roof")
|
|
|
|
|
|
async def _async_handle_stop_roof(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_stop_roof")
|
|
|
|
|
|
async def _async_handle_power_on(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_turn_power_on")
|
|
|
|
|
|
async def _async_handle_power_off(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_turn_power_off")
|
|
|
|
|
|
async def _async_handle_clear_schedule(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_clear_schedule")
|
|
|
|
|
|
async def _async_handle_clear_open_schedule(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_clear_open_schedule")
|
|
|
|
|
|
async def _async_handle_clear_close_schedule(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_clear_close_schedule")
|
|
|
|
|
|
async def _async_handle_set_nautical_schedule(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_set_next_nautical_schedule")
|
|
|
|
|
|
async def _async_handle_set_nautical_open_schedule(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_set_next_nautical_open_schedule")
|
|
|
|
|
|
async def _async_handle_set_nautical_close_schedule(hass: HomeAssistant, call: ServiceCall) -> None:
|
|
await _async_call_on_all(hass, "async_set_next_nautical_close_schedule")
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up Sternwarte from a config entry."""
|
|
coordinator = SternwarteDataUpdateCoordinator(hass, entry)
|
|
await coordinator.async_config_entry_first_refresh()
|
|
|
|
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
|
|
|
|
if not hass.services.has_service(DOMAIN, SERVICE_OPEN_ROOF):
|
|
hass.services.async_register(DOMAIN, SERVICE_OPEN_ROOF, partial(_async_handle_open_roof, hass))
|
|
hass.services.async_register(DOMAIN, SERVICE_CLOSE_ROOF, partial(_async_handle_close_roof, hass))
|
|
hass.services.async_register(DOMAIN, SERVICE_STOP_ROOF, partial(_async_handle_stop_roof, hass))
|
|
hass.services.async_register(DOMAIN, SERVICE_POWER_ON, partial(_async_handle_power_on, hass))
|
|
hass.services.async_register(DOMAIN, SERVICE_POWER_OFF, partial(_async_handle_power_off, hass))
|
|
hass.services.async_register(DOMAIN, SERVICE_CLEAR_SCHEDULE, partial(_async_handle_clear_schedule, hass))
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_CLEAR_OPEN_SCHEDULE,
|
|
partial(_async_handle_clear_open_schedule, hass),
|
|
)
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_CLEAR_CLOSE_SCHEDULE,
|
|
partial(_async_handle_clear_close_schedule, hass),
|
|
)
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_SET_NAUTICAL_SCHEDULE,
|
|
partial(_async_handle_set_nautical_schedule, hass),
|
|
)
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_SET_NAUTICAL_OPEN_SCHEDULE,
|
|
partial(_async_handle_set_nautical_open_schedule, hass),
|
|
)
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_SET_NAUTICAL_CLOSE_SCHEDULE,
|
|
partial(_async_handle_set_nautical_close_schedule, hass),
|
|
)
|
|
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
if unload_ok:
|
|
hass.data[DOMAIN].pop(entry.entry_id, None)
|
|
if not hass.data[DOMAIN]:
|
|
for service in (
|
|
SERVICE_OPEN_ROOF,
|
|
SERVICE_CLOSE_ROOF,
|
|
SERVICE_STOP_ROOF,
|
|
SERVICE_POWER_ON,
|
|
SERVICE_POWER_OFF,
|
|
SERVICE_CLEAR_SCHEDULE,
|
|
SERVICE_CLEAR_OPEN_SCHEDULE,
|
|
SERVICE_CLEAR_CLOSE_SCHEDULE,
|
|
SERVICE_SET_NAUTICAL_SCHEDULE,
|
|
SERVICE_SET_NAUTICAL_OPEN_SCHEDULE,
|
|
SERVICE_SET_NAUTICAL_CLOSE_SCHEDULE,
|
|
):
|
|
if hass.services.has_service(DOMAIN, service):
|
|
hass.services.async_remove(DOMAIN, service)
|
|
return unload_ok
|