homeassistant-stubs 2024.7.3__py3-none-any.whl → 2024.8.0__py3-none-any.whl

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -9,6 +9,7 @@ from aiounifi.models.event import Event, EventKey
 from collections.abc import Callable as Callable, Mapping
 from dataclasses import dataclass
 from datetime import timedelta
+from functools import cached_property as cached_property
 from homeassistant.components.device_tracker import DOMAIN as DOMAIN, ScannerEntity as ScannerEntity, SourceType as SourceType
 from homeassistant.core import Event as core_Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -36,7 +37,7 @@ class UnifiTrackerEntityDescription(UnifiEntityDescription[HandlerT, ApiItemT]):
     ip_address_fn: Callable[[aiounifi.Controller, str], str | None]
     is_connected_fn: Callable[[UnifiHub, str], bool]
     hostname_fn: Callable[[aiounifi.Controller, str], str | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn=..., available_fn=..., name_fn=..., supported_fn=..., event_is_on=..., event_to_subscribe=..., should_poll=..., heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiTrackerEntityDescription, ...]
 
@@ -45,7 +46,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry,
 
 class UnifiScannerEntity(UnifiEntity[HandlerT, ApiItemT], ScannerEntity):
     entity_description: UnifiTrackerEntityDescription
-    _event_is_on: tuple[EventKey, ...]
+    _event_is_on: set[EventKey]
     _ignore_events: bool
     _is_connected: bool
     def async_initiate_state(self) -> None: ...
@@ -55,11 +56,11 @@ class UnifiScannerEntity(UnifiEntity[HandlerT, ApiItemT], ScannerEntity):
     def hostname(self) -> str | None: ...
     @property
     def ip_address(self) -> str | None: ...
-    @property
+    @cached_property
     def mac_address(self) -> str: ...
-    @property
+    @cached_property
     def source_type(self) -> SourceType: ...
-    @property
+    @cached_property
     def unique_id(self) -> str: ...
     def _make_disconnected(self, *_: core_Event) -> None: ...
     def async_update_state(self, event: ItemEvent, obj_id: str) -> None: ...

homeassistant-stubs/components/unifi/entity.pyi

@@ -35,16 +35,17 @@ class UnifiEntityDescription(EntityDescription, Generic[HandlerT, ApiItemT]):
     name_fn: Callable[[ApiItemT], str | None] = ...
     supported_fn: Callable[[UnifiHub, str], bool] = ...
     has_entity_name = ...
-    event_is_on: tuple[EventKey, ...] | None = ...
+    event_is_on: set[EventKey] | None = ...
     event_to_subscribe: tuple[EventKey, ...] | None = ...
     should_poll: bool = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn=..., available_fn=..., name_fn=..., supported_fn=..., event_is_on=..., event_to_subscribe=..., should_poll=...) -> None: ...
 
 class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
     entity_description: UnifiEntityDescription[HandlerT, ApiItemT]
     _attr_unique_id: str
     _obj_id: Incomplete
     hub: Incomplete
+    api: Incomplete
     _removed: bool
     _attr_available: Incomplete
     _attr_device_info: Incomplete

homeassistant-stubs/components/unifi/hub/entity_loader.pyi

@@ -2,21 +2,30 @@ from ..const import LOGGER as LOGGER, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_C
 from ..entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable, Coroutine, Sequence
 from homeassistant.const import Platform as Platform
 from homeassistant.core import callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+from typing import Any
 
 CHECK_HEARTBEAT_INTERVAL: Incomplete
+POLL_INTERVAL: Incomplete
 
 class UnifiEntityLoader:
     hub: Incomplete
     api_updaters: Incomplete
+    polling_api_updaters: Incomplete
     wireless_clients: Incomplete
+    _dataUpdateCoordinator: Incomplete
+    _update_listener: Incomplete
     platforms: Incomplete
     known_objects: Incomplete
     def __init__(self, hub: UnifiHub) -> None: ...
     async def initialize(self) -> None: ...
