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

Collapse websocket_api _state_diff into _state_diff_event (#118170)

Browse Source
This commit is contained in:
J. Nick Koston 2024-05-26 04:04:34 -10:00 committed by GitHub
parent 4a5c5fa311
commit caa65708fb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
homeassistant/components/websocket_api/messages.py
View File

@ -15,7 +15,7 @@ from homeassistant.const import (
COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_LAST_UPDATED,
COMPRESSED_STATE_STATE, COMPRESSED_STATE_STATE,
) )
from homeassistant.core import Event, EventStateChangedData, State from homeassistant.core import CompressedState, Event, EventStateChangedData
from homeassistant.helpers import config_validation as cv from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.json import ( from homeassistant.helpers.json import (
JSON_DUMP, JSON_DUMP,
@ -177,7 +177,14 @@ def _partial_cached_state_diff_message(event: Event[EventStateChangedData]) -> b
) )
def _state_diff_event(event: Event[EventStateChangedData]) -> dict: def _state_diff_event(
event: Event[EventStateChangedData],
) -> dict[
str,
list[str]
| dict[str, CompressedState]
| dict[str, dict[str, dict[str, str | list[str]]]],
]:
"""Convert a state_changed event to the minimal version. """Convert a state_changed event to the minimal version.
State update example State update example
@ -188,21 +195,10 @@ def _state_diff_event(event: Event[EventStateChangedData]) -> dict:
"r": [entity_id,…] "r": [entity_id,…]
} }
""" """
if (event_new_state := event.data["new_state"]) is None: if (new_state := event.data["new_state"]) is None:
return {ENTITY_EVENT_REMOVE: [event.data["entity_id"]]} return {ENTITY_EVENT_REMOVE: [event.data["entity_id"]]}
if (event_old_state := event.data["old_state"]) is None: if (old_state := event.data["old_state"]) is None:
return { return {ENTITY_EVENT_ADD: {new_state.entity_id: new_state.as_compressed_state}}
ENTITY_EVENT_ADD: {
event_new_state.entity_id: event_new_state.as_compressed_state
}
}
return _state_diff(event_old_state, event_new_state)
def _state_diff(
old_state: State, new_state: State
) -> dict[str, dict[str, dict[str, dict[str, str | list[str]]]]]:
"""Create a diff dict that can be used to overlay changes."""
additions: dict[str, Any] = {} additions: dict[str, Any] = {}
diff: dict[str, dict[str, Any]] = {STATE_DIFF_ADDITIONS: additions} diff: dict[str, dict[str, Any]] = {STATE_DIFF_ADDITIONS: additions}
new_state_context = new_state.context new_state_context = new_state.context