mirror of
https://github.com/home-assistant/core
synced 2024-08-02 23:40:32 +02:00
e56c64db07
* Adjust screenlogic * Adjust mypy_config * Adjust coordinator cast * Move cast higher up Co-authored-by: epenet <epenet@users.noreply.github.com>
90 lines
3.1 KiB
Python
90 lines
3.1 KiB
Python
"""Services for ScreenLogic integration."""
|
|
|
|
import logging
|
|
|
|
from screenlogicpy import ScreenLogicError
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.core import HomeAssistant, ServiceCall, callback
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.service import async_extract_config_entry_ids
|
|
|
|
from .const import (
|
|
ATTR_COLOR_MODE,
|
|
DOMAIN,
|
|
SERVICE_SET_COLOR_MODE,
|
|
SUPPORTED_COLOR_MODES,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SET_COLOR_MODE_SCHEMA = cv.make_entity_service_schema(
|
|
{
|
|
vol.Required(ATTR_COLOR_MODE): vol.In(SUPPORTED_COLOR_MODES),
|
|
},
|
|
)
|
|
|
|
|
|
@callback
|
|
def async_load_screenlogic_services(hass: HomeAssistant):
|
|
"""Set up services for the ScreenLogic integration."""
|
|
if hass.services.has_service(DOMAIN, SERVICE_SET_COLOR_MODE):
|
|
# Integration-level services have already been added. Return.
|
|
return
|
|
|
|
async def extract_screenlogic_config_entry_ids(service_call: ServiceCall):
|
|
return [
|
|
entry_id
|
|
for entry_id in await async_extract_config_entry_ids(hass, service_call)
|
|
if (entry := hass.config_entries.async_get_entry(entry_id))
|
|
and entry.domain == DOMAIN
|
|
]
|
|
|
|
async def async_set_color_mode(service_call: ServiceCall) -> None:
|
|
if not (
|
|
screenlogic_entry_ids := await extract_screenlogic_config_entry_ids(
|
|
service_call
|
|
)
|
|
):
|
|
raise HomeAssistantError(
|
|
f"Failed to call service '{SERVICE_SET_COLOR_MODE}'. Config entry for target not found"
|
|
)
|
|
color_num = SUPPORTED_COLOR_MODES[service_call.data[ATTR_COLOR_MODE]]
|
|
for entry_id in screenlogic_entry_ids:
|
|
coordinator = hass.data[DOMAIN][entry_id]
|
|
_LOGGER.debug(
|
|
"Service %s called on %s with mode %s",
|
|
SERVICE_SET_COLOR_MODE,
|
|
coordinator.gateway.name,
|
|
color_num,
|
|
)
|
|
try:
|
|
if not await coordinator.gateway.async_set_color_lights(color_num):
|
|
raise HomeAssistantError(
|
|
f"Failed to call service '{SERVICE_SET_COLOR_MODE}'"
|
|
)
|
|
# Debounced refresh to catch any secondary
|
|
# changes in the device
|
|
await coordinator.async_request_refresh()
|
|
except ScreenLogicError as error:
|
|
raise HomeAssistantError(error) from error
|
|
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_SET_COLOR_MODE, async_set_color_mode, SET_COLOR_MODE_SCHEMA
|
|
)
|
|
|
|
|
|
@callback
|
|
def async_unload_screenlogic_services(hass: HomeAssistant):
|
|
"""Unload services for the ScreenLogic integration."""
|
|
if hass.data[DOMAIN]:
|
|
# There is still another config entry for this domain, don't remove services.
|
|
return
|
|
|
|
if not hass.services.has_service(DOMAIN, SERVICE_SET_COLOR_MODE):
|
|
return
|
|
|
|
_LOGGER.info("Unloading ScreenLogic Services")
|
|
hass.services.async_remove(domain=DOMAIN, service=SERVICE_SET_COLOR_MODE)
|