homeassistant-stubs 2024.6.3__py3-none-any.whl → 2024.7.0__py3-none-any.whl

homeassistant-stubs/components/unifi/sensor.pyi

@@ -25,8 +25,9 @@ def async_client_rx_value_fn(hub: UnifiHub, client: Client) -> float: ...
 def async_client_tx_value_fn(hub: UnifiHub, client: Client) -> float: ...
 def async_client_uptime_value_fn(hub: UnifiHub, client: Client) -> datetime: ...
 def async_wlan_client_value_fn(hub: UnifiHub, wlan: Wlan) -> int: ...
+def async_device_clients_value_fn(hub: UnifiHub, device: Device) -> int: ...
 def async_device_uptime_value_fn(hub: UnifiHub, device: Device) -> datetime | None: ...
-def async_device_uptime_value_changed_fn(old: StateType | date | datetime | Decimal, new: datetime | float | str | None) -> bool: ...
+def async_uptime_value_changed_fn(old: StateType | date | datetime | Decimal, new: datetime | float | str | None) -> bool: ...
 def async_device_outlet_power_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 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: ...

homeassistant-stubs/components/unifi/services.pyi

@@ -1,5 +1,6 @@
 from _typeshed import Incomplete
 from collections.abc import Mapping
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC

homeassistant-stubs/components/unifiprotect/__init__.pyi

@@ -1,12 +1,11 @@
-from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_ALLOW_EA as CONF_ALLOW_EA, DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN, MIN_REQUIRED_PROTECT_V as MIN_REQUIRED_PROTECT_V, OUTDATED_LOG_MESSAGE as OUTDATED_LOG_MESSAGE, PLATFORMS as PLATFORMS
-from .data import ProtectData as ProtectData, async_ufp_instance_for_config_entry_ids as async_ufp_instance_for_config_entry_ids
+from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_ALLOW_EA as CONF_ALLOW_EA, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN, MIN_REQUIRED_PROTECT_V as MIN_REQUIRED_PROTECT_V, OUTDATED_LOG_MESSAGE as OUTDATED_LOG_MESSAGE, PLATFORMS as PLATFORMS
+from .data import ProtectData as ProtectData, UFPConfigEntry as UFPConfigEntry
 from .discovery import async_start_discovery as async_start_discovery
 from .migrate import async_migrate_data as async_migrate_data
-from .services import async_cleanup_services as async_cleanup_services, async_setup_services as async_setup_services
+from .services import async_setup_services as async_setup_services
 from .utils import _async_unifi_mac_from_hass as _async_unifi_mac_from_hass, async_create_api_client as async_create_api_client, async_get_devices as async_get_devices
 from .views import ThumbnailProxyView as ThumbnailProxyView, VideoProxyView as VideoProxyView
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
@@ -22,8 +21,8 @@ CONFIG_SCHEMA: Incomplete
 EARLY_ACCESS_URL: str
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def _async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, data_service: ProtectData, bootstrap: Bootstrap) -> None: ...
-async def _async_options_updated(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: ConfigEntry, device_entry: dr.DeviceEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...
+async def _async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, data_service: ProtectData, bootstrap: Bootstrap) -> None: ...
+async def _async_options_updated(hass: HomeAssistant, entry: UFPConfigEntry) -> None: ...
+async def async_unload_entry(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: UFPConfigEntry, device_entry: dr.DeviceEntry) -> bool: ...

homeassistant-stubs/components/unifiprotect/binary_sensor.pyi

@@ -1,64 +1,72 @@
 import dataclasses
-from .const import DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
-from .entity import EventEntityMixin as EventEntityMixin, ProtectDeviceEntity as ProtectDeviceEntity, ProtectNVREntity as ProtectNVREntity, async_all_device_entities as async_all_device_entities
-from .models import PermRequired as PermRequired, ProtectEventMixin as ProtectEventMixin, ProtectRequiredKeysMixin as ProtectRequiredKeysMixin
+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 .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectEventMixin as ProtectEventMixin
 from _typeshed import Incomplete
+from collections.abc import Sequence
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 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 typing import Any
-from uiprotect.data import Camera, Light as Light, NVR as NVR, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId, Sensor
+from uiprotect.data import Camera, Light as Light, ModelType, NVR as NVR, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId, Sensor as Sensor
 from uiprotect.data.nvr import UOSDisk as UOSDisk
 
