from dataclasses import dataclass from enum import Enum class DeviceEntryType(Enum): SERVICE = "service" def DeviceInfo(**kwargs): return kwargs @dataclass class DeviceEntry: identifiers: set class DeviceRegistry: def __init__(self): self.devices = {} def async_get(self, device_id): return self.devices.get(device_id) REGISTRY = DeviceRegistry() def async_get(hass): return REGISTRY