mirror of
https://github.com/home-assistant/core.git
synced 2026-05-29 19:57:40 +01:00
epenet
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
"""Utility methods for the Tuya integration."""
|
|
|
|
from tuya_device_handlers import TUYA_QUIRKS_REGISTRY
|
|
from tuya_sharing import CustomerDevice
|
|
|
|
from homeassistant.const import UnitOfTemperature
|
|
from homeassistant.exceptions import ServiceValidationError
|
|
from homeassistant.helpers.device_registry import DeviceInfo
|
|
|
|
from .const import CELSIUS_ALIASES, DOMAIN, FAHRENHEIT_ALIASES, DPCode
|
|
|
|
_TEMP_UNIT_CONVERT_MAPPING = {
|
|
"c": UnitOfTemperature.CELSIUS,
|
|
"f": UnitOfTemperature.FAHRENHEIT,
|
|
}
|
|
|
|
|
|
def get_temperature_unit(
|
|
device: CustomerDevice, dpcode_uom: str | None
|
|
) -> UnitOfTemperature | None:
|
|
"""Convert the DPCode unit of measurement to a temperature unit."""
|
|
if not dpcode_uom:
|
|
return get_device_temp_unit_convert(device)
|
|
|
|
dpcode_uom = dpcode_uom.lower()
|
|
if dpcode_uom in CELSIUS_ALIASES:
|
|
return UnitOfTemperature.CELSIUS
|
|
if dpcode_uom in FAHRENHEIT_ALIASES:
|
|
return UnitOfTemperature.FAHRENHEIT
|
|
return None
|
|
|
|
|
|
def get_device_temp_unit_convert(device: CustomerDevice) -> UnitOfTemperature | None:
|
|
"""Return the temperature unit from TEMP_UNIT_CONVERT, or None if unrecognised."""
|
|
if temp_unit_convert := device.status.get(DPCode.TEMP_UNIT_CONVERT):
|
|
return _TEMP_UNIT_CONVERT_MAPPING.get(temp_unit_convert)
|
|
return None
|
|
|
|
|
|
class ActionDPCodeNotFoundError(ServiceValidationError):
|
|
"""Custom exception for action DP code not found errors."""
|
|
|
|
def __init__(
|
|
self, device: CustomerDevice, expected: str | tuple[str, ...] | None
|
|
) -> None:
|
|
"""Initialize the error with device and expected DP codes."""
|
|
if expected is None:
|
|
expected = () # empty tuple for no expected codes
|
|
elif not isinstance(expected, tuple):
|
|
expected = (expected,)
|
|
|
|
super().__init__(
|
|
translation_domain=DOMAIN,
|
|
translation_key="action_dpcode_not_found",
|
|
translation_placeholders={
|
|
"expected": str(
|
|
sorted(
|
|
[dp.value if isinstance(dp, DPCode) else dp for dp in expected]
|
|
)
|
|
),
|
|
"available": str(sorted(device.function.keys())),
|
|
},
|
|
)
|
|
|
|
|
|
def get_device_info(device: CustomerDevice, *, initial: bool = False) -> DeviceInfo:
|
|
"""Get device info."""
|
|
manufacturer = "Tuya"
|
|
model: str | None = device.product_name
|
|
model_id: str | None = device.product_id
|
|
|
|
if initial:
|
|
# Note: the model is overridden via entity.device_info property
|
|
# when the entity is created. If no entities are generated, it will
|
|
# stay as unsupported
|
|
model = f"{device.product_name} (unsupported)"
|
|
|
|
if (
|
|
quirk := TUYA_QUIRKS_REGISTRY.get_quirk_for_device(device)
|
|
) and quirk.manufacturer:
|
|
# If the manufacturer is not set, we cannot trust the model/model_id
|
|
manufacturer = quirk.manufacturer
|
|
model = quirk.model
|
|
model_id = quirk.model_id
|
|
|
|
return DeviceInfo(
|
|
identifiers={(DOMAIN, device.id)},
|
|
manufacturer=manufacturer,
|
|
name=device.name,
|
|
model=model,
|
|
model_id=model_id,
|
|
)
|