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

Avoid many string lowers in the state machine (#109607)

Browse Source
This commit is contained in:
J. Nick Koston 2024-02-04 10:35:35 -06:00 committed by GitHub
parent 2c91b31233
commit e35c7fde89
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
homeassistant/core.py
View File

@ -1757,7 +1757,9 @@ class StateMachine:
Async friendly. Async friendly.
""" """
return self._states_data.get(entity_id.lower()) return self._states_data.get(entity_id) or self._states_data.get(
entity_id.lower()
)
def is_state(self, entity_id: str, state: str) -> bool: def is_state(self, entity_id: str, state: str) -> bool:
"""Test if entity exists and is in specified state. """Test if entity exists and is in specified state.
@ -1870,10 +1872,16 @@ class StateMachine:
This method must be run in the event loop. This method must be run in the event loop.
""" """
entity_id = entity_id.lower()
new_state = str(new_state) new_state = str(new_state)
attributes = attributes or {} attributes = attributes or {}
if (old_state := self._states_data.get(entity_id)) is None: old_state = self._states_data.get(entity_id)
if old_state is None:
# If the state is missing, try to convert the entity_id to lowercase
# and try again.
entity_id = entity_id.lower()
old_state = self._states_data.get(entity_id)
if old_state is None:
same_state = False same_state = False
same_attr = False same_attr = False
last_changed = None last_changed = None

13
tests/test_core.py
View File

@ -1194,8 +1194,8 @@ async def test_statemachine_remove(hass: HomeAssistant) -> None:
assert len(events) == 1 assert len(events) == 1
async def test_statemachine_case_insensitivty(hass: HomeAssistant) -> None: async def test_state_machine_case_insensitivity(hass: HomeAssistant) -> None:
"""Test insensitivty.""" """Test setting and getting states entity_id insensitivity."""
events = async_capture_events(hass, EVENT_STATE_CHANGED) events = async_capture_events(hass, EVENT_STATE_CHANGED)
hass.states.async_set("light.BOWL", "off") hass.states.async_set("light.BOWL", "off")
@ -1204,6 +1204,15 @@ async def test_statemachine_case_insensitivty(hass: HomeAssistant) -> None:
assert hass.states.is_state("light.bowl", "off") assert hass.states.is_state("light.bowl", "off")
assert len(events) == 1 assert len(events) == 1
hass.states.async_set("ligHT.Bowl", "on")
assert hass.states.get("light.bowl").state == "on"
hass.states.async_set("light.BOWL", "off")
assert hass.states.get("light.BoWL").state == "off"
hass.states.async_set("light.bowl", "on")
assert hass.states.get("light.bowl").state == "on"
async def test_statemachine_last_changed_not_updated_on_same_state( async def test_statemachine_last_changed_not_updated_on_same_state(
hass: HomeAssistant, hass: HomeAssistant,