jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 08:47:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve websocket_api schema typing (#120411)

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

6
homeassistant/components/websocket_api/__init__.py
View File

@ -4,11 +4,9 @@ from __future__ import annotations
from typing import Final, cast
import voluptuous as vol
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.typing import ConfigType
from homeassistant.helpers.typing import ConfigType, VolSchemaType
from homeassistant.loader import bind_hass
from . import commands, connection, const, decorators, http, messages # noqa: F401
@ -55,7 +53,7 @@ def async_register_command(
hass: HomeAssistant,
command_or_handler: str | const.WebSocketCommandHandler,
handler: const.WebSocketCommandHandler | None = None,
schema: vol.Schema | None = None,
schema: VolSchemaType | None = None,
) -> None:
"""Register a websocket command."""
if handler is None:

3
homeassistant/components/websocket_api/decorators.py
View File

@ -11,6 +11,7 @@ import voluptuous as vol
from homeassistant.const import HASSIO_USER_NAME
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import Unauthorized
from homeassistant.helpers.typing import VolDictType
from . import const, messages
from .connection import ActiveConnection
@ -130,7 +131,7 @@ def ws_require_user(
def websocket_command(
schema: dict[vol.Marker, Any] | vol.All,
schema: VolDictType | vol.All,
) -> Callable[[const.WebSocketCommandHandler], const.WebSocketCommandHandler]:
"""Tag a function as a websocket command.