Improve typing in core service registration (#63005)
Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
parent
1320f27fd7
commit
9971831b51
3 changed files with 4 additions and 4 deletions
|
@ -1330,7 +1330,7 @@ class ServiceRegistry:
|
|||
self,
|
||||
domain: str,
|
||||
service: str,
|
||||
service_func: Callable,
|
||||
service_func: Callable[[ServiceCall], Awaitable[None] | None],
|
||||
schema: vol.Schema | None = None,
|
||||
) -> None:
|
||||
"""
|
||||
|
@ -1347,7 +1347,7 @@ class ServiceRegistry:
|
|||
self,
|
||||
domain: str,
|
||||
service: str,
|
||||
service_func: Callable,
|
||||
service_func: Callable[[ServiceCall], Awaitable[None] | None],
|
||||
schema: vol.Schema | None = None,
|
||||
) -> None:
|
||||
"""
|
||||
|
|
|
@ -198,7 +198,7 @@ class EntityComponent:
|
|||
if isinstance(schema, dict):
|
||||
schema = cv.make_entity_service_schema(schema)
|
||||
|
||||
async def handle_service(call: Callable) -> None:
|
||||
async def handle_service(call: ServiceCall) -> None:
|
||||
"""Handle the service."""
|
||||
await self.hass.helpers.service.entity_service_call(
|
||||
self._platforms.values(), func, call, required_features
|
||||
|
|
|
@ -716,7 +716,7 @@ def async_register_admin_service(
|
|||
hass: HomeAssistant,
|
||||
domain: str,
|
||||
service: str,
|
||||
service_func: Callable[[ServiceCall], Awaitable | None],
|
||||
service_func: Callable[[ServiceCall], Awaitable[None] | None],
|
||||
schema: vol.Schema = vol.Schema({}, extra=vol.PREVENT_EXTRA),
|
||||
) -> None:
|
||||
"""Register a service that requires admin access."""
|
||||
|
|
Loading…
Add table
Reference in a new issue