+    async def _refresh_data(self, updaters: Sequence[Callable[[], Coroutine[Any, Any, None]]]) -> None: ...
+    async def _update_pollable_api_data(self) -> None: ...
     async def _refresh_api_data(self) -> None: ...
     def _restore_inactive_clients(self) -> None: ...
     def register_platform(self, async_add_entities: AddEntitiesCallback, entity_class: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], requires_admin: bool = False) -> None: ...

homeassistant-stubs/components/unifi/image.pyi

@@ -18,7 +18,7 @@ def async_wlan_qr_code_image_fn(hub: UnifiHub, wlan: Wlan) -> bytes: ...
 class UnifiImageEntityDescription(ImageEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     image_fn: Callable[[UnifiHub, ApiItemT], bytes]
     value_fn: Callable[[ApiItemT], str | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, image_fn, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn=..., available_fn=..., name_fn=..., supported_fn=..., event_is_on=..., event_to_subscribe=..., should_poll=..., image_fn, value_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiImageEntityDescription, ...]
 

homeassistant-stubs/components/unifi/sensor.pyi

@@ -6,18 +6,20 @@ from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
 from aiounifi.models.client import Client
-from aiounifi.models.device import Device
+from aiounifi.models.device import Device, TypedDeviceUptimeStatsWanMonitor as TypedDeviceUptimeStatsWanMonitor
 from aiounifi.models.wlan import Wlan
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from datetime import date, datetime
 from decimal import Decimal
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass, UnitOfTemperature as UnitOfTemperature
-from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfDataRate as UnitOfDataRate, UnitOfPower as UnitOfPower
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfDataRate as UnitOfDataRate, UnitOfPower as UnitOfPower, UnitOfTime as UnitOfTime
 from homeassistant.core import Event as core_Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
+from homeassistant.util import slugify as slugify
+from typing import Literal
 
 def async_bandwidth_sensor_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_uptime_sensor_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
@@ -33,13 +35,17 @@ def async_device_outlet_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def device_system_stats_supported_fn(stat_index: int, hub: UnifiHub, obj_id: str) -> bool: ...
 def async_client_is_connected_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_device_state_value_fn(hub: UnifiHub, device: Device) -> str: ...
+def async_device_wan_latency_supported_fn(wan: Literal['WAN', 'WAN2'], monitor_target: str, hub: UnifiHub, obj_id: str) -> bool: ...
+def async_device_wan_latency_value_fn(wan: Literal['WAN', 'WAN2'], monitor_target: str, hub: UnifiHub, device: Device) -> int | None: ...
+def _device_wan_latency_monitor(wan: Literal['WAN', 'WAN2'], monitor_target: str, device: Device) -> TypedDeviceUptimeStatsWanMonitor | None: ...
+def make_wan_latency_sensors() -> tuple[UnifiSensorEntityDescription, ...]: ...
 
 @dataclass(frozen=True, kw_only=True)
 class UnifiSensorEntityDescription(SensorEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     value_fn: Callable[[UnifiHub, ApiItemT], datetime | float | str | None]
     is_connected_fn: Callable[[UnifiHub, str], bool] | None = ...
     value_changed_fn: Callable[[StateType | date | datetime | Decimal, datetime | float | str | None], bool] = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn, is_connected_fn, value_changed_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn=..., available_fn=..., name_fn=..., supported_fn=..., event_is_on=..., event_to_subscribe=..., should_poll=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn, is_connected_fn=..., value_changed_fn=...) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSensorEntityDescription, ...]
 

homeassistant-stubs/components/unifi/switch.pyi

@@ -23,13 +23,14 @@ CLIENT_UNBLOCKED: Incomplete
 def async_block_client_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_dpi_group_is_on_fn(hub: UnifiHub, dpi_group: DPIRestrictionGroup) -> bool: ...
 def async_dpi_group_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
-def async_port_forward_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+def async_unifi_network_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
 async def async_block_client_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_dpi_group_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 def async_outlet_switching_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 async def async_outlet_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_poe_port_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_port_forward_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_traffic_rule_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_wlan_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
