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

Avoid creating template objects in mqtt sensor if they are not configured (#118194)

Browse Source
This commit is contained in:
J. Nick Koston 2024-05-26 12:09:06 -10:00 committed by GitHub
parent 841d5dfd4f
commit 98d7821f47
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
homeassistant/components/mqtt/sensor.py
View File

@ -131,8 +131,12 @@ class MqttSensor(MqttEntity, RestoreSensor):
_expiration_trigger: CALLBACK_TYPE | None = None
_expire_after: int | None
_expired: bool | None
_template: Callable[[ReceivePayloadType, PayloadSentinel], ReceivePayloadType]
_last_reset_template: Callable[[ReceivePayloadType], ReceivePayloadType]
_template: (
Callable[[ReceivePayloadType, PayloadSentinel], ReceivePayloadType] | None
) = None
_last_reset_template: Callable[[ReceivePayloadType], ReceivePayloadType] | None = (
None
)
async def mqtt_async_added_to_hass(self) -> None:
"""Restore state for entities with expire_after set."""
@ -172,8 +176,7 @@ class MqttSensor(MqttEntity, RestoreSensor):
)
async def async_will_remove_from_hass(self) -> None:
"""Remove exprire triggers."""
# Clean up expire triggers
"""Remove expire triggers."""
if self._expiration_trigger:
_LOGGER.debug("Clean up expire after trigger for %s", self.entity_id)
self._expiration_trigger()
@ -202,12 +205,14 @@ class MqttSensor(MqttEntity, RestoreSensor):
else:
self._expired = None
self._template = MqttValueTemplate(
self._config.get(CONF_VALUE_TEMPLATE), entity=self
).async_render_with_possible_json_value
self._last_reset_template = MqttValueTemplate(
self._config.get(CONF_LAST_RESET_VALUE_TEMPLATE), entity=self
).async_render_with_possible_json_value
if value_template := config.get(CONF_VALUE_TEMPLATE):
self._template = MqttValueTemplate(
value_template, entity=self
).async_render_with_possible_json_value
if last_reset_template := config.get(CONF_LAST_RESET_VALUE_TEMPLATE):
self._last_reset_template = MqttValueTemplate(
last_reset_template, entity=self
).async_render_with_possible_json_value
@callback
def _update_state(self, msg: ReceiveMessage) -> None:
@ -226,7 +231,10 @@ class MqttSensor(MqttEntity, RestoreSensor):
self.hass, self._expire_after, self._value_is_expired
)
payload = self._template(msg.payload, PayloadSentinel.DEFAULT)
if template := self._template:
payload = template(msg.payload, PayloadSentinel.DEFAULT)
else:
payload = msg.payload
if payload is PayloadSentinel.DEFAULT:
return
new_value = str(payload)
@ -260,8 +268,8 @@ class MqttSensor(MqttEntity, RestoreSensor):
@callback
def _update_last_reset(self, msg: ReceiveMessage) -> None:
payload = self._last_reset_template(msg.payload)
template = self._last_reset_template
payload = msg.payload if template is None else template(msg.payload)
if not payload:
_LOGGER.debug("Ignoring empty last_reset message from '%s'", msg.topic)
return