Changed to use dictionary

This commit is contained in:
Rowan Hine 2016-02-23 18:01:50 +00:00
parent 532d807771
commit f8240a9cda

View File

@ -46,26 +46,20 @@ class SteamSensor(Entity):
@property @property
def state(self): def state(self):
""" State of the sensor. """ """ State of the sensor. """
if self._profile.status == 1: return self._state
account_state = 'Online'
elif self._profile.status == 2:
account_state = 'Busy'
elif self._profile.status == 3:
account_state = 'Away'
elif self._profile.status == 4:
account_state = 'Snooze'
elif self._profile.status == 5:
account_state = 'Trade'
elif self._profile.status == 6:
account_state = 'Play'
else:
account_state = 'Offline'
return account_state
# pylint: disable=no-member # pylint: disable=no-member
def update(self): def update(self):
""" Update device state. """ """ Update device state. """
self._profile = self._steamod.user.profile(self._account) self._profile = self._steamod.user.profile(self._account)
self._state = {
1: 'Online',
2: 'Busy',
3: 'Away',
4: 'Snooze',
5: 'Trade',
6: 'Play',
}.get(self._profile.status, 'Offline')
@property @property
def device_state_attributes(self): def device_state_attributes(self):