jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-09 10:59:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small entity attribute cleanup in AirVisual (#51601)

Browse Source 
* Small entity attribute cleanup in AirVisual

* Fix icon in sensor update
This commit is contained in:
Franck Nijhof
2021-06-08 08:29:04 +02:00
committed by GitHub
parent a4587b5f3b
commit 4007430d72
3 changed files with 7 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/airvisual/sensor.py
View File

@@ -154,12 +154,13 @@ class AirVisualGeographySensor(AirVisualEntity, SensorEntity):
}
)
self._config_entry = config_entry
self._icon = icon
self._kind = kind
self._locale = locale
self._name = name
self._state = None
self._unit = unit
self._attr_icon = icon
self._attr_unit_of_measurement = unit
@property
def available(self):
@@ -196,7 +197,7 @@ class AirVisualGeographySensor(AirVisualEntity, SensorEntity):
if self._kind == SENSOR_KIND_LEVEL:
aqi = data[f"aqi{self._locale}"]
self._state, self._icon = async_get_pollutant_level_info(aqi)
self._state, self._attr_icon = async_get_pollutant_level_info(aqi)
elif self._kind == SENSOR_KIND_AQI:
self._state = data[f"aqi{self._locale}"]
elif self._kind == SENSOR_KIND_POLLUTANT:
@@ -244,16 +245,12 @@ class AirVisualNodeProSensor(AirVisualEntity, SensorEntity):
"""Initialize."""
super().__init__(coordinator)
self._device_class = device_class
self._kind = kind
self._name = name
self._state = None
self._unit = unit
@property
def device_class(self):
"""Return the device class."""
return self._device_class
self._attr_device_class = device_class
self._attr_unit_of_measurement = unit
@property
def device_info(self):