hass-core/homeassistant/helpers/icon.py
J. Nick Koston e2e9c84c88
Cache construction of battery icon (#102194)
This was being built every time state was written. When
a robo vac is operating it writes state often which mean building
the icon string over and over again when it rarely changes.
2023-10-18 07:35:39 +02:00

34 lines
1 KiB
Python

"""Icon helper methods."""
from __future__ import annotations
from functools import lru_cache
@lru_cache
def icon_for_battery_level(
battery_level: int | None = None, charging: bool = False
) -> str:
"""Return a battery icon valid identifier."""
icon = "mdi:battery"
if battery_level is None:
return f"{icon}-unknown"
if charging and battery_level > 10:
icon += f"-charging-{int(round(battery_level / 20 - 0.01)) * 20}"
elif charging:
icon += "-outline"
elif battery_level <= 5:
icon += "-alert"
elif 5 < battery_level < 95:
icon += f"-{int(round(battery_level / 10 - 0.01)) * 10}"
return icon
def icon_for_signal_level(signal_level: int | None = None) -> str:
"""Return a signal icon valid identifier."""
if signal_level is None or signal_level == 0:
return "mdi:signal-cellular-outline"
if signal_level > 70:
return "mdi:signal-cellular-3"
if signal_level > 30:
return "mdi:signal-cellular-2"
return "mdi:signal-cellular-1"