core/homeassistant/components/wiz/utils.py

26 lines
725 B
Python

"""WiZ utils."""
from __future__ import annotations
from pywizlight import BulbType
from pywizlight.bulblibrary import BulbClass
from .const import DEFAULT_NAME
def _short_mac(mac: str) -> str:
"""Get the short mac address from the full mac."""
return mac.replace(":", "").upper()[-6:]
def name_from_bulb_type_and_mac(bulb_type: BulbType, mac: str) -> str:
"""Generate a name from bulb_type and mac."""
if bulb_type.bulb_type == BulbClass.RGB:
if bulb_type.white_channels == 2:
description = "RGBWW Tunable"
else:
description = "RGBW Tunable"
else:
description = bulb_type.bulb_type.value
return f"{DEFAULT_NAME} {description} {_short_mac(mac)}"