Make Arlo battery_level icon dynamic (#9747)

* Make Arlo battery_level icon dynamic

* makes lint happy
This commit is contained in:
Marcelo Moreira de Mello 2017-10-08 04:05:41 -04:00 committed by Pascal Vizeli
parent 9eff9fa703
commit f56bdd29ff

View file

@ -16,6 +16,7 @@ from homeassistant.components.arlo import (
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import (ATTR_ATTRIBUTION, CONF_MONITORED_CONDITIONS)
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.icon import icon_for_battery_level
_LOGGER = logging.getLogger(__name__)
@ -84,6 +85,9 @@ class ArloSensor(Entity):
@property
def icon(self):
"""Icon to use in the frontend, if any."""
if self._sensor_type == 'battery_level' and self._state is not None:
return icon_for_battery_level(battery_level=int(self._state),
charging=False)
return self._icon
@property