From f56bdd29ff08fc71cbc9bcfd3d0324fc1264c340 Mon Sep 17 00:00:00 2001 From: Marcelo Moreira de Mello Date: Sun, 8 Oct 2017 04:05:41 -0400 Subject: [PATCH] Make Arlo battery_level icon dynamic (#9747) * Make Arlo battery_level icon dynamic * makes lint happy --- homeassistant/components/sensor/arlo.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/homeassistant/components/sensor/arlo.py b/homeassistant/components/sensor/arlo.py index 231a0ae69f8..f665d8e70ab 100644 --- a/homeassistant/components/sensor/arlo.py +++ b/homeassistant/components/sensor/arlo.py @@ -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