Automatically add newly added devices for UniFi Protect (#73879)

homeassistant/components/unifiprotect/number.py

@@ -4,19 +4,27 @@ from __future__ import annotations
 from dataclasses import dataclass
 from datetime import timedelta
 
-from pyunifiprotect.data import Camera, Doorlock, Light, ProtectModelWithId
+from pyunifiprotect.data import (
+    Camera,
+    Doorlock,
+    Light,
+    ProtectAdoptableDeviceModel,
+    ProtectModelWithId,
+)
 
 from homeassistant.components.number import NumberEntity, NumberEntityDescription
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import PERCENTAGE, TIME_SECONDS
 from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.dispatcher import async_dispatcher_connect
 from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from .const import DOMAIN
+from .const import DISPATCH_ADOPT, DOMAIN
 from .data import ProtectData
 from .entity import ProtectDeviceEntity, async_all_device_entities
 from .models import PermRequired, ProtectSetableKeysMixin, T
+from .utils import async_dispatch_id as _ufpd
 
 
 @dataclass
@@ -184,6 +192,22 @@ async def async_setup_entry(
 ) -> None:
     """Set up number entities for UniFi Protect integration."""
     data: ProtectData = hass.data[DOMAIN][entry.entry_id]
+
+    async def _add_new_device(device: ProtectAdoptableDeviceModel) -> None:
+        entities = async_all_device_entities(
+            data,
+            ProtectNumbers,
+            camera_descs=CAMERA_NUMBERS,
+            light_descs=LIGHT_NUMBERS,
+            sense_descs=SENSE_NUMBERS,
+            lock_descs=DOORLOCK_NUMBERS,
+            chime_descs=CHIME_NUMBERS,
+            ufp_device=device,
+        )
+        async_add_entities(entities)
+
+    async_dispatcher_connect(hass, _ufpd(entry, DISPATCH_ADOPT), _add_new_device)
+
     entities: list[ProtectDeviceEntity] = async_all_device_entities(
         data,
         ProtectNumbers,