Add season icons (#18221)

* Add season icons

* Use STATE constants

* Calm down hound

* Update season.py
This commit is contained in:
Jorim Tielemans 2018-11-06 12:34:11 +01:00 committed by Pascal Vizeli
parent 087bffeaae
commit 121ec5c684

View File

@ -34,6 +34,13 @@ HEMISPHERE_SEASON_SWAP = {STATE_WINTER: STATE_SUMMER,
STATE_AUTUMN: STATE_SPRING, STATE_AUTUMN: STATE_SPRING,
STATE_SUMMER: STATE_WINTER} 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({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_TYPE, default=TYPE_ASTRONOMICAL): vol.In(VALID_TYPES) vol.Optional(CONF_TYPE, default=TYPE_ASTRONOMICAL): vol.In(VALID_TYPES)
@ -116,6 +123,11 @@ class Season(Entity):
"""Return the current season.""" """Return the current season."""
return self.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): def update(self):
"""Update season.""" """Update season."""
self.datetime = datetime.now() self.datetime = datetime.now()