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

homeassistant-stubs/components/tplink/sensor.pyi

@@ -1,35 +1,27 @@
-from . import legacy_device_id as legacy_device_id
-from .const import ATTR_CURRENT_A as ATTR_CURRENT_A, ATTR_CURRENT_POWER_W as ATTR_CURRENT_POWER_W, ATTR_TODAY_ENERGY_KWH as ATTR_TODAY_ENERGY_KWH, ATTR_TOTAL_ENERGY_KWH as ATTR_TOTAL_ENERGY_KWH, DOMAIN as DOMAIN
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .const import UNIT_MAPPING as UNIT_MAPPING
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
-from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity
-from .models import TPLinkData as TPLinkData
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import ATTR_VOLTAGE as ATTR_VOLTAGE, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfPower as UnitOfPower
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from kasa import SmartDevice as SmartDevice
+from kasa import Device as Device, Feature
 
-@dataclass(frozen=True)
-class TPLinkSensorEntityDescription(SensorEntityDescription):
-    emeter_attr: str | None = ...
-    precision: int | 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, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, emeter_attr, precision) -> None: ...
+@dataclass(frozen=True, kw_only=True)
+class TPLinkSensorEntityDescription(SensorEntityDescription, TPLinkFeatureEntityDescription):
+    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, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
 
-ENERGY_SENSORS: tuple[TPLinkSensorEntityDescription, ...]
+SENSOR_DESCRIPTIONS: tuple[TPLinkSensorEntityDescription, ...]
+SENSOR_DESCRIPTIONS_MAP: Incomplete
 
-def async_emeter_from_device(device: SmartDevice, description: TPLinkSensorEntityDescription) -> float | None: ...
-def _async_sensors_for_device(device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, has_parent: bool = False) -> list[SmartPlugSensor]: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class SmartPlugSensor(CoordinatedTPLinkEntity, SensorEntity):
+class TPLinkSensorEntity(CoordinatedTPLinkFeatureEntity, SensorEntity):
     entity_description: TPLinkSensorEntityDescription
-    _attr_unique_id: Incomplete
-    _attr_translation_placeholders: Incomplete
-    _attr_translation_key: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkSensorEntityDescription, has_parent: bool = False) -> None: ...
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature: Feature, description: TPLinkSensorEntityDescription, parent: Device | None = None) -> None: ...
+    _attr_suggested_display_precision: Incomplete
     _attr_native_value: Incomplete
+    _attr_native_unit_of_measurement: Incomplete
     def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...

homeassistant-stubs/components/tplink/switch.pyi

@@ -1,48 +1,28 @@
-from . import legacy_device_id as legacy_device_id
-from .const import DOMAIN as DOMAIN
+from . import TPLinkConfigEntry as TPLinkConfigEntry
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
-from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity, async_refresh_after as async_refresh_after
-from .models import TPLinkData as TPLinkData
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription, async_refresh_after as async_refresh_after
 from _typeshed import Incomplete
-from homeassistant.components.switch import SwitchEntity as SwitchEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import EntityCategory as EntityCategory
+from dataclasses import dataclass
+from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from kasa import SmartDevice as SmartDevice, SmartPlug
+from kasa import Device as Device, Feature
 from typing import Any
 
 _LOGGER: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+@dataclass(frozen=True, kw_only=True)
+class TPLinkSwitchEntityDescription(SwitchEntityDescription, TPLinkFeatureEntityDescription):
+    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) -> None: ...
 
-class SmartPlugLedSwitch(CoordinatedTPLinkEntity, SwitchEntity):
-    device: SmartPlug
-    _attr_translation_key: str
-    _attr_entity_category: Incomplete
-    _attr_unique_id: Incomplete
-    def __init__(self, device: SmartPlug, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-    _attr_is_on: Incomplete
-    def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...
+SWITCH_DESCRIPTIONS: tuple[TPLinkSwitchEntityDescription, ...]
+SWITCH_DESCRIPTIONS_MAP: Incomplete
 
-class SmartPlugSwitch(CoordinatedTPLinkEntity, SwitchEntity):
-    _attr_name: str | None
-    _attr_unique_id: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-    _attr_is_on: Incomplete
-    def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class SmartPlugSwitchChild(SmartPlugSwitch):
-    _plug: Incomplete
-    _attr_unique_id: Incomplete
-    _attr_name: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, plug: SmartDevice) -> None: ...
+class TPLinkSwitch(CoordinatedTPLinkFeatureEntity, SwitchEntity):
+    entity_description: TPLinkSwitchEntityDescription
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature: Feature, description: TPLinkSwitchEntityDescription, parent: Device | None = None) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     _attr_is_on: Incomplete