-_LOGGER: Incomplete
 _KEY_DOOR: str
 
 @dataclasses.dataclass(frozen=True, kw_only=True)
-class ProtectBinaryEntityDescription(ProtectRequiredKeysMixin, 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) -> None: ...
+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: ...
 
 @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, ufp_event_obj) -> 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, ...]
 LIGHT_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
+MOUNTABLE_SENSE_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
 SENSE_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
 EVENT_SENSORS: tuple[ProtectBinaryEventEntityDescription, ...]
 DOORLOCK_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
 VIEWER_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
 DISK_SENSORS: tuple[ProtectBinaryEntityDescription, ...]
-
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
-def _async_event_entities(data: ProtectData, ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[ProtectDeviceEntity]: ...
-def _async_nvr_entities(data: ProtectData) -> list[ProtectDeviceEntity]: ...
+_MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
+_MOUNTABLE_MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
 
 class ProtectDeviceBinarySensor(ProtectDeviceEntity, BinarySensorEntity):
     device: Camera | Light | Sensor
     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, ...]
     _attr_device_class: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
 
 class ProtectDiskBinarySensor(ProtectNVREntity, BinarySensorEntity):
     _disk: UOSDisk
     entity_description: ProtectBinaryEntityDescription
+    _state_attrs: Incomplete
     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_get_state_attrs(self) -> tuple[Any, ...]: ...
 
 class ProtectEventBinarySensor(EventEntityMixin, BinarySensorEntity):
     entity_description: ProtectBinaryEventEntityDescription
-    _attr_is_on: Incomplete
-    _event: Incomplete
+    _state_attrs: Incomplete
+    _attr_is_on: bool
     _attr_extra_state_attributes: Incomplete
