mirror of
https://github.com/home-assistant/core.git
synced 2026-05-22 16:30:27 +01:00
epenet
57 lines
1.6 KiB
Python
57 lines
1.6 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
|
|
|
|
|
|
def get_dpcode(
|
|
device: CustomerDevice, dpcodes: str | tuple[str, ...] | None
|
|
) -> str | None:
|
|
"""Get the first matching DPCode from the device or return None."""
|
|
if dpcodes is None:
|
|
return None
|
|
|
|
if not isinstance(dpcodes, tuple):
|
|
dpcodes = (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
|
|
|
|
|
|
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())),
|
|
},
|
|
)
|