Expose the condition code on condition sensors (#14011)

* expose the condition code on condition sensors

* 💄

* like thisss duh

* add test for condition_code

* It’s a string
This commit is contained in:
Jon Maddox 2018-04-21 04:20:33 -04:00 committed by Martin Hjelmare
parent cb490780c9
commit f12ff6f297
2 changed files with 8 additions and 3 deletions

View File

@ -131,9 +131,12 @@ class YahooWeatherSensor(Entity):
@property
def device_state_attributes(self):
"""Return the state attributes."""
return {
ATTR_ATTRIBUTION: CONF_ATTRIBUTION,
}
attrs = {ATTR_ATTRIBUTION: CONF_ATTRIBUTION}
if self._code is not None and "weather" in self._type:
attrs['condition_code'] = self._code
return attrs
def update(self):
"""Get the latest data from Yahoo! and updates the states."""

View File

@ -162,6 +162,8 @@ class TestWeather(unittest.TestCase):
state = self.hass.states.get('sensor.yweather_condition')
assert state is not None
self.assertEqual(state.state, 'Mostly Cloudy')
self.assertEqual(state.attributes.get('condition_code'),
'28')
self.assertEqual(state.attributes.get('friendly_name'),
'Yweather Condition')