home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-05 05:35:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Define entity attributes as entity class variables

Browse Source
This commit is contained in:
Franck Nijhof
2021-05-21 13:13:56 +02:00
parent 00208ff0d8
commit 7ff574b7ea
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/helpers/entity.py
View File

@@ -168,6 +168,11 @@ class Entity(ABC):
# If entity is added to an entity platform # If entity is added to an entity platform
_added = False _added = False
# Entity Properties
_device_class: str | None = None
_icon: str | None = None
_unit_of_measurement: str | None = None
@property @property
def should_poll(self) -> bool: def should_poll(self) -> bool:
"""Return True if entity has to be polled for state. """Return True if entity has to be polled for state.
@@ -240,17 +245,17 @@ class Entity(ABC):
@property @property
def device_class(self) -> str | None: def device_class(self) -> str | None:
"""Return the class of this device, from component DEVICE_CLASSES.""" """Return the class of this device, from component DEVICE_CLASSES."""
return None return self._device_class
@property @property
def unit_of_measurement(self) -> str | None: def unit_of_measurement(self) -> str | None:
"""Return the unit of measurement of this entity, if any.""" """Return the unit of measurement of this entity, if any."""
return None return self._unit_of_measurement
@property @property
def icon(self) -> str | None: def icon(self) -> str | None:
"""Return the icon to use in the frontend, if any.""" """Return the icon to use in the frontend, if any."""
return None return self._icon
@property @property
def entity_picture(self) -> str | None: def entity_picture(self) -> str | None: