1
mirror of https://github.com/home-assistant/core synced 2024-08-28 03:36:46 +02:00

Add support for attribute caching to the time platform (#106339)

This commit is contained in:
J. Nick Koston 2023-12-23 15:14:31 -10:00 committed by GitHub
parent b6c2842b01
commit b5e1074062
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,7 +3,7 @@ from __future__ import annotations
from datetime import time, timedelta from datetime import time, timedelta
import logging import logging
from typing import final from typing import TYPE_CHECKING, final
import voluptuous as vol import voluptuous as vol
@ -21,6 +21,12 @@ from homeassistant.helpers.typing import ConfigType
from .const import DOMAIN, SERVICE_SET_VALUE from .const import DOMAIN, SERVICE_SET_VALUE
if TYPE_CHECKING:
from functools import cached_property
else:
from homeassistant.backports.functools import cached_property
SCAN_INTERVAL = timedelta(seconds=30) SCAN_INTERVAL = timedelta(seconds=30)
ENTITY_ID_FORMAT = DOMAIN + ".{}" ENTITY_ID_FORMAT = DOMAIN + ".{}"
@ -65,7 +71,10 @@ class TimeEntityDescription(EntityDescription, frozen_or_thawed=True):
"""A class that describes time entities.""" """A class that describes time entities."""
class TimeEntity(Entity): CACHED_PROPERTIES_WITH_ATTR_ = {"native_value"}
class TimeEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
"""Representation of a Time entity.""" """Representation of a Time entity."""
entity_description: TimeEntityDescription entity_description: TimeEntityDescription
@ -73,13 +82,13 @@ class TimeEntity(Entity):
_attr_device_class: None = None _attr_device_class: None = None
_attr_state: None = None _attr_state: None = None
@property @cached_property
@final @final
def device_class(self) -> None: def device_class(self) -> None:
"""Return the device class for the entity.""" """Return the device class for the entity."""
return None return None
@property @cached_property
@final @final
def state_attributes(self) -> None: def state_attributes(self) -> None:
"""Return the state attributes.""" """Return the state attributes."""
@ -93,7 +102,7 @@ class TimeEntity(Entity):
return None return None
return self.native_value.isoformat() return self.native_value.isoformat()
@property @cached_property
def native_value(self) -> time | None: def native_value(self) -> time | None:
"""Return the value reported by the time.""" """Return the value reported by the time."""
return self._attr_native_value return self._attr_native_value