homeassistant-stubs/components/tplink_omada/controller.pyi

@@ -1,25 +1,13 @@
-from .coordinator import OmadaCoordinator as OmadaCoordinator
+from .coordinator import OmadaClientsCoordinator as OmadaClientsCoordinator, OmadaGatewayCoordinator as OmadaGatewayCoordinator, OmadaSwitchPortCoordinator as OmadaSwitchPortCoordinator
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
 from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
-from tplink_omada_client.devices import OmadaGateway, OmadaSwitch as OmadaSwitch, OmadaSwitchPortDetails
-
-POLL_SWITCH_PORT: int
-POLL_GATEWAY: int
-
-class OmadaSwitchPortCoordinator(OmadaCoordinator[OmadaSwitchPortDetails]):
-    _network_switch: Incomplete
-    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, network_switch: OmadaSwitch) -> None: ...
-    async def poll_update(self) -> dict[str, OmadaSwitchPortDetails]: ...
-
-class OmadaGatewayCoordinator(OmadaCoordinator[OmadaGateway]):
-    mac: Incomplete
-    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, mac: str) -> None: ...
-    async def poll_update(self) -> dict[str, OmadaGateway]: ...
+from tplink_omada_client.devices import OmadaSwitch as OmadaSwitch
 
 class OmadaSiteController:
     _gateway_coordinator: OmadaGatewayCoordinator | None
     _initialized_gateway_coordinator: bool
+    _clients_coordinator: OmadaClientsCoordinator | None
     _hass: Incomplete
     _omada_client: Incomplete
     _switch_port_coordinators: Incomplete
@@ -28,3 +16,4 @@ class OmadaSiteController:
     def omada_client(self) -> OmadaSiteClient: ...
     def get_switch_port_coordinator(self, switch: OmadaSwitch) -> OmadaSwitchPortCoordinator: ...
     async def get_gateway_coordinator(self) -> OmadaGatewayCoordinator | None: ...
+    def get_clients_coordinator(self) -> OmadaClientsCoordinator: ...

homeassistant-stubs/components/tplink_omada/coordinator.pyi

@@ -1,12 +1,31 @@
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
+from tplink_omada_client import OmadaSiteClient as OmadaSiteClient, OmadaSwitchPortDetails
+from tplink_omada_client.clients import OmadaWirelessClient
+from tplink_omada_client.devices import OmadaGateway, OmadaSwitch as OmadaSwitch
 
 _LOGGER: Incomplete
+POLL_SWITCH_PORT: int
+POLL_GATEWAY: int
+POLL_CLIENTS: int
 
 class OmadaCoordinator(DataUpdateCoordinator[dict[str, _T]]):
     omada_client: Incomplete
     def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, name: str, poll_delay: int = 300) -> None: ...
     async def _async_update_data(self) -> dict[str, _T]: ...
     async def poll_update(self) -> dict[str, _T]: ...
+
+class OmadaSwitchPortCoordinator(OmadaCoordinator[OmadaSwitchPortDetails]):
+    _network_switch: Incomplete
+    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, network_switch: OmadaSwitch) -> None: ...
+    async def poll_update(self) -> dict[str, OmadaSwitchPortDetails]: ...
+
+class OmadaGatewayCoordinator(OmadaCoordinator[OmadaGateway]):
+    mac: Incomplete
+    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, mac: str) -> None: ...
+    async def poll_update(self) -> dict[str, OmadaGateway]: ...
+
+class OmadaClientsCoordinator(OmadaCoordinator[OmadaWirelessClient]):
+    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient) -> None: ...
+    async def poll_update(self) -> dict[str, OmadaWirelessClient]: ...

homeassistant-stubs/components/tplink_omada/device_tracker.pyi

