jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 06:07:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve async_register_admin_service schema typing (#120405)

Browse Source
This commit is contained in:
Marc Mueller 2024-06-25 12:48:00 +02:00 committed by GitHub
parent de8bccb650
commit 62fd692d27
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
homeassistant/components/zha/websocket_api.py
View File

@ -21,6 +21,7 @@ from homeassistant.helpers import entity_registry as er
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.service import async_register_admin_service from homeassistant.helpers.service import async_register_admin_service
from homeassistant.helpers.typing import VolDictType, VolSchemaType
from .api import ( from .api import (
async_change_channel, async_change_channel,
@ -126,7 +127,7 @@ def _ensure_list_if_present[_T](value: _T | None) -> list[_T] | list[Any] | None
return cast("list[_T]", value) if isinstance(value, list) else [value] return cast("list[_T]", value) if isinstance(value, list) else [value]
SERVICE_PERMIT_PARAMS = { SERVICE_PERMIT_PARAMS: VolDictType = {
vol.Optional(ATTR_IEEE): IEEE_SCHEMA, vol.Optional(ATTR_IEEE): IEEE_SCHEMA,
vol.Optional(ATTR_DURATION, default=60): vol.All( vol.Optional(ATTR_DURATION, default=60): vol.All(
vol.Coerce(int), vol.Range(0, 254) vol.Coerce(int), vol.Range(0, 254)
@ -138,7 +139,7 @@ SERVICE_PERMIT_PARAMS = {
vol.Exclusive(ATTR_QR_CODE, "install_code"): vol.All(cv.string, qr_to_install_code), vol.Exclusive(ATTR_QR_CODE, "install_code"): vol.All(cv.string, qr_to_install_code),
} }
SERVICE_SCHEMAS = { SERVICE_SCHEMAS: dict[str, VolSchemaType] = {
SERVICE_PERMIT: vol.Schema( SERVICE_PERMIT: vol.Schema(
vol.All( vol.All(
cv.deprecated(ATTR_IEEE_ADDRESS, replacement_key=ATTR_IEEE), cv.deprecated(ATTR_IEEE_ADDRESS, replacement_key=ATTR_IEEE),

4
homeassistant/helpers/service.py
View File

@ -63,7 +63,7 @@ from . import (
) )
from .group import expand_entity_ids from .group import expand_entity_ids
from .selector import TargetSelector from .selector import TargetSelector
from .typing import ConfigType, TemplateVarsType from .typing import ConfigType, TemplateVarsType, VolSchemaType
if TYPE_CHECKING: if TYPE_CHECKING:
from .entity import Entity from .entity import Entity
@ -1100,7 +1100,7 @@ def async_register_admin_service(
domain: str, domain: str,
service: str, service: str,
service_func: Callable[[ServiceCall], Awaitable[None] | None], service_func: Callable[[ServiceCall], Awaitable[None] | None],
schema: vol.Schema = vol.Schema({}, extra=vol.PREVENT_EXTRA), schema: VolSchemaType = vol.Schema({}, extra=vol.PREVENT_EXTRA),
) -> None: ) -> None:
"""Register a service that requires admin access.""" """Register a service that requires admin access."""
hass.services.async_register( hass.services.async_register(