@@ -38,7 +39,7 @@ class UnifiSwitchEntityDescription(SwitchEntityDescription, UnifiEntityDescripti
     is_on_fn: Callable[[UnifiHub, ApiItemT], bool]
     custom_subscribe: Callable[[aiounifi.Controller], SubscriptionT] | None = ...
     only_event_for_state_change: bool = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, control_fn, is_on_fn, custom_subscribe, only_event_for_state_change) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn=..., available_fn=..., name_fn=..., supported_fn=..., event_is_on=..., event_to_subscribe=..., should_poll=..., control_fn, is_on_fn, custom_subscribe=..., only_event_for_state_change=...) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSwitchEntityDescription, ...]
 

homeassistant-stubs/components/unifi/update.pyi

@@ -22,7 +22,7 @@ async def async_device_control_fn(api: aiounifi.Controller, obj_id: str) -> None
 class UnifiUpdateEntityDescription(UpdateEntityDescription, UnifiEntityDescription[_HandlerT, _DataT]):
     control_fn: Callable[[aiounifi.Controller, str], Coroutine[Any, Any, None]]
     state_fn: Callable[[aiounifi.Controller, _DataT], bool]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, control_fn, state_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn=..., available_fn=..., name_fn=..., supported_fn=..., event_is_on=..., event_to_subscribe=..., should_poll=..., control_fn, state_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiUpdateEntityDescription, ...]
 

homeassistant-stubs/components/unifiprotect/binary_sensor.pyi

@@ -1,6 +1,6 @@
 import dataclasses
-from .data import ProtectData as ProtectData, UFPConfigEntry as UFPConfigEntry
-from .entity import BaseProtectEntity as BaseProtectEntity, EventEntityMixin as EventEntityMixin, ProtectDeviceEntity as ProtectDeviceEntity, ProtectNVREntity as ProtectNVREntity, async_all_device_entities as async_all_device_entities
+from .data import ProtectData as ProtectData, ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
+from .entity import BaseProtectEntity as BaseProtectEntity, EventEntityMixin as EventEntityMixin, ProtectDeviceEntity as ProtectDeviceEntity, ProtectIsOnEntity as ProtectIsOnEntity, ProtectNVREntity as ProtectNVREntity, async_all_device_entities as async_all_device_entities
 from .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectEventMixin as ProtectEventMixin
 from _typeshed import Incomplete
 from collections.abc import Sequence
@@ -8,18 +8,18 @@ from homeassistant.components.binary_sensor import BinarySensorDeviceClass as Bi
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from uiprotect.data import Camera, Light as Light, ModelType, NVR as NVR, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId, Sensor as Sensor
+from uiprotect.data import ModelType, NVR as NVR, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, Sensor as Sensor
 from uiprotect.data.nvr import UOSDisk as UOSDisk
 
 _KEY_DOOR: str
 
 @dataclasses.dataclass(frozen=True, kw_only=True)
 class ProtectBinaryEntityDescription(ProtectEntityDescription, BinarySensorEntityDescription):
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=...) -> None: ...
 
 @dataclasses.dataclass(frozen=True, kw_only=True)
 class ProtectBinaryEventEntityDescription(ProtectEventMixin, BinarySensorEntityDescription):
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled, ufp_event_obj, ufp_obj_type) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=..., ufp_event_obj=..., ufp_obj_type=...) -> None: ...
 
 MOUNT_DEVICE_CLASS_MAP: Incomplete
 CAMERA_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
@@ -33,18 +33,14 @@ DISK_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
 _MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
 _MOUNTABLE_MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
 
-class ProtectDeviceBinarySensor(ProtectDeviceEntity, BinarySensorEntity):
-    device: Camera | Light | Sensor
+class ProtectDeviceBinarySensor(ProtectIsOnEntity, ProtectDeviceEntity, BinarySensorEntity):
     entity_description: ProtectBinaryEntityDescription
-    _state_attrs: tuple[str, ...]
-    _attr_is_on: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
 
 class MountableProtectDeviceBinarySensor(ProtectDeviceBinarySensor):
     device: Sensor
-    _state_attrs: tuple[str, ...]
+    _state_attrs: Incomplete
     _attr_device_class: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
 
 class ProtectDiskBinarySensor(ProtectNVREntity, BinarySensorEntity):
     _disk: UOSDisk