@@ -0,0 +1,36 @@
+from .config_flow import CONF_SITE as CONF_SITE
+from .const import DOMAIN as DOMAIN
+from .controller import OmadaClientsCoordinator as OmadaClientsCoordinator, OmadaSiteController as OmadaSiteController
+from _typeshed import Incomplete
+from homeassistant.components.device_tracker import ScannerEntity as ScannerEntity, SourceType as SourceType
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+from tplink_omada_client.clients import OmadaWirelessClient
+
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class OmadaClientScannerEntity(CoordinatorEntity[OmadaClientsCoordinator], ScannerEntity):
+    _client_details: OmadaWirelessClient | None
+    _site_id: Incomplete
+    _client_id: Incomplete
+    _attr_name: Incomplete
+    def __init__(self, site_id: str, client_id: str, display_name: str, coordinator: OmadaClientsCoordinator) -> None: ...
+    @property
+    def source_type(self) -> SourceType: ...
+    def _do_update(self) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    def _handle_coordinator_update(self) -> None: ...
+    @property
+    def ip_address(self) -> str | None: ...
+    @property
+    def mac_address(self) -> str | None: ...
+    @property
+    def hostname(self) -> str | None: ...
+    @property
+    def is_connected(self) -> bool: ...
+    @property
+    def unique_id(self) -> str | None: ...

homeassistant-stubs/components/transmission/__init__.pyi

@@ -3,11 +3,12 @@ from .const import ATTR_DELETE_DATA as ATTR_DELETE_DATA, ATTR_TORRENT as ATTR_TO
 from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from .errors import AuthenticationError as AuthenticationError, CannotConnect as CannotConnect, UnknownError as UnknownError
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_ID as CONF_ID, CONF_NAME as CONF_NAME, CONF_PASSWORD as CONF_PASSWORD, CONF_PATH as CONF_PATH, CONF_PORT as CONF_PORT, CONF_SSL as CONF_SSL, CONF_USERNAME as CONF_USERNAME, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
-from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady, HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import selector as selector
+from homeassistant.helpers.typing import ConfigType as ConfigType
 from typing import Any
 
 _LOGGER: Incomplete
@@ -18,8 +19,13 @@ SERVICE_ADD_TORRENT_SCHEMA: Incomplete
 SERVICE_REMOVE_TORRENT_SCHEMA: Incomplete
 SERVICE_START_TORRENT_SCHEMA: Incomplete
 SERVICE_STOP_TORRENT_SCHEMA: Incomplete
+CONFIG_SCHEMA: Incomplete
+TransmissionConfigEntry = ConfigEntry[TransmissionDataUpdateCoordinator]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
 async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+def _get_coordinator_from_service_data(hass: HomeAssistant, entry_id: str) -> TransmissionDataUpdateCoordinator: ...
+def setup_hass_services(hass: HomeAssistant) -> None: ...
 async def get_api(hass: HomeAssistant, entry: dict[str, Any]) -> transmission_rpc.Client: ...

homeassistant-stubs/components/transmission/sensor.pyi

