From 121ec5c684e43d82203d984979827015bce0c689 Mon Sep 17 00:00:00 2001 From: Jorim Tielemans Date: Tue, 6 Nov 2018 12:34:11 +0100 Subject: [PATCH] Add season icons (#18221) * Add season icons * Use STATE constants * Calm down hound * Update season.py --- homeassistant/components/sensor/season.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/homeassistant/components/sensor/season.py b/homeassistant/components/sensor/season.py index 3b14c3854cd..c9b222f2b26 100644 --- a/homeassistant/components/sensor/season.py +++ b/homeassistant/components/sensor/season.py @@ -34,6 +34,13 @@ HEMISPHERE_SEASON_SWAP = {STATE_WINTER: STATE_SUMMER, STATE_AUTUMN: STATE_SPRING, STATE_SUMMER: STATE_WINTER} +SEASON_ICONS = { + STATE_SPRING: 'mdi:flower', + STATE_SUMMER: 'mdi:sunglasses', + STATE_AUTUMN: 'mdi:leaf', + STATE_WINTER: 'mdi:snowflake' +} + PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ vol.Optional(CONF_TYPE, default=TYPE_ASTRONOMICAL): vol.In(VALID_TYPES) @@ -116,6 +123,11 @@ class Season(Entity): """Return the current season.""" return self.season + @property + def icon(self): + """Icon to use in the frontend, if any.""" + return SEASON_ICONS.get(self.season, 'mdi:cloud') + def update(self): """Update season.""" self.datetime = datetime.now()