Allow device class to control icons for tesla (#37526)

This commit is contained in:
Chris Talkington 2020-08-04 19:00:05 -05:00 committed by GitHub
parent 17c9e31e2c
commit d89bfe79f9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -15,12 +15,10 @@ from homeassistant.const import (
CONF_SCAN_INTERVAL,
CONF_TOKEN,
CONF_USERNAME,
DEVICE_CLASS_BATTERY,
)
from homeassistant.core import callback
from homeassistant.helpers import aiohttp_client, config_validation as cv
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.icon import icon_for_battery_level
from homeassistant.util import slugify
from .config_flow import (
@ -223,14 +221,8 @@ class TeslaDevice(Entity):
@property
def icon(self):
"""Return the icon of the sensor."""
if (
self.device_class == DEVICE_CLASS_BATTERY
and self.tesla_device.has_battery()
):
return icon_for_battery_level(
battery_level=self.tesla_device.battery_level(),
charging=self.tesla_device.battery_charging(),
)
if self.device_class:
return None
return self._icon