@@ -1,10 +1,10 @@
+from . import TransmissionConfigEntry as TransmissionConfigEntry
 from .const import DOMAIN as DOMAIN, STATE_ATTR_TORRENT_INFO as STATE_ATTR_TORRENT_INFO, STATE_DOWNLOADING as STATE_DOWNLOADING, STATE_SEEDING as STATE_SEEDING, STATE_UP_DOWN as STATE_UP_DOWN, SUPPORTED_ORDER_MODES as SUPPORTED_ORDER_MODES
 from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import STATE_IDLE as STATE_IDLE, UnitOfDataRate as UnitOfDataRate
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -24,7 +24,7 @@ class TransmissionSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_TYPES: tuple[TransmissionSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TransmissionSensor(CoordinatorEntity[TransmissionDataUpdateCoordinator], SensorEntity):
     entity_description: TransmissionSensorEntityDescription

homeassistant-stubs/components/transmission/switch.pyi

@@ -1,18 +1,16 @@
+from . import TransmissionConfigEntry as TransmissionConfigEntry
 from .const import DOMAIN as DOMAIN
 from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from typing import Any
 
-_LOGGING: Incomplete
-
 @dataclass(frozen=True, kw_only=True)
 class TransmissionSwitchEntityDescription(SwitchEntityDescription):
     is_on_func: Callable[[TransmissionDataUpdateCoordinator], bool | None]
@@ -22,7 +20,7 @@ class TransmissionSwitchEntityDescription(SwitchEntityDescription):
 
 SWITCH_TYPES: tuple[TransmissionSwitchEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TransmissionSwitch(CoordinatorEntity[TransmissionDataUpdateCoordinator], SwitchEntity):
     entity_description: TransmissionSwitchEntityDescription

homeassistant-stubs/components/trend/__init__.pyi

@@ -1,7 +1,8 @@
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import Platform as Platform
+from homeassistant.const import CONF_ENTITY_ID as CONF_ENTITY_ID, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device import async_remove_stale_devices_links_keep_entity_device as async_remove_stale_devices_links_keep_entity_device
 
 PLATFORMS: Incomplete
 

homeassistant-stubs/components/trend/binary_sensor.pyi

@@ -2,10 +2,12 @@ from . import PLATFORMS as PLATFORMS
 from .const import ATTR_GRADIENT as ATTR_GRADIENT, ATTR_INVERT as ATTR_INVERT, ATTR_MIN_GRADIENT as ATTR_MIN_GRADIENT, ATTR_SAMPLE_COUNT as ATTR_SAMPLE_COUNT, ATTR_SAMPLE_DURATION as ATTR_SAMPLE_DURATION, CONF_INVERT as CONF_INVERT, CONF_MAX_SAMPLES as CONF_MAX_SAMPLES, CONF_MIN_GRADIENT as CONF_MIN_GRADIENT, CONF_MIN_SAMPLES as CONF_MIN_SAMPLES, CONF_SAMPLE_DURATION as CONF_SAMPLE_DURATION, DEFAULT_MAX_SAMPLES as DEFAULT_MAX_SAMPLES, DEFAULT_MIN_GRADIENT as DEFAULT_MIN_GRADIENT, DEFAULT_MIN_SAMPLES as DEFAULT_MIN_SAMPLES, DEFAULT_SAMPLE_DURATION as DEFAULT_SAMPLE_DURATION, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, ENTITY_ID_FORMAT as ENTITY_ID_FORMAT, PLATFORM_SCHEMA as PLATFORM_SCHEMA
+from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, ENTITY_ID_FORMAT as ENTITY_ID_FORMAT
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, CONF_ATTRIBUTE as CONF_ATTRIBUTE, CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_FRIENDLY_NAME as CONF_FRIENDLY_NAME, CONF_SENSORS as CONF_SENSORS, STATE_ON as STATE_ON, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.device import async_device_info_to_link_from_entity as async_device_info_to_link_from_entity
 from homeassistant.helpers.entity import generate_entity_id as generate_entity_id
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.event import async_track_state_change_event as async_track_state_change_event
@@ -20,6 +22,7 @@ _LOGGER: Incomplete
 def _validate_min_max(data: dict[str, Any]) -> dict[str, Any]: ...
 
 SENSOR_SCHEMA: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
@@ -38,8 +41,9 @@ class SensorTrend(BinarySensorEntity, RestoreEntity):
     _attr_name: Incomplete
     _attr_device_class: Incomplete
     _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
     entity_id: Incomplete
-    def __init__(self, name: str, entity_id: str, attribute: str | None, invert: bool, sample_duration: int, min_gradient: float, min_samples: int, max_samples: int, unique_id: str | None = None, device_class: BinarySensorDeviceClass | None = None, sensor_entity_id: str | None = None) -> None: ...
+    def __init__(self, name: str, entity_id: str, attribute: str | None, invert: bool, sample_duration: int, min_gradient: float, min_samples: int, max_samples: int, unique_id: str | None = None, device_class: BinarySensorDeviceClass | None = None, sensor_entity_id: str | None = None, device_info: dr.DeviceInfo | None = None) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
     @property

homeassistant-stubs/components/tts/notify.pyi

@@ -1,6 +1,6 @@
 from . import ATTR_LANGUAGE as ATTR_LANGUAGE, ATTR_MEDIA_PLAYER_ENTITY_ID as ATTR_MEDIA_PLAYER_ENTITY_ID, ATTR_MESSAGE as ATTR_MESSAGE, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.components.notify import BaseNotificationService as BaseNotificationService, PLATFORM_SCHEMA as PLATFORM_SCHEMA
+from homeassistant.components.notify import BaseNotificationService as BaseNotificationService
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant, split_entity_id as split_entity_id
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
@@ -10,6 +10,7 @@ CONF_MEDIA_PLAYER: str
 CONF_TTS_SERVICE: str
 ENTITY_LEGACY_PROVIDER_GROUP: str
 _LOGGER: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 async def async_get_service(hass: HomeAssistant, config: ConfigType, discovery_info: DiscoveryInfoType | None = None) -> TTSNotificationService: ...
 

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -6,7 +6,7 @@ from _typeshed import Incomplete
 from aiounifi.interfaces.sites import Sites as Sites
 from collections.abc import Mapping
 from homeassistant.components import ssdp as ssdp
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import format_mac as format_mac

homeassistant-stubs/components/unifi/sensor.pyi

@@ -25,6 +25,7 @@ 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_device_outlet_power_supported_fn(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