homeassistant-stubs 2024.6.4__py3-none-any.whl → 2024.7.0b0__py3-none-any.whl

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: ...

homeassistant-stubs/components/unifiprotect/models.pyi

@@ -1,45 +1,44 @@
-from .utils import get_nested_attr as get_nested_attr
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from enum import Enum
 from homeassistant.helpers.entity import EntityDescription as EntityDescription
 from typing import Any, Generic, TypeVar
-from uiprotect.data import Event as Event, NVR, ProtectAdoptableDeviceModel
+from uiprotect.data import Event as Event, NVR, ProtectAdoptableDeviceModel, SmartDetectObjectType as SmartDetectObjectType
 
 _LOGGER: Incomplete
 T = TypeVar('T', bound=ProtectAdoptableDeviceModel | NVR)
 
-def split_tuple(value: tuple[str, ...] | str | None) -> tuple[str, ...] | None: ...
-
 class PermRequired(int, Enum):
     NO_WRITE: int
     WRITE: int
     DELETE: int
 
 @dataclass(frozen=True, kw_only=True)
-class ProtectRequiredKeysMixin(EntityDescription, Generic[T]):
-    ufp_required_field: tuple[str, ...] | str | None = ...
-    ufp_value: tuple[str, ...] | str | None = ...
+class ProtectEntityDescription(EntityDescription, Generic[T]):
+    ufp_required_field: str | None = ...
+    ufp_value: str | None = ...
     ufp_value_fn: Callable[[T], Any] | None = ...
-    ufp_enabled: tuple[str, ...] | str | None = ...
+    ufp_enabled: str | None = ...
     ufp_perm: PermRequired | None = ...
-    def __post_init__(self) -> None: ...
+    has_required: Callable[[T], bool] = ...
+    get_ufp_enabled: Callable[[T], bool] | None = ...
     def get_ufp_value(self, obj: T) -> Any: ...
-    def get_ufp_enabled(self, obj: T) -> bool: ...
-    def has_required(self, obj: T) -> 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, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm) -> None: ...
+    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: ...
 
 @dataclass(frozen=True, kw_only=True)
-class ProtectEventMixin(ProtectRequiredKeysMixin[T]):
+class ProtectEventMixin(ProtectEntityDescription[T]):
     ufp_event_obj: str | None = ...
+    ufp_obj_type: SmartDetectObjectType | None = ...
     def get_event_obj(self, obj: T) -> Event | None: ...
-    def get_is_on(self, obj: T, event: Event | 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, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, ufp_event_obj) -> 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: ...
 
 @dataclass(frozen=True, kw_only=True)
-class ProtectSetableKeysMixin(ProtectRequiredKeysMixin[T]):
+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, 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,26 +1,21 @@
-from .const import DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN
-from .data import ProtectData as ProtectData
+from .data import ProtectData as ProtectData, 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.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 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.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 as Camera, Doorlock, Light, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
-
-_LOGGER: Incomplete
+from uiprotect.data import Camera as Camera, Doorlock, Light, ModelType, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId
 
 @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, 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: ...
@@ -33,12 +28,14 @@ LIGHT_NUMBERS: tuple[ProtectNumberEntityDescription, ...]
 SENSE_NUMBERS: tuple[ProtectNumberEntityDescription, ...]
 DOORLOCK_NUMBERS: tuple[ProtectNumberEntityDescription, ...]
 CHIME_NUMBERS: tuple[ProtectNumberEntityDescription, ...]
+_MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
 
-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 ProtectNumbers(ProtectDeviceEntity, NumberEntity):
     device: Camera | Light
     entity_description: ProtectNumberEntityDescription
+    _state_attrs: Incomplete
     _attr_native_max_value: Incomplete
     _attr_native_min_value: Incomplete
     _attr_native_step: Incomplete
@@ -46,4 +43,3 @@ class ProtectNumbers(ProtectDeviceEntity, NumberEntity):
     _attr_native_value: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
     async def async_set_native_value(self, value: float) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...

homeassistant-stubs/components/unifiprotect/repairs.pyi

@@ -1,4 +1,5 @@
 from .const import CONF_ALLOW_EA as CONF_ALLOW_EA
+from .data import UFPConfigEntry as UFPConfigEntry, async_get_data_for_entry_id as async_get_data_for_entry_id
 from .utils import async_create_api_client as async_create_api_client
 from homeassistant import data_entry_flow as data_entry_flow
 from homeassistant.components.repairs import ConfirmRepairFlow as ConfirmRepairFlow, RepairsFlow as RepairsFlow
@@ -9,8 +10,8 @@ from uiprotect.data import Bootstrap as Bootstrap, Camera as Camera
 
 class ProtectRepair(RepairsFlow):
     _api: ProtectApiClient