@@ -53,7 +49,7 @@ class ProtectDiskBinarySensor(ProtectNVREntity, BinarySensorEntity):
     def __init__(self, data: ProtectData, device: NVR, description: ProtectBinaryEntityDescription, disk: UOSDisk) -> None: ...
     _attr_available: bool
     _attr_is_on: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
 
 class ProtectEventBinarySensor(EventEntityMixin, BinarySensorEntity):
     entity_description: ProtectBinaryEventEntityDescription
@@ -63,7 +59,7 @@ class ProtectEventBinarySensor(EventEntityMixin, BinarySensorEntity):
     def _set_event_done(self) -> None: ...
     _event: Incomplete
     _event_end: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
 
 MODEL_DESCRIPTIONS_WITH_CLASS: Incomplete
 

homeassistant-stubs/components/unifiprotect/button.pyi

@@ -1,5 +1,5 @@
 from .const import DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN
-from .data import UFPConfigEntry as UFPConfigEntry
+from .data import ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity, async_all_device_entities as async_all_device_entities
 from .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectSetableKeysMixin as ProtectSetableKeysMixin, T as T
 from _typeshed import Incomplete
@@ -11,17 +11,16 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Final
-from uiprotect.data import ModelType, ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import ModelType, ProtectAdoptableDeviceModel
 
 _LOGGER: Incomplete
 
 @dataclass(frozen=True, kw_only=True)
 class ProtectButtonEntityDescription(ProtectSetableKeysMixin[T], ButtonEntityDescription):
     ufp_press: str | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled, ufp_set_method, ufp_set_method_fn, ufp_press) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=..., ufp_set_method=..., ufp_set_method_fn=..., ufp_press=...) -> None: ...
 
 DEVICE_CLASS_CHIME_BUTTON: Final[str]
-KEY_ADOPT: str
 ALL_DEVICE_BUTTONS: tuple[ProtectButtonEntityDescription, ...]
 ADOPT_BUTTON: Incomplete
 SENSOR_BUTTONS: tuple[ProtectButtonEntityDescription, ...]
@@ -33,6 +32,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_ad
 
 class ProtectButton(ProtectDeviceEntity, ButtonEntity):
     entity_description: ProtectButtonEntityDescription
-    _attr_available: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
     async def async_press(self) -> None: ...
+
+class ProtectAdoptButton(ProtectButton):
+    _attr_available: Incomplete
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...

homeassistant-stubs/components/unifiprotect/camera.pyi

@@ -1,15 +1,15 @@
 from .const import ATTR_BITRATE as ATTR_BITRATE, ATTR_CHANNEL_ID as ATTR_CHANNEL_ID, ATTR_FPS as ATTR_FPS, ATTR_HEIGHT as ATTR_HEIGHT, ATTR_WIDTH as ATTR_WIDTH, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData, UFPConfigEntry as UFPConfigEntry
+from .data import ProtectData as ProtectData, ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity
 from .utils import get_camera_base_name as get_camera_base_name
 from _typeshed import Incomplete
+from collections.abc import Generator
 from homeassistant.components.camera import Camera as Camera, CameraEntityFeature as CameraEntityFeature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity
-from typing_extensions import Generator
-from uiprotect.data import Camera as UFPCamera, CameraChannel as CameraChannel, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import Camera as UFPCamera, CameraChannel as CameraChannel, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel
 
 _LOGGER: Incomplete
 
@@ -38,7 +38,7 @@ class ProtectCamera(ProtectDeviceEntity, Camera):
     _attr_is_recording: Incomplete
     _attr_available: Incomplete
     _attr_extra_state_attributes: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
     async def async_camera_image(self, width: int | None = None, height: int | None = None) -> bytes | None: ...
     async def stream_source(self) -> str | None: ...
     async def async_enable_motion_detection(self) -> None: ...

homeassistant-stubs/components/unifiprotect/data.pyi

@@ -1,20 +1,19 @@
 from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_DISABLE_RTSP as CONF_DISABLE_RTSP, CONF_MAX_MEDIA as CONF_MAX_MEDIA, DEFAULT_MAX_MEDIA as DEFAULT_MAX_MEDIA, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DISPATCH_ADD as DISPATCH_ADD, DISPATCH_ADOPT as DISPATCH_ADOPT, DISPATCH_CHANNELS as DISPATCH_CHANNELS, DOMAIN as DOMAIN
 from .utils import async_get_devices_by_type as async_get_devices_by_type
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Iterable
+from collections.abc import Callable as Callable, Generator, Iterable
 from datetime import datetime, timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
 from typing import Any
