ha-core/homeassistant/components/rfxtrx/helpers.py

25 lines
795 B
Python

"""Provides helpers for RFXtrx."""
from RFXtrx import RFXtrxDevice, get_device
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import device_registry as dr
from . import get_device_tuple_from_identifiers
@callback
def async_get_device_object(hass: HomeAssistant, device_id: str) -> RFXtrxDevice:
"""Get a device for the given device registry id."""
device_registry = dr.async_get(hass)
registry_device = device_registry.async_get(device_id)
if registry_device is None:
raise ValueError(f"Device {device_id} not found")
device_tuple = get_device_tuple_from_identifiers(registry_device.identifiers)
assert device_tuple
return get_device(
int(device_tuple[0], 16), int(device_tuple[1], 16), device_tuple[2]
)