mirror of
https://github.com/home-assistant/core.git
synced 2025-07-18 02:37:08 +00:00
Use entity class attributes for Bizkaibus (#52888)
This commit is contained in:
parent
5d2e5d2612
commit
01afd141c6
@ -31,40 +31,24 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
|
|||||||
route = config[CONF_ROUTE]
|
route = config[CONF_ROUTE]
|
||||||
|
|
||||||
data = Bizkaibus(stop, route)
|
data = Bizkaibus(stop, route)
|
||||||
add_entities([BizkaibusSensor(data, stop, route, name)], True)
|
add_entities([BizkaibusSensor(data, name)], True)
|
||||||
|
|
||||||
|
|
||||||
class BizkaibusSensor(SensorEntity):
|
class BizkaibusSensor(SensorEntity):
|
||||||
"""The class for handling the data."""
|
"""The class for handling the data."""
|
||||||
|
|
||||||
def __init__(self, data, stop, route, name):
|
_attr_unit_of_measurement = TIME_MINUTES
|
||||||
|
|
||||||
|
def __init__(self, data, name):
|
||||||
"""Initialize the sensor."""
|
"""Initialize the sensor."""
|
||||||
self.data = data
|
self.data = data
|
||||||
self.stop = stop
|
self._attr_name = name
|
||||||
self.route = route
|
|
||||||
self._name = name
|
|
||||||
self._state = None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def name(self):
|
|
||||||
"""Return the name of the sensor."""
|
|
||||||
return self._name
|
|
||||||
|
|
||||||
@property
|
|
||||||
def state(self):
|
|
||||||
"""Return the state of the sensor."""
|
|
||||||
return self._state
|
|
||||||
|
|
||||||
@property
|
|
||||||
def unit_of_measurement(self):
|
|
||||||
"""Return the unit of measurement of the sensor."""
|
|
||||||
return TIME_MINUTES
|
|
||||||
|
|
||||||
def update(self):
|
def update(self):
|
||||||
"""Get the latest data from the webservice."""
|
"""Get the latest data from the webservice."""
|
||||||
self.data.update()
|
self.data.update()
|
||||||
with suppress(TypeError):
|
with suppress(TypeError):
|
||||||
self._state = self.data.info[0][ATTR_DUE_IN]
|
self._attr_state = self.data.info[0][ATTR_DUE_IN]
|
||||||
|
|
||||||
|
|
||||||
class Bizkaibus:
|
class Bizkaibus:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user