-from typing_extensions import Generator
 from uiprotect import ProtectApiClient as ProtectApiClient
 from uiprotect.data import Camera, ModelType, NVR, ProtectAdoptableDeviceModel, WSSubscriptionMessage as WSSubscriptionMessage
 from uiprotect.websocket import WebsocketState
 
 _LOGGER: Incomplete
-ProtectDeviceType: Incomplete
+ProtectDeviceType = ProtectAdoptableDeviceModel | NVR
 UFPConfigEntry = ConfigEntry[ProtectData]
 
 def async_last_update_was_successful(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...

homeassistant-stubs/components/unifiprotect/entity.pyi

@@ -1,5 +1,5 @@
 from .const import ATTR_EVENT_ID as ATTR_EVENT_ID, ATTR_EVENT_SCORE as ATTR_EVENT_SCORE, DEFAULT_ATTRIBUTION as DEFAULT_ATTRIBUTION, DEFAULT_BRAND as DEFAULT_BRAND, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .data import ProtectData as ProtectData, ProtectDeviceType as ProtectDeviceType
 from .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectEventMixin as ProtectEventMixin
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Sequence
@@ -7,7 +7,7 @@ from datetime import datetime
 from homeassistant.core import callback as callback
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity import Entity as Entity, EntityDescription as EntityDescription
-from uiprotect.data import Event as Event, ModelType, NVR as NVR, ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import Event as Event, ModelType, NVR as NVR, ProtectAdoptableDeviceModel
 
 _LOGGER: Incomplete
 
@@ -19,7 +19,7 @@ def _combine_model_descs(model_type: ModelType, model_descriptions: dict[ModelTy
 def async_all_device_entities(data: ProtectData, klass: type[BaseProtectEntity], model_descriptions: dict[ModelType, Sequence[ProtectEntityDescription]] | None = None, all_descs: Sequence[ProtectEntityDescription] | None = None, unadopted_descs: list[ProtectEntityDescription] | None = None, ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[BaseProtectEntity]: ...
 
 class BaseProtectEntity(Entity):
-    device: ProtectAdoptableDeviceModel | NVR
+    device: ProtectDeviceType
     _attr_should_poll: bool
     _attr_attribution = DEFAULT_ATTRIBUTION
     _state_attrs: tuple[str, ...]
@@ -30,24 +30,28 @@ class BaseProtectEntity(Entity):
     _attr_name: Incomplete
     entity_description: Incomplete
     _state_getters: Incomplete
-    def __init__(self, data: ProtectData, device: ProtectAdoptableDeviceModel | NVR, description: EntityDescription | None = None) -> None: ...
+    def __init__(self, data: ProtectData, device: ProtectDeviceType, description: EntityDescription | None = None) -> None: ...
     async def async_update(self) -> None: ...
-    _attr_device_info: Incomplete
     def _async_set_device_info(self) -> None: ...
     _attr_available: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
-    def _async_updated_event(self, device: ProtectAdoptableDeviceModel | NVR) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
+    def _async_updated_event(self, device: ProtectDeviceType) -> None: ...
     async def async_added_to_hass(self) -> None: ...
 
+class ProtectIsOnEntity(BaseProtectEntity):
+    _state_attrs: tuple[str, ...]
+    _attr_is_on: bool | None
+    entity_description: ProtectEntityDescription
+    def _async_update_device_from_protect(self, device: ProtectAdoptableDeviceModel | NVR) -> None: ...
+
 class ProtectDeviceEntity(BaseProtectEntity):
-    device: ProtectAdoptableDeviceModel
+    _attr_device_info: Incomplete
+    def _async_set_device_info(self) -> None: ...
 
 class ProtectNVREntity(BaseProtectEntity):
     device: NVR
     _attr_device_info: Incomplete
     def _async_set_device_info(self) -> None: ...
-    _attr_available: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
 
 class EventEntityMixin(ProtectDeviceEntity):
     entity_description: ProtectEventMixin

homeassistant-stubs/components/unifiprotect/event.pyi

@@ -0,0 +1,25 @@
+import dataclasses
+from .const import ATTR_EVENT_ID as ATTR_EVENT_ID
+from .data import ProtectData as ProtectData, ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
+from .entity import EventEntityMixin as EventEntityMixin, ProtectDeviceEntity as ProtectDeviceEntity
+from .models import ProtectEventMixin as ProtectEventMixin
+from _typeshed import Incomplete
+from homeassistant.components.event import EventDeviceClass as EventDeviceClass, EventEntity as EventEntity, EventEntityDescription as EventEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from uiprotect.data import ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel
+
+@dataclasses.dataclass(frozen=True, kw_only=True)
+class ProtectEventEntityDescription(ProtectEventMixin, EventEntityDescription):
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., event_types=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=..., ufp_event_obj=..., ufp_obj_type=...) -> None: ...
+
+EVENT_DESCRIPTIONS: tuple[ProtectEventEntityDescription, ...]
+
+class ProtectDeviceEventEntity(EventEntityMixin, ProtectDeviceEntity, EventEntity):
+    entity_description: ProtectEventEntityDescription
+    _event: Incomplete
+    _event_end: Incomplete
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
+
+def _async_event_entities(data: ProtectData, ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[ProtectDeviceEntity]: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...

homeassistant-stubs/components/unifiprotect/light.pyi

@@ -1,11 +1,11 @@
-from .data import UFPConfigEntry as UFPConfigEntry
+from .data import ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity
 from _typeshed import Incomplete
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ColorMode as ColorMode, LightEntity as LightEntity
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
-from uiprotect.data import Light as Light, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import Light as Light, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel
 
 _LOGGER: Incomplete
 
@@ -21,6 +21,6 @@ class ProtectLight(ProtectDeviceEntity, LightEntity):
     _state_attrs: Incomplete
     _attr_is_on: Incomplete
     _attr_brightness: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/unifiprotect/lock.pyi

@@ -1,11 +1,11 @@
-from .data import UFPConfigEntry as UFPConfigEntry
+from .data import ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity
 from _typeshed import Incomplete
 from homeassistant.components.lock import LockEntity as LockEntity, LockEntityDescription as LockEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
-from uiprotect.data import Doorlock, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import Doorlock, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel
 
 _LOGGER: Incomplete
 
@@ -20,6 +20,6 @@ class ProtectLock(ProtectDeviceEntity, LockEntity):
     _attr_is_unlocking: bool
     _attr_is_jammed: bool
     _attr_available: bool
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
     async def async_unlock(self, **kwargs: Any) -> None: ...
     async def async_lock(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/unifiprotect/media_player.pyi

@@ -1,4 +1,4 @@
-from .data import UFPConfigEntry as UFPConfigEntry
+from .data import ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity
 from _typeshed import Incomplete
 from homeassistant.components import media_source as media_source
@@ -7,7 +7,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
-from uiprotect.data import Camera, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import Camera, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel
 
 _LOGGER: Incomplete
 _SPEAKER_DESCRIPTION: Incomplete
@@ -23,7 +23,7 @@ class ProtectMediaPlayer(ProtectDeviceEntity, MediaPlayerEntity):
     _attr_volume_level: Incomplete
     _attr_state: Incomplete
     _attr_available: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
     async def async_set_volume_level(self, volume: float) -> None: ...
     async def async_media_stop(self) -> None: ...
     async def async_play_media(self, media_type: MediaType | str, media_id: str, **kwargs: Any) -> None: ...

homeassistant-stubs/components/unifiprotect/media_source.pyi

@@ -18,20 +18,20 @@ THUMBNAIL_WIDTH: int
 THUMBNAIL_HEIGHT: int
 
 class SimpleEventType(str, Enum):
-    ALL: str
-    RING: str
-    MOTION: str
-    SMART: str
-    AUDIO: str
+    ALL = 'all'
+    RING = 'ring'
+    MOTION = 'motion'
+    SMART = 'smart'
+    AUDIO = 'audio'
 
 class IdentifierType(str, Enum):
-    EVENT: str
-    EVENT_THUMB: str
-    BROWSE: str
+    EVENT = 'event'
+    EVENT_THUMB = 'eventthumb'
+    BROWSE = 'browse'
 
 class IdentifierTimeType(str, Enum):
-    RECENT: str
-    RANGE: str
+    RECENT = 'recent'
+    RANGE = 'range'
 
 EVENT_MAP: dict[SimpleEventType, set[EventType]]
 EVENT_NAME_MAP: Incomplete

homeassistant-stubs/components/unifiprotect/models.pyi

@@ -10,9 +10,9 @@ _LOGGER: Incomplete
 T = TypeVar('T', bound=ProtectAdoptableDeviceModel | NVR)
 
 class PermRequired(int, Enum):
-    NO_WRITE: int
-    WRITE: int
-    DELETE: int
+    NO_WRITE = 1
+    WRITE = 2
+    DELETE = 3
 
 @dataclass(frozen=True, kw_only=True)
 class ProtectEntityDescription(EntityDescription, Generic[T]):
@@ -25,7 +25,7 @@ class ProtectEntityDescription(EntityDescription, Generic[T]):
     get_ufp_enabled: Callable[[T], bool] | None = ...
     def get_ufp_value(self, obj: T) -> Any: ...
     def __post_init__(self) -> None: ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=...) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class ProtectEventMixin(ProtectEntityDescription[T]):
@@ -34,11 +34,11 @@ class ProtectEventMixin(ProtectEntityDescription[T]):
     def get_event_obj(self, obj: T) -> Event | None: ...
     def has_matching_smart(self, event: Event) -> bool: ...
     def __post_init__(self) -> None: ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled, ufp_event_obj, ufp_obj_type) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=..., ufp_event_obj=..., ufp_obj_type=...) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class ProtectSetableKeysMixin(ProtectEntityDescription[T]):
     ufp_set_method: str | None = ...
     ufp_set_method_fn: Callable[[T, Any], Coroutine[Any, Any, None]] | None = ...
     async def ufp_set(self, obj: T, value: Any) -> None: ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled, ufp_set_method, ufp_set_method_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=..., ufp_set_method=..., ufp_set_method_fn=...) -> None: ...

homeassistant-stubs/components/unifiprotect/number.pyi

@@ -1,4 +1,4 @@
-from .data import ProtectData as ProtectData, UFPConfigEntry as UFPConfigEntry
+from .data import ProtectData as ProtectData, ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity, async_all_device_entities as async_all_device_entities
 from .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectSetableKeysMixin as ProtectSetableKeysMixin, T as T
 from _typeshed import Incomplete
@@ -8,14 +8,14 @@ from homeassistant.components.number import NumberEntity as NumberEntity, Number
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from uiprotect.data import Camera as Camera, Doorlock, Light, ModelType, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import Camera as Camera, Doorlock, Light, ModelType, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel
 
 @dataclass(frozen=True, kw_only=True)
 class ProtectNumberEntityDescription(ProtectSetableKeysMixin[T], NumberEntityDescription):
     ufp_max: int | float
     ufp_min: int | float
     ufp_step: int | float
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, has_required, get_ufp_enabled, ufp_set_method, ufp_set_method_fn, ufp_max, ufp_min, ufp_step) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., max_value=..., min_value=..., mode=..., native_max_value=..., native_min_value=..., native_step=..., native_unit_of_measurement=..., step=..., ufp_required_field=..., ufp_value=..., ufp_value_fn=..., ufp_enabled=..., ufp_perm=..., has_required=..., get_ufp_enabled=..., ufp_set_method=..., ufp_set_method_fn=..., ufp_max, ufp_min, ufp_step) -> None: ...
 
 def _get_pir_duration(obj: Light) -> int: ...
 async def _set_pir_duration(obj: Light, value: float) -> None: ...
@@ -41,5 +41,5 @@ class ProtectNumbers(ProtectDeviceEntity, NumberEntity):
     _attr_native_step: Incomplete
     def __init__(self, data: ProtectData, device: Camera | Light, description: ProtectNumberEntityDescription) -> None: ...
     _attr_native_value: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None: ...
     async def async_set_native_value(self, value: float) -> None: ...