Fixing nzbget units display (#27521)

This commit is contained in:
chriscla 2019-10-12 20:18:30 -07:00 committed by Paulus Schoutsen
parent df646f5db1
commit 6cbc9d6abb

View File

@ -34,9 +34,9 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
name = discovery_info["client_name"] name = discovery_info["client_name"]
devices = [] devices = []
for sensor_type, sensor_config in SENSOR_TYPES.items(): for sensor_config in SENSOR_TYPES.values():
new_sensor = NZBGetSensor( new_sensor = NZBGetSensor(
nzbget_data, sensor_type, name, sensor_config[0], sensor_config[1] nzbget_data, sensor_config[0], name, sensor_config[1], sensor_config[2]
) )
devices.append(new_sensor) devices.append(new_sensor)
@ -50,8 +50,8 @@ class NZBGetSensor(Entity):
self, nzbget_data, sensor_type, client_name, sensor_name, unit_of_measurement self, nzbget_data, sensor_type, client_name, sensor_name, unit_of_measurement
): ):
"""Initialize a new NZBGet sensor.""" """Initialize a new NZBGet sensor."""
self._name = f"{client_name} {sensor_type}" self._name = f"{client_name} {sensor_name}"
self.type = sensor_name self.type = sensor_type
self.client_name = client_name self.client_name = client_name
self.nzbget_data = nzbget_data self.nzbget_data = nzbget_data
self._state = None self._state = None