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

Small cleanups to template helper (#87944)

Browse Source 
- reduce dict lookups
- avoid split_entity_id
This commit is contained in:
J. Nick Koston 2023-02-12 13:11:48 -06:00 committed by GitHub
parent 69dc50c917
commit e19f8595fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/helpers/template.py
View File

@ -821,8 +821,8 @@ class TemplateStateBase(State):
self._as_dict: ReadOnlyDict[str, Collection[Any]] | None = None self._as_dict: ReadOnlyDict[str, Collection[Any]] | None = None
def _collect_state(self) -> None: def _collect_state(self) -> None:
if self._collect and _RENDER_INFO in self._hass.data: if self._collect and (_render_info := self._hass.data.get(_RENDER_INFO)):
self._hass.data[_RENDER_INFO].entities.add(self._entity_id) _render_info.entities.add(self._entity_id)
# Jinja will try __getitem__ first and it avoids the need # Jinja will try __getitem__ first and it avoids the need
# to call is_safe_attribute # to call is_safe_attribute
@ -830,8 +830,8 @@ class TemplateStateBase(State):
"""Return a property as an attribute for jinja.""" """Return a property as an attribute for jinja."""
if item in _COLLECTABLE_STATE_ATTRIBUTES: if item in _COLLECTABLE_STATE_ATTRIBUTES:
# _collect_state inlined here for performance # _collect_state inlined here for performance
if self._collect and _RENDER_INFO in self._hass.data: if self._collect and (_render_info := self._hass.data.get(_RENDER_INFO)):
self._hass.data[_RENDER_INFO].entities.add(self._entity_id) _render_info.entities.add(self._entity_id)
return getattr(self._state, item) return getattr(self._state, item)
if item == "entity_id": if item == "entity_id":
return self._entity_id return self._entity_id
@ -910,12 +910,13 @@ class TemplateStateBase(State):
) )
self._collect_state() self._collect_state()
unit = self._state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) if rounded and self._state.domain == SENSOR_DOMAIN:
if rounded and split_entity_id(self._entity_id)[0] == SENSOR_DOMAIN:
state = async_rounded_state(self._hass, self._entity_id, self._state) state = async_rounded_state(self._hass, self._entity_id, self._state)
else: else:
state = self._state.state state = self._state.state
return f"{state} {unit}" if with_unit and unit else state if with_unit and (unit := self._state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)):
return f"{state} {unit}"
return state
def __eq__(self, other: Any) -> bool: def __eq__(self, other: Any) -> bool:
"""Ensure we collect on equality check.""" """Ensure we collect on equality check."""