+    def _set_event_done(self) -> None: ...
+    _event: Incomplete
+    _event_end: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
+
+MODEL_DESCRIPTIONS_WITH_CLASS: Incomplete
+
+def _async_event_entities(data: ProtectData, ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[ProtectDeviceEntity]: ...
+def _async_nvr_entities(data: ProtectData) -> list[BaseProtectEntity]: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...

homeassistant-stubs/components/unifiprotect/button.pyi

@@ -1,24 +1,24 @@
-from .const import DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DISPATCH_ADD as DISPATCH_ADD, DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .const import DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN
+from .data import UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity, async_all_device_entities as async_all_device_entities
-from .models import PermRequired as PermRequired, ProtectSetableKeysMixin as ProtectSetableKeysMixin, T as T
+from .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectSetableKeysMixin as ProtectSetableKeysMixin, T as T
 from _typeshed import Incomplete
+from collections.abc import Sequence
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform
 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 typing import Final
-from uiprotect.data import ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
+from uiprotect.data import ModelType, ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 _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, 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
@@ -26,14 +26,13 @@ ALL_DEVICE_BUTTONS: tuple[ProtectButtonEntityDescription, ...]
 ADOPT_BUTTON: Incomplete
 SENSOR_BUTTONS: tuple[ProtectButtonEntityDescription, ...]
 CHIME_BUTTONS: tuple[ProtectButtonEntityDescription, ...]
+_MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
 
 def _async_remove_adopt_button(hass: HomeAssistant, device: ProtectAdoptableDeviceModel) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class ProtectButton(ProtectDeviceEntity, ButtonEntity):
     entity_description: ProtectButtonEntityDescription
-    _attr_name: Incomplete
-    def __init__(self, data: ProtectData, device: ProtectAdoptableDeviceModel, description: ProtectButtonEntityDescription) -> None: ...
     _attr_available: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
     async def async_press(self) -> None: ...

homeassistant-stubs/components/unifiprotect/camera.pyi

@@ -1,27 +1,28 @@
-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, DISPATCH_ADOPT as DISPATCH_ADOPT, DISPATCH_CHANNELS as DISPATCH_CHANNELS, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+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 .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.config_entries import ConfigEntry as ConfigEntry
 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 import Any
+from typing_extensions import Generator
 from uiprotect.data import Camera as UFPCamera, CameraChannel as CameraChannel, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 _LOGGER: Incomplete
 
-def _create_rtsp_repair(hass: HomeAssistant, entry: ConfigEntry, data: ProtectData, camera: UFPCamera) -> None: ...
-def _get_camera_channels(hass: HomeAssistant, entry: ConfigEntry, data: ProtectData, ufp_device: UFPCamera | None = None) -> Generator[tuple[UFPCamera, CameraChannel, bool], None, None]: ...
-def _async_camera_entities(hass: HomeAssistant, entry: ConfigEntry, data: ProtectData, ufp_device: UFPCamera | None = None) -> list[ProtectDeviceEntity]: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+def _create_rtsp_repair(hass: HomeAssistant, entry: UFPConfigEntry, data: ProtectData, camera: UFPCamera) -> None: ...
+def _get_camera_channels(hass: HomeAssistant, entry: UFPConfigEntry, data: ProtectData, ufp_device: UFPCamera | None = None) -> Generator[tuple[UFPCamera, CameraChannel, bool]]: ...
+def _async_camera_entities(hass: HomeAssistant, entry: UFPConfigEntry, data: ProtectData, ufp_device: UFPCamera | None = None) -> list[ProtectDeviceEntity]: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+_EMPTY_CAMERA_FEATURES: Incomplete
 
 class ProtectCamera(ProtectDeviceEntity, Camera):
     device: UFPCamera
+    _state_attrs: Incomplete
     channel: Incomplete
     _secure: Incomplete
     _disable_stream: Incomplete
@@ -33,7 +34,6 @@ class ProtectCamera(ProtectDeviceEntity, Camera):
     _stream_source: Incomplete
     _attr_supported_features: Incomplete
     def _async_set_stream_source(self) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
     _attr_motion_detection_enabled: Incomplete
     _attr_is_recording: Incomplete
     _attr_available: Incomplete

homeassistant-stubs/components/unifiprotect/const.pyi

@@ -22,7 +22,6 @@ CONFIG_OPTIONS: Incomplete
 DEFAULT_PORT: int
 DEFAULT_ATTRIBUTION: str
 DEFAULT_BRAND: str
-DEFAULT_SCAN_INTERVAL: int
 DEFAULT_VERIFY_SSL: bool
 DEFAULT_MAX_MEDIA: int
 DEVICES_THAT_ADOPT: Incomplete

homeassistant-stubs/components/unifiprotect/data.pyi

@@ -1,50 +1,63 @@
 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, Generator, Iterable
+from collections.abc import Callable as Callable, 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_send as async_dispatcher_send
+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 Bootstrap as Bootstrap, ModelType, NVR, ProtectAdoptableDeviceModel, WSSubscriptionMessage as WSSubscriptionMessage
+from uiprotect.data import Camera, ModelType, NVR, ProtectAdoptableDeviceModel, WSSubscriptionMessage as WSSubscriptionMessage
+from uiprotect.websocket import WebsocketState
 
 _LOGGER: Incomplete
 ProtectDeviceType: Incomplete
+UFPConfigEntry = ConfigEntry[ProtectData]
 
-def async_last_update_was_successful(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+def async_last_update_was_successful(hass: HomeAssistant, entry: UFPConfigEntry) -> bool: ...
+def _async_dispatch_id(entry: UFPConfigEntry, dispatch: str) -> str: ...
 
 class ProtectData:
-    _hass: Incomplete
     _entry: Incomplete
+    _hass: Incomplete
     _update_interval: Incomplete
     _subscriptions: Incomplete
     _pending_camera_ids: Incomplete
-    _unsub_interval: Incomplete
-    _unsub_websocket: Incomplete
+    _unsubs: Incomplete
     _auth_failures: int
     last_update_success: bool
     api: Incomplete
-    def __init__(self, hass: HomeAssistant, protect: ProtectApiClient, update_interval: timedelta, entry: ConfigEntry) -> None: ...
+    adopt_signal: Incomplete
+    add_signal: Incomplete
+    channels_signal: Incomplete
+    def __init__(self, hass: HomeAssistant, protect: ProtectApiClient, update_interval: timedelta, entry: UFPConfigEntry) -> None: ...
     @property
     def disable_stream(self) -> bool: ...
     @property
     def max_events(self) -> int: ...
-    def get_by_types(self, device_types: Iterable[ModelType], ignore_unadopted: bool = True) -> Generator[ProtectAdoptableDeviceModel, None, None]: ...
-    async def async_setup(self) -> None: ...
+    def async_subscribe_adopt(self, add_callback: Callable[[ProtectAdoptableDeviceModel], None]) -> None: ...
+    def get_by_types(self, device_types: Iterable[ModelType], ignore_unadopted: bool = True) -> Generator[ProtectAdoptableDeviceModel]: ...
+    def get_cameras(self, ignore_unadopted: bool = True) -> Generator[Camera]: ...
+    def async_setup(self) -> None: ...
+    def _async_websocket_state_changed(self, state: WebsocketState) -> None: ...
+    def _async_update_change(self, success: bool, force_update: bool = False, exception: Exception | None = None) -> None: ...
     async def async_stop(self, *args: Any) -> None: ...
-    async def async_refresh(self, *_: Any, force: bool = False) -> None: ...
+    async def async_refresh(self) -> None: ...
     def async_add_pending_camera_id(self, camera_id: str) -> None: ...
     def _async_add_device(self, device: ProtectAdoptableDeviceModel) -> None: ...
     def _async_remove_device(self, device: ProtectAdoptableDeviceModel) -> None: ...
     def _async_update_device(self, device: ProtectAdoptableDeviceModel | NVR, changed_data: dict[str, Any]) -> None: ...
     def _async_process_ws_message(self, message: WSSubscriptionMessage) -> None: ...
-    def _async_process_updates(self, updates: Bootstrap | None) -> None: ...
+    def _async_process_updates(self) -> None: ...
     def _async_poll(self, now: datetime) -> None: ...
-    def async_subscribe_device_id(self, mac: str, update_callback: Callable[[ProtectDeviceType], None]) -> CALLBACK_TYPE: ...
-    def async_unsubscribe_device_id(self, mac: str, update_callback: Callable[[ProtectDeviceType], None]) -> None: ...
+    def async_subscribe(self, mac: str, update_callback: Callable[[ProtectDeviceType], None]) -> CALLBACK_TYPE: ...
+    def _async_unsubscribe(self, mac: str, update_callback: Callable[[ProtectDeviceType], None]) -> None: ...
     def _async_signal_device_update(self, device: ProtectDeviceType) -> None: ...
 
 def async_ufp_instance_for_config_entry_ids(hass: HomeAssistant, config_entry_ids: set[str]) -> ProtectApiClient | None: ...
+def async_get_ufp_entries(hass: HomeAssistant) -> list[UFPConfigEntry]: ...
+def async_get_data_for_nvr_id(hass: HomeAssistant, nvr_id: str) -> ProtectData | None: ...
+def async_get_data_for_entry_id(hass: HomeAssistant, entry_id: str) -> ProtectData | None: ...

homeassistant-stubs/components/unifiprotect/diagnostics.pyi

@@ -1,7 +1,5 @@
-from .const import DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from .data import UFPConfigEntry as UFPConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: UFPConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/unifiprotect/entity.pyi

@@ -1,51 +1,61 @@
 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 .models import PermRequired as PermRequired, ProtectEventMixin as ProtectEventMixin, ProtectRequiredKeysMixin as ProtectRequiredKeysMixin
+from .models import PermRequired as PermRequired, ProtectEntityDescription as ProtectEntityDescription, ProtectEventMixin as ProtectEventMixin
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Sequence
+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 homeassistant.helpers.typing import UNDEFINED as UNDEFINED
-from typing import Any
 from uiprotect.data import Event as Event, ModelType, NVR as NVR, ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 _LOGGER: Incomplete
 
-def _async_device_entities(data: ProtectData, klass: type[ProtectDeviceEntity], model_type: ModelType, descs: Sequence[ProtectRequiredKeysMixin], unadopted_descs: Sequence[ProtectRequiredKeysMixin], ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[ProtectDeviceEntity]: ...
-def async_all_device_entities(data: ProtectData, klass: type[ProtectDeviceEntity], camera_descs: Sequence[ProtectRequiredKeysMixin] | None = None, light_descs: Sequence[ProtectRequiredKeysMixin] | None = None, sense_descs: Sequence[ProtectRequiredKeysMixin] | None = None, viewer_descs: Sequence[ProtectRequiredKeysMixin] | None = None, lock_descs: Sequence[ProtectRequiredKeysMixin] | None = None, chime_descs: Sequence[ProtectRequiredKeysMixin] | None = None, all_descs: Sequence[ProtectRequiredKeysMixin] | None = None, unadopted_descs: Sequence[ProtectRequiredKeysMixin] | None = None, ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[ProtectDeviceEntity]: ...
+def _async_device_entities(data: ProtectData, klass: type[BaseProtectEntity], model_type: ModelType, descs: Sequence[ProtectEntityDescription], unadopted_descs: Sequence[ProtectEntityDescription] | None = None, ufp_device: ProtectAdoptableDeviceModel | None = None) -> list[BaseProtectEntity]: ...
 
-class ProtectDeviceEntity(Entity):
-    device: ProtectAdoptableDeviceModel
+_ALL_MODEL_TYPES: Incomplete
+
+def _combine_model_descs(model_type: ModelType, model_descriptions: dict[ModelType, Sequence[ProtectEntityDescription]] | None, all_descs: Sequence[ProtectEntityDescription] | None) -> list[ProtectEntityDescription]: ...
+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
     _attr_should_poll: bool
+    _attr_attribution = DEFAULT_ATTRIBUTION
+    _state_attrs: tuple[str, ...]
+    _attr_has_entity_name: bool
+    _async_get_ufp_enabled: Callable[[ProtectAdoptableDeviceModel], bool] | None
     data: Incomplete
-    _async_get_ufp_enabled: Incomplete
     _attr_unique_id: Incomplete
     _attr_name: Incomplete
     entity_description: Incomplete
-    _attr_attribution: Incomplete
-    def __init__(self, data: ProtectData, device: ProtectAdoptableDeviceModel, description: EntityDescription | None = None) -> None: ...
+    _state_getters: Incomplete
+    def __init__(self, data: ProtectData, device: ProtectAdoptableDeviceModel | NVR, 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_get_state_attrs(self) -> tuple[Any, ...]: ...
     def _async_updated_event(self, device: ProtectAdoptableDeviceModel | NVR) -> None: ...
     async def async_added_to_hass(self) -> None: ...
 
-class ProtectNVREntity(ProtectDeviceEntity):
+class ProtectDeviceEntity(BaseProtectEntity):
+    device: ProtectAdoptableDeviceModel
+
+class ProtectNVREntity(BaseProtectEntity):
     device: NVR
-    def __init__(self, entry: ProtectData, device: NVR, description: EntityDescription | None = None) -> 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: ...
 
 class EventEntityMixin(ProtectDeviceEntity):
-    _unrecorded_attributes: Incomplete
     entity_description: ProtectEventMixin
-    _event: Incomplete
-    def __init__(self, *args: Any, **kwarg: Any) -> None: ...
+    _unrecorded_attributes: Incomplete
+    _event: Event | None
+    _event_end: datetime | None
+    def _set_event_done(self) -> None: ...
     _attr_extra_state_attributes: Incomplete
-    def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
+    def _set_event_attrs(self, event: Event) -> None: ...
+    def _async_event_with_immediate_end(self) -> None: ...
+    def _event_already_ended(self, prev_event: Event | None, prev_event_end: datetime | None) -> bool: ...

homeassistant-stubs/components/unifiprotect/light.pyi

@@ -1,18 +1,15 @@
-from .const import DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .data import 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.config_entries import ConfigEntry as ConfigEntry
 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 typing import Any
 from uiprotect.data import Light as Light, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 _LOGGER: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 def unifi_brightness_to_hass(value: int) -> int: ...
 def hass_to_unifi_brightness(value: int) -> int: ...
 
@@ -21,7 +18,7 @@ class ProtectLight(ProtectDeviceEntity, LightEntity):
     _attr_icon: str
     _attr_color_mode: Incomplete
     _attr_supported_color_modes: Incomplete
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
+    _state_attrs: Incomplete
     _attr_is_on: Incomplete
     _attr_brightness: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...

homeassistant-stubs/components/unifiprotect/lock.pyi

@@ -1,25 +1,20 @@
-from .const import DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .data import 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.config_entries import ConfigEntry as ConfigEntry
 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 typing import Any
 from uiprotect.data import Doorlock, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 _LOGGER: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class ProtectLock(ProtectDeviceEntity, LockEntity):
     device: Doorlock
     entity_description: LockEntityDescription
-    _attr_name: Incomplete
-    def __init__(self, data: ProtectData, doorlock: Doorlock) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
+    _state_attrs: Incomplete
     _attr_is_locked: bool
     _attr_is_locking: bool
     _attr_is_unlocking: bool

homeassistant-stubs/components/unifiprotect/media_player.pyi

@@ -1,33 +1,29 @@
-from .const import DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .data import UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity
 from _typeshed import Incomplete
 from homeassistant.components import media_source as media_source
 from homeassistant.components.media_player import BrowseMedia as BrowseMedia, MediaPlayerDeviceClass as MediaPlayerDeviceClass, MediaPlayerEntity as MediaPlayerEntity, MediaPlayerEntityDescription as MediaPlayerEntityDescription, MediaPlayerEntityFeature as MediaPlayerEntityFeature, MediaPlayerState as MediaPlayerState, MediaType as MediaType, async_process_play_media_url as async_process_play_media_url
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
-from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
 from uiprotect.data import Camera, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 _LOGGER: Incomplete
+_SPEAKER_DESCRIPTION: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UFPConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class ProtectMediaPlayer(ProtectDeviceEntity, MediaPlayerEntity):
     device: Camera
     entity_description: MediaPlayerEntityDescription
     _attr_supported_features: Incomplete
-    _attr_name: Incomplete
     _attr_media_content_type: Incomplete
-    def __init__(self, data: ProtectData, camera: Camera) -> None: ...
+    _state_attrs: Incomplete
     _attr_volume_level: Incomplete
     _attr_state: Incomplete
     _attr_available: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
     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

@@ -1,5 +1,5 @@
 from .const import DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .data import ProtectData as ProtectData, async_get_ufp_entries as async_get_ufp_entries
 from .views import async_generate_event_video_url as async_generate_event_video_url, async_generate_thumbnail_url as async_generate_thumbnail_url
 from _typeshed import Incomplete
 from datetime import date, datetime, timedelta
@@ -11,7 +11,7 @@ from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers import entity_registry as er
 from typing import Any, NoReturn
-from uiprotect.data import Camera, Event, EventType
+from uiprotect.data import Camera as Camera, Event, EventType
 
 VIDEO_FORMAT: str
 THUMBNAIL_WIDTH: int
@@ -36,7 +36,6 @@ class IdentifierTimeType(str, Enum):
 EVENT_MAP: dict[SimpleEventType, set[EventType]]
 EVENT_NAME_MAP: Incomplete
 
-def get_ufp_event(event_type: SimpleEventType) -> set[EventType]: ...
 async def async_get_media_source(hass: HomeAssistant) -> MediaSource: ...
 def _get_month_start_end(start: datetime) -> tuple[datetime, datetime]: ...
 def _bad_identifier(identifier: str, err: Exception | None = None) -> NoReturn: ...

homeassistant-stubs/components/unifiprotect/migrate.pyi

@@ -1,8 +1,8 @@
 from .const import DOMAIN as DOMAIN
+from .data import UFPConfigEntry as UFPConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.automation import automations_with_entity as automations_with_entity
 from homeassistant.components.script import scripts_with_entity as scripts_with_entity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity
@@ -20,7 +20,7 @@ class EntityUsage(TypedDict):
     automations: dict[str, list[str]]
     scripts: dict[str, list[str]]
 
-def check_if_used(hass: HomeAssistant, entry: ConfigEntry, entities: dict[str, EntityRef]) -> dict[str, EntityUsage]: ...
-def create_repair_if_used(hass: HomeAssistant, entry: ConfigEntry, breaks_in: str, entities: dict[str, EntityRef]) -> None: ...
-async def async_migrate_data(hass: HomeAssistant, entry: ConfigEntry, protect: ProtectApiClient, bootstrap: Bootstrap) -> None: ...
-def async_deprecate_hdr_package(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+def check_if_used(hass: HomeAssistant, entry: UFPConfigEntry, entities: dict[str, EntityRef]) -> dict[str, EntityUsage]: ...
+def create_repair_if_used(hass: HomeAssistant, entry: UFPConfigEntry, breaks_in: str, entities: dict[str, EntityRef]) -> None: ...
+async def async_migrate_data(hass: HomeAssistant, entry: UFPConfigEntry, protect: ProtectApiClient, bootstrap: Bootstrap) -> None: ...
+def async_deprecate_hdr_package(hass: HomeAssistant, entry: UFPConfigEntry) -> None: ...