From 15a5cebd5f3533ed671107efc5f11d40b892aae8 Mon Sep 17 00:00:00 2001 From: Quentame Date: Thu, 5 Mar 2020 00:02:22 +0100 Subject: [PATCH] Add icons to Coronavirus (#32480) * Add icons to Coronavirus * Update homeassistant/components/coronavirus/sensor.py Co-authored-by: Paulus Schoutsen --- homeassistant/components/coronavirus/sensor.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/coronavirus/sensor.py b/homeassistant/components/coronavirus/sensor.py index 20f18896431..3885dbebf24 100644 --- a/homeassistant/components/coronavirus/sensor.py +++ b/homeassistant/components/coronavirus/sensor.py @@ -5,6 +5,13 @@ from homeassistant.helpers.entity import Entity from . import get_coordinator from .const import ATTRIBUTION, OPTION_WORLDWIDE +SENSORS = { + "confirmed": "mdi:emoticon-neutral-outline", + "current": "mdi:emoticon-frown-outline", + "recovered": "mdi:emoticon-happy-outline", + "deaths": "mdi:emoticon-dead-outline", +} + async def async_setup_entry(hass, config_entry, async_add_entities): """Defer sensor setup to the shared sensor module.""" @@ -12,7 +19,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities): async_add_entities( CoronavirusSensor(coordinator, config_entry.data["country"], info_type) - for info_type in ("confirmed", "recovered", "deaths", "current") + for info_type in SENSORS ) @@ -50,6 +57,11 @@ class CoronavirusSensor(Entity): return getattr(self.coordinator.data[self.country], self.info_type) + @property + def icon(self): + """Return the icon.""" + return SENSORS[self.info_type] + @property def unit_of_measurement(self): """Return unit of measurement."""