|
7 | 7 | from aiohue.v2 import HueBridgeV2 |
8 | 8 | from aiohue.v2.controllers.events import EventType |
9 | 9 | from aiohue.v2.controllers.groups import Room, Zone |
10 | | -from aiohue.v2.models.device import Device, DeviceArchetypes |
| 10 | +from aiohue.v2.models.device import Device |
11 | 11 | from aiohue.v2.models.resource import ResourceTypes |
12 | 12 |
|
13 | 13 | from homeassistant.const import ( |
@@ -66,7 +66,7 @@ def add_device(hue_resource: Device | Room | Zone) -> dr.DeviceEntry: |
66 | 66 | } |
67 | 67 | if room := dev_controller.get_room(hue_resource.id): |
68 | 68 | params[ATTR_SUGGESTED_AREA] = room.metadata.name |
69 | | - if hue_resource.metadata.archetype == DeviceArchetypes.BRIDGE_V2: |
| 69 | + if hue_resource.id == api.config.bridge_device.id: |
70 | 70 | params[ATTR_IDENTIFIERS].add((DOMAIN, api.config.bridge_id)) |
71 | 71 | else: |
72 | 72 | params[ATTR_VIA_DEVICE] = (DOMAIN, api.config.bridge_device.id) |
@@ -97,9 +97,7 @@ def handle_device_event( |
97 | 97 | # create/update all current devices found in controllers |
98 | 98 | # sort the devices to ensure bridges are added first |
99 | 99 | hue_devices = list(dev_controller) |
100 | | - hue_devices.sort( |
101 | | - key=lambda dev: dev.metadata.archetype != DeviceArchetypes.BRIDGE_V2 |
102 | | - ) |
| 100 | + hue_devices.sort(key=lambda dev: dev.id != api.config.bridge_device.id) |
103 | 101 | known_devices = [add_device(hue_device) for hue_device in hue_devices] |
104 | 102 | known_devices += [add_device(hue_room) for hue_room in api.groups.room] |
105 | 103 | known_devices += [add_device(hue_zone) for hue_zone in api.groups.zone] |
|
0 commit comments