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

Add support for attribute caching to the switch platform (#106258)

Browse Source
This commit is contained in:
J. Nick Koston 2023-12-23 09:22:02 -10:00 committed by GitHub
parent ebdf7b9c8c
commit 83e1ba338a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/switch/__init__.py
View File

@ -5,6 +5,7 @@ from datetime import timedelta
from enum import StrEnum from enum import StrEnum
from functools import partial from functools import partial
import logging import logging
from typing import TYPE_CHECKING
import voluptuous as vol import voluptuous as vol
@ -32,6 +33,11 @@ from homeassistant.loader import bind_hass
from .const import DOMAIN from .const import DOMAIN
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 + ".{}"
@ -108,13 +114,18 @@ class SwitchEntityDescription(ToggleEntityDescription, frozen_or_thawed=True):
device_class: SwitchDeviceClass | None = None device_class: SwitchDeviceClass | None = None
class SwitchEntity(ToggleEntity): CACHED_PROPERTIES_WITH_ATTR_ = {
"device_class",
}
class SwitchEntity(ToggleEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
"""Base class for switch entities.""" """Base class for switch entities."""
entity_description: SwitchEntityDescription entity_description: SwitchEntityDescription
_attr_device_class: SwitchDeviceClass | None _attr_device_class: SwitchDeviceClass | None
@property @cached_property
def device_class(self) -> SwitchDeviceClass | None: def device_class(self) -> SwitchDeviceClass | 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"):