Refactor and unify device fetching for UniFi Protect (#77341)

This commit is contained in:
Christopher Bailey 2022-08-26 07:46:11 -04:00 committed by GitHub
parent dfc3e7d80f
commit 1fb8fbf5de
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 50 additions and 44 deletions

View file

@ -10,6 +10,7 @@ from pyunifiprotect.data import (
Camera,
Event,
Light,
ModelType,
MountType,
ProtectAdoptableDeviceModel,
ProtectModelWithId,
@ -409,12 +410,9 @@ def _async_motion_entities(
) -> list[ProtectDeviceEntity]:
entities: list[ProtectDeviceEntity] = []
devices = (
data.api.bootstrap.cameras.values() if ufp_device is None else [ufp_device]
data.get_by_types({ModelType.CAMERA}) if ufp_device is None else [ufp_device]
)
for device in devices:
if not device.is_adopted:
continue
for description in MOTION_SENSORS:
entities.append(ProtectEventBinarySensor(data, device, description))
_LOGGER.debug(