evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for attribute caching to the button platform (#106259)

Browse source
This commit is contained in:
J. Nick Koston 2023-12-23 00:09:46 -10:00 • committed by GitHub
parent 98dd69ba09
commit ef0031cbcf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
homeassistant/components/button/__init__.py
View file

@ -4,7 +4,7 @@ from __future__ import annotations
from datetime import datetime, timedelta from datetime import datetime, timedelta
from enum import StrEnum from enum import StrEnum
import logging import logging
from typing import final from typing import TYPE_CHECKING, final
import voluptuous as vol import voluptuous as vol
@ -22,6 +22,11 @@ from homeassistant.util import dt as dt_util
from .const import DOMAIN, SERVICE_PRESS from .const import DOMAIN, SERVICE_PRESS
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 + ".{}"
@ -78,7 +83,12 @@ class ButtonEntityDescription(EntityDescription, frozen_or_thawed=True):
device_class: ButtonDeviceClass | None = None device_class: ButtonDeviceClass | None = None
class ButtonEntity(RestoreEntity): CACHED_PROPERTIES_WITH_ATTR_ = {
"device_class",
}
class ButtonEntity(RestoreEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
"""Representation of a Button entity.""" """Representation of a Button entity."""
entity_description: ButtonEntityDescription entity_description: ButtonEntityDescription
@ -94,7 +104,7 @@ class ButtonEntity(RestoreEntity):
""" """
return self.device_class is not None return self.device_class is not None
@property @cached_property
def device_class(self) -> ButtonDeviceClass | None: def device_class(self) -> ButtonDeviceClass | None:
"""Return the class of this entity.""" """Return the class of this entity."""
if hasattr(self, "_attr_device_class"): if hasattr(self, "_attr_device_class"):