Change default icons for Tesla components (#30288)

This commit is contained in:
Alan Tse 2019-12-30 08:52:34 -08:00 committed by Fabian Affolter
parent d0c9a42b81
commit 40e3d6f773

View File

@ -26,7 +26,7 @@ from .config_flow import (
configured_instances,
validate_input,
)
from .const import DATA_LISTENER, DOMAIN, TESLA_COMPONENTS
from .const import DATA_LISTENER, DOMAIN, SENSOR_ICONS, TESLA_COMPONENTS
_LOGGER = logging.getLogger(__name__)
@ -186,6 +186,11 @@ class TeslaDevice(Entity):
self._name = self.tesla_device.name
self.tesla_id = slugify(self.tesla_device.uniq_name)
self._attributes = {}
self._icon = (
SENSOR_ICONS[self.tesla_device.type]
if self.tesla_device.type and self.tesla_device.type in SENSOR_ICONS.keys()
else None
)
@property
def name(self):
@ -197,6 +202,11 @@ class TeslaDevice(Entity):
"""Return a unique ID."""
return self.tesla_id
@property
def icon(self):
"""Return the icon of the sensor."""
return self._icon
@property
def should_poll(self):
"""Return the polling state."""