mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-08-03 09:24:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fixed sensor icons

Browse source
This commit is contained in:
tomaae 2020-12-28 13:07:28 +01:00
parent e8c2c4bd1e
commit b026c18cbc
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
custom_components/mikrotik_router/sensor.py
View file

@ -358,7 +358,12 @@ class MikrotikControllerSensor(Entity):
self._device_class = None
self._state = None
self._icon = None
if ATTR_ICON in self._type:
self._icon = self._type[ATTR_ICON]
else:
self._icon = None
self._unit_of_measurement = None
self._attrs = {ATTR_ATTRIBUTION: ATTRIBUTION}
@ -384,10 +389,10 @@ class MikrotikControllerSensor(Entity):
@property
def icon(self) -> str:
"""Return the icon."""
if ATTR_ICON in self._type:
if self._icon:
return self._icon
else:
return ""
return ""
@property
def device_class(self) -> Optional[str]: