mirror of
https://github.com/home-assistant/core.git
synced 2026-05-26 18:26:25 +01:00
epenet
105 lines
3.0 KiB
Python
105 lines
3.0 KiB
Python
"""Utility methods for the Tuya integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from tuya_sharing import CustomerDevice
|
|
|
|
from homeassistant.exceptions import ServiceValidationError
|
|
|
|
from .const import DOMAIN, DPCode, DPType
|
|
|
|
_DPTYPE_MAPPING: dict[str, DPType] = {
|
|
"bitmap": DPType.BITMAP,
|
|
"bool": DPType.BOOLEAN,
|
|
"enum": DPType.ENUM,
|
|
"json": DPType.JSON,
|
|
"raw": DPType.RAW,
|
|
"string": DPType.STRING,
|
|
"value": DPType.INTEGER,
|
|
}
|
|
|
|
|
|
def get_dpcode(
|
|
device: CustomerDevice, dpcodes: str | DPCode | tuple[DPCode, ...] | None
|
|
) -> DPCode | None:
|
|
"""Get the first matching DPCode from the device or return None."""
|
|
if dpcodes is None:
|
|
return None
|
|
|
|
if isinstance(dpcodes, DPCode):
|
|
dpcodes = (dpcodes,)
|
|
elif isinstance(dpcodes, str):
|
|
dpcodes = (DPCode(dpcodes),)
|
|
|
|
for dpcode in dpcodes:
|
|
if (
|
|
dpcode in device.function
|
|
or dpcode in device.status
|
|
or dpcode in device.status_range
|
|
):
|
|
return dpcode
|
|
|
|
return None
|
|
|
|
|
|
def get_dptype(
|
|
device: CustomerDevice, dpcode: DPCode | None, *, prefer_function: bool = False
|
|
) -> DPType | None:
|
|
"""Find a matching DPType type information for this device DPCode."""
|
|
if dpcode is None:
|
|
return None
|
|
|
|
lookup_tuple = (
|
|
(device.function, device.status_range)
|
|
if prefer_function
|
|
else (device.status_range, device.function)
|
|
)
|
|
|
|
for device_specs in lookup_tuple:
|
|
if current_definition := device_specs.get(dpcode):
|
|
current_type = current_definition.type
|
|
try:
|
|
return DPType(current_type)
|
|
except ValueError:
|
|
# Sometimes, we get ill-formed DPTypes from the cloud,
|
|
# this fixes them and maps them to the correct DPType.
|
|
return _DPTYPE_MAPPING.get(current_type)
|
|
|
|
return None
|
|
|
|
|
|
def remap_value(
|
|
value: float,
|
|
from_min: float = 0,
|
|
from_max: float = 255,
|
|
to_min: float = 0,
|
|
to_max: float = 255,
|
|
reverse: bool = False,
|
|
) -> float:
|
|
"""Remap a value from its current range, to a new range."""
|
|
if reverse:
|
|
value = from_max - value + from_min
|
|
return ((value - from_min) / (from_max - from_min)) * (to_max - to_min) + to_min
|
|
|
|
|
|
class ActionDPCodeNotFoundError(ServiceValidationError):
|
|
"""Custom exception for action DP code not found errors."""
|
|
|
|
def __init__(
|
|
self, device: CustomerDevice, expected: str | DPCode | tuple[DPCode, ...] | None
|
|
) -> None:
|
|
"""Initialize the error with device and expected DP codes."""
|
|
if expected is None:
|
|
expected = () # empty tuple for no expected codes
|
|
elif isinstance(expected, str):
|
|
expected = (DPCode(expected),)
|
|
|
|
super().__init__(
|
|
translation_domain=DOMAIN,
|
|
translation_key="action_dpcode_not_found",
|
|
translation_placeholders={
|
|
"expected": str(sorted([dp.value for dp in expected])),
|
|
"available": str(sorted(device.function.keys())),
|
|
},
|
|
)
|