-    _entry: ConfigEntry
-    def __init__(self, *, api: ProtectApiClient, entry: ConfigEntry) -> None: ...
+    _entry: UFPConfigEntry
+    def __init__(self, *, api: ProtectApiClient, entry: UFPConfigEntry) -> None: ...
     def _async_get_placeholders(self) -> dict[str, str]: ...
 
 class EAConfirmRepair(ProtectRepair):
@@ -26,7 +27,7 @@ class RTSPRepair(ProtectRepair):
     _camera_id: str
     _camera: Camera | None
     _bootstrap: Bootstrap | None
-    def __init__(self, *, api: ProtectApiClient, entry: ConfigEntry, camera_id: str) -> None: ...
+    def __init__(self, *, api: ProtectApiClient, entry: UFPConfigEntry, camera_id: str) -> None: ...
     def _async_get_placeholders(self) -> dict[str, str]: ...
     async def _get_boostrap(self) -> Bootstrap: ...
     async def _get_camera(self) -> Camera: ...
@@ -35,4 +36,5 @@ class RTSPRepair(ProtectRepair):
     async def async_step_start(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
     async def async_step_confirm(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
 
+def _async_get_or_create_api_client(hass: HomeAssistant, entry: ConfigEntry) -> ProtectApiClient: ...
 async def async_create_fix_flow(hass: HomeAssistant, issue_id: str, data: dict[str, str | int | float | None] | None) -> RepairsFlow: ...

homeassistant-stubs/components/unifiprotect/select.pyi

@@ -1,21 +1,19 @@
-from .const import DISPATCH_ADOPT as DISPATCH_ADOPT, DOMAIN as DOMAIN, TYPE_EMPTY_VALUE as TYPE_EMPTY_VALUE
-from .data import ProtectData as ProtectData
+from .const import TYPE_EMPTY_VALUE as TYPE_EMPTY_VALUE
+from .data import ProtectData as ProtectData, 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 .utils import async_get_light_motion_current as async_get_light_motion_current
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable
+from collections.abc import Callable as Callable, Sequence
 from dataclasses import dataclass
 from enum import Enum
 from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
-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, Final
 from uiprotect.api import ProtectApiClient as ProtectApiClient
-from uiprotect.data import Camera, Doorlock, Light, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId, Sensor, Viewer
+from uiprotect.data import Camera, Doorlock, Light, ModelType, ProtectAdoptableDeviceModel as ProtectAdoptableDeviceModel, ProtectModelWithId as ProtectModelWithId, Sensor, Viewer
 
 _LOGGER: Incomplete
 _KEY_LIGHT_MOTION: str
@@ -38,7 +36,7 @@ class ProtectSelectEntityDescription(ProtectSetableKeysMixin[T], SelectEntityDes
     ufp_options: list[dict[str, Any]] | None = ...
     ufp_options_fn: Callable[[ProtectApiClient], list[dict[str, Any]]] | None = ...
     ufp_enum_type: type[Enum] | 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, options, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, ufp_set_method, ufp_set_method_fn, ufp_options, ufp_options_fn, ufp_enum_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, options, 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_options, ufp_options_fn, ufp_enum_type) -> None: ...
 
 def _get_viewer_options(api: ProtectApiClient) -> list[dict[str, Any]]: ...
 def _get_doorbell_options(api: ProtectApiClient) -> list[dict[str, Any]]: ...
@@ -55,13 +53,14 @@ LIGHT_SELECTS: tuple[ProtectSelectEntityDescription, ...]
 SENSE_SELECTS: tuple[ProtectSelectEntityDescription, ...]
 DOORLOCK_SELECTS: tuple[ProtectSelectEntityDescription, ...]
 VIEWER_SELECTS: tuple[ProtectSelectEntityDescription, ...]
+_MODEL_DESCRIPTIONS: dict[ModelType, Sequence[ProtectEntityDescription]]
 
-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 ProtectSelects(ProtectDeviceEntity, SelectEntity):
     device: Camera | Light | Viewer
     entity_description: ProtectSelectEntityDescription
-    _attr_name: Incomplete
+    _state_attrs: Incomplete
     def __init__(self, data: ProtectData, device: Camera | Light | Viewer, description: ProtectSelectEntityDescription) -> None: ...
     _attr_current_option: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
@@ -70,4 +69,3 @@ class ProtectSelects(ProtectDeviceEntity, SelectEntity):
     _unifi_to_hass_options: Incomplete
     def _async_set_options(self, data: ProtectData, description: ProtectSelectEntityDescription) -> None: ...
     async def async_select_option(self, option: str) -> None: ...
-    def _async_get_state_attrs(self) -> tuple[Any, ...]: ...