homeassistant-stubs 2023.7.2__py3-none-any.whl → 2023.8.0__py3-none-any.whl

homeassistant-stubs/components/tractive/switch.pyi

@@ -24,7 +24,6 @@ SWITCH_TYPES: tuple[TractiveSwitchEntityDescription, ...]
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TractiveSwitch(TractiveEntity, SwitchEntity):
-    _attr_has_entity_name: bool
     entity_description: TractiveSwitchEntityDescription
     _attr_unique_id: Incomplete
     _attr_available: bool

homeassistant-stubs/components/tradfri/base_class.pyi

@@ -14,11 +14,11 @@ from typing import Any
 def handle_error(func: Callable[[Command | list[Command]], Any]) -> Callable[[Command | list[Command]], Coroutine[Any, Any, None]]: ...
 
 class TradfriBaseEntity(CoordinatorEntity[TradfriDeviceDataUpdateCoordinator], metaclass=abc.ABCMeta):
+    _attr_has_entity_name: bool
     _gateway_id: Incomplete
     _device: Incomplete
     _device_id: Incomplete
     _api: Incomplete
-    _attr_name: Incomplete
     _attr_unique_id: Incomplete
     def __init__(self, device_coordinator: TradfriDeviceDataUpdateCoordinator, gateway_id: str, api: Callable[[Command | list[Command]], Any]) -> None: ...
     @abstractmethod

homeassistant-stubs/components/tradfri/cover.pyi

@@ -13,6 +13,7 @@ from typing import Any
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TradfriCover(TradfriBaseEntity, CoverEntity):
+    _attr_name: Incomplete
     _device_control: Incomplete
     _device_data: Incomplete
     def __init__(self, device_coordinator: TradfriDeviceDataUpdateCoordinator, api: Callable[[Command | list[Command]], Any], gateway_id: str) -> None: ...

homeassistant-stubs/components/tradfri/fan.pyi

@@ -18,6 +18,7 @@ def _from_fan_speed(fan_speed: int) -> int: ...
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TradfriAirPurifierFan(TradfriBaseEntity, FanEntity):
+    _attr_name: Incomplete
     _attr_supported_features: Incomplete
     _device_control: Incomplete
     _device_data: Incomplete

homeassistant-stubs/components/tradfri/light.pyi

@@ -13,6 +13,7 @@ from typing import Any
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TradfriLight(TradfriBaseEntity, LightEntity):
+    _attr_name: Incomplete
     _attr_supported_features: Incomplete
     _device_control: Incomplete
     _device_data: Incomplete

homeassistant-stubs/components/tradfri/sensor.pyi

@@ -8,7 +8,6 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, PERCENTAGE as PERCENTAGE, Platform as Platform, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.typing import UNDEFINED as UNDEFINED
 from pytradfri.command import Command as Command
 from pytradfri.device import Device as Device
 from typing import Any
@@ -32,7 +31,6 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
 class TradfriSensor(TradfriBaseEntity, SensorEntity):
     entity_description: TradfriSensorEntityDescription
     _attr_unique_id: Incomplete
-    _attr_name: Incomplete
     def __init__(self, device_coordinator: TradfriDeviceDataUpdateCoordinator, api: Callable[[Command | list[Command]], Any], gateway_id: str, description: TradfriSensorEntityDescription) -> None: ...
     _attr_native_value: Incomplete
     def _refresh(self) -> None: ...

homeassistant-stubs/components/tradfri/switch.pyi

@@ -13,6 +13,7 @@ from typing import Any
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TradfriSwitch(TradfriBaseEntity, SwitchEntity):
+    _attr_name: Incomplete
     _device_control: Incomplete
     _device_data: Incomplete
     def __init__(self, device_coordinator: TradfriDeviceDataUpdateCoordinator, api: Callable[[Command | list[Command]], Any], gateway_id: str) -> None: ...

homeassistant-stubs/components/trafikverket_train/__init__.pyi

@@ -1,4 +1,5 @@
 from .const import CONF_FROM as CONF_FROM, CONF_TO as CONF_TO, DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
+from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/components/trafikverket_train/coordinator.pyi

@@ -0,0 +1,39 @@
+from .const import CONF_TIME as CONF_TIME, DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from datetime import date, datetime
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_WEEKDAY as CONF_WEEKDAY, WEEKDAYS as WEEKDAYS
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+from pytrafikverket.trafikverket_train import StationInfo as StationInfo, TrainStop as TrainStop
+
+class TrainData:
+    departure_time: datetime | None
+    departure_state: str
+    cancelled: bool
+    delayed_time: int | None
+    planned_time: datetime | None
+    estimated_time: datetime | None
+    actual_time: datetime | None
+    other_info: str | None
+    deviation: str | None
+    def __init__(self, departure_time, departure_state, cancelled, delayed_time, planned_time, estimated_time, actual_time, other_info, deviation) -> None: ...
+
+_LOGGER: Incomplete
+TIME_BETWEEN_UPDATES: Incomplete
+
+def _next_weekday(fromdate: date, weekday: int) -> date: ...
+def _next_departuredate(departure: list[str]) -> date: ...
+def _get_as_utc(date_value: datetime | None) -> datetime | None: ...
+def _get_as_joined(information: list[str] | None) -> str | None: ...
+
+class TVDataUpdateCoordinator(DataUpdateCoordinator[TrainData]):
+    _train_api: Incomplete
+    from_station: Incomplete
+    to_station: Incomplete
+    _time: Incomplete
+    _weekdays: Incomplete
+    def __init__(self, hass: HomeAssistant, entry: ConfigEntry, to_station: StationInfo, from_station: StationInfo) -> None: ...
+    async def _async_update_data(self) -> TrainData: ...

homeassistant-stubs/components/trafikverket_train/sensor.pyi

@@ -1,18 +1,18 @@
-from .const import CONF_FROM as CONF_FROM, CONF_TIME as CONF_TIME, CONF_TO as CONF_TO, DOMAIN as DOMAIN
+from .const import CONF_TIME as CONF_TIME, DOMAIN as DOMAIN
+from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from .util import create_unique_id as create_unique_id
 from _typeshed import Incomplete
-from datetime import date, datetime, time
+from datetime import time
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_NAME as CONF_NAME, CONF_WEEKDAY as CONF_WEEKDAY, WEEKDAYS as WEEKDAYS
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.const import CONF_NAME as CONF_NAME, CONF_WEEKDAY as CONF_WEEKDAY
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType
 from homeassistant.helpers.entity import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from pytrafikverket import TrafikverketTrain as TrafikverketTrain
-from pytrafikverket.trafikverket_train import StationInfo as StationInfo, TrainStop as TrainStop
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+from pytrafikverket.trafikverket_train import StationInfo as StationInfo
 
-_LOGGER: Incomplete
 ATTR_DEPARTURE_STATE: str
 ATTR_CANCELED: str
 ATTR_DELAY_TIME: str
@@ -25,24 +25,16 @@ ICON: str
 SCAN_INTERVAL: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
-def next_weekday(fromdate: date, weekday: int) -> date: ...
-def next_departuredate(departure: list[str]) -> date: ...
-def _to_iso_format(traintime: datetime) -> str: ...
 
-class TrainSensor(SensorEntity):
+class TrainSensor(CoordinatorEntity[TVDataUpdateCoordinator], SensorEntity):
     _attr_icon = ICON
     _attr_device_class: Incomplete
     _attr_has_entity_name: bool
-    _train_api: Incomplete
-    _from_station: Incomplete
-    _to_station: Incomplete
-    _weekday: Incomplete
-    _time: Incomplete
+    _attr_name: Incomplete
     _attr_device_info: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, train_api: TrafikverketTrain, name: str, from_station: StationInfo, to_station: StationInfo, weekday: list, departuretime: time | None, entry_id: str) -> None: ...
-    _attr_available: bool
+    def __init__(self, coordinator: TVDataUpdateCoordinator, name: str, from_station: StationInfo, to_station: StationInfo, weekday: list, departuretime: time | None, entry_id: str) -> None: ...
+    def _handle_coordinator_update(self) -> None: ...
     _attr_native_value: Incomplete
     _attr_extra_state_attributes: Incomplete
-    async def async_update(self) -> None: ...
-    def _update_attributes(self, state: TrainStop) -> None: ...
+    def _update_attr(self) -> None: ...

homeassistant-stubs/components/twentemilieu/sensor.pyi

@@ -1,4 +1,4 @@
-from .const import DOMAIN as DOMAIN, WASTE_TYPE_TO_DESCRIPTION as WASTE_TYPE_TO_DESCRIPTION
+from .const import DOMAIN as DOMAIN
 from .entity import TwenteMilieuEntity as TwenteMilieuEntity
 from _typeshed import Incomplete
 from datetime import date

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -37,7 +37,7 @@ class UnifiEntityTrackerDescriptionMixin(Generic[HandlerT, ApiItemT]):
     def __init__(self, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn) -> None: ...
 
 class UnifiTrackerEntityDescription(UnifiEntityDescription[HandlerT, ApiItemT], UnifiEntityTrackerDescriptionMixin[HandlerT, ApiItemT]):
-    def __init__(self, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
+    def __init__(self, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiTrackerEntityDescription, ...]
 

homeassistant-stubs/components/unifi/entity.pyi

@@ -1,6 +1,6 @@
 import abc
 import aiounifi
-from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER
+from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, DOMAIN as DOMAIN
 from .controller import UniFiController as UniFiController
 from _typeshed import Incomplete
 from abc import abstractmethod
@@ -9,7 +9,7 @@ from aiounifi.models.api import ApiItemT
 from aiounifi.models.event import Event as Event, EventKey as EventKey
 from collections.abc import Callable
 from homeassistant.core import callback as callback
-from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC
+from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceEntryType as DeviceEntryType
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity import DeviceInfo as DeviceInfo, Entity as Entity, EntityDescription as EntityDescription
 from typing import Generic, TypeVar
@@ -19,6 +19,7 @@ SubscriptionT = Callable[[CallbackType, ItemEvent], UnsubscribeType]
 
 def async_device_available_fn(controller: UniFiController, obj_id: str) -> bool: ...
 def async_device_device_info_fn(api: aiounifi.Controller, obj_id: str) -> DeviceInfo: ...
+def async_wlan_device_info_fn(api: aiounifi.Controller, obj_id: str) -> DeviceInfo: ...
 
 class UnifiDescription(Generic[HandlerT, ApiItemT]):
     allowed_fn: Callable[[UniFiController, str], bool]
@@ -29,22 +30,23 @@ class UnifiDescription(Generic[HandlerT, ApiItemT]):
     event_to_subscribe: tuple[EventKey, ...] | None
     name_fn: Callable[[ApiItemT], str | None]
     object_fn: Callable[[aiounifi.Controller, str], ApiItemT]
+    should_poll: bool
     supported_fn: Callable[[UniFiController, str], bool | None]
     unique_id_fn: Callable[[UniFiController, str], str]
-    def __init__(self, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, supported_fn, unique_id_fn) -> None: ...
+    def __init__(self, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn) -> None: ...
 
 class UnifiEntityDescription(EntityDescription, UnifiDescription[HandlerT, ApiItemT]):
-    def __init__(self, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
+    def __init__(self, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
 
 class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
     entity_description: UnifiEntityDescription[HandlerT, ApiItemT]
-    _attr_should_poll: bool
     _attr_unique_id: str
     _obj_id: Incomplete
     controller: Incomplete
     _removed: bool
     _attr_available: Incomplete
     _attr_device_info: Incomplete
+    _attr_should_poll: Incomplete
     _attr_name: Incomplete
     def __init__(self, obj_id: str, controller: UniFiController, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
     async def async_added_to_hass(self) -> None: ...
@@ -52,6 +54,7 @@ class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
     def async_signal_reachable_callback(self) -> None: ...
     async def async_signal_options_updated(self) -> None: ...
     async def remove_item(self, keys: set) -> None: ...
+    async def async_update(self) -> None: ...
     def async_initiate_state(self) -> None: ...
     @abstractmethod
     def async_update_state(self, event: ItemEvent, obj_id: str) -> None: ...

homeassistant-stubs/components/unifi/image.pyi

@@ -0,0 +1,37 @@
+from .controller import UniFiController as UniFiController
+from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_wlan_device_info_fn as async_wlan_device_info_fn
+from _typeshed import Incomplete
+from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
+from aiounifi.models.api import ApiItemT
+from aiounifi.models.wlan import Wlan
+from collections.abc import Callable as Callable
+from homeassistant.components.image import ImageEntity as ImageEntity, ImageEntityDescription as ImageEntityDescription
+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.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Generic
+
+def async_wlan_qr_code_image_fn(controller: UniFiController, wlan: Wlan) -> bytes: ...
+
+class UnifiImageEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
+    image_fn: Callable[[UniFiController, ApiItemT], bytes]
+    value_fn: Callable[[ApiItemT], str]
+    def __init__(self, image_fn, value_fn) -> None: ...
+
+class UnifiImageEntityDescription(ImageEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT], UnifiImageEntityDescriptionMixin[HandlerT, ApiItemT]):
+    def __init__(self, image_fn, value_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
+
+ENTITY_DESCRIPTIONS: tuple[UnifiImageEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class UnifiImageEntity(UnifiEntity[HandlerT, ApiItemT], ImageEntity):
+    entity_description: UnifiImageEntityDescription[HandlerT, ApiItemT]
+    _attr_content_type: str
+    current_image: bytes | None
+    previous_value: str
+    def __init__(self, obj_id: str, controller: UniFiController, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
+    def image(self) -> bytes | None: ...
+    _attr_image_last_updated: Incomplete
+    def async_update_state(self, event: ItemEvent, obj_id: str) -> None: ...

homeassistant-stubs/components/unifi/sensor.pyi

@@ -1,10 +1,11 @@
 import aiounifi
 from .controller import UniFiController as UniFiController
-from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn
+from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
 from aiounifi.models.client import Client
+from aiounifi.models.wlan import Wlan
 from collections.abc import Callable as Callable
 from datetime import datetime
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
@@ -19,6 +20,7 @@ from typing import Generic
 def async_client_rx_value_fn(controller: UniFiController, client: Client) -> float: ...
 def async_client_tx_value_fn(controller: UniFiController, client: Client) -> float: ...
 def async_client_uptime_value_fn(controller: UniFiController, client: Client) -> datetime: ...
+def async_wlan_client_value_fn(controller: UniFiController, wlan: Wlan) -> int: ...
 def async_client_device_info_fn(api: aiounifi.Controller, obj_id: str) -> DeviceInfo: ...
 
 class UnifiSensorEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
@@ -26,7 +28,7 @@ class UnifiSensorEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
     def __init__(self, value_fn) -> None: ...
 
 class UnifiSensorEntityDescription(SensorEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT], UnifiSensorEntityDescriptionMixin[HandlerT, ApiItemT]):
-    def __init__(self, value_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
+    def __init__(self, value_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSensorEntityDescription, ...]
 

homeassistant-stubs/components/unifi/switch.pyi

@@ -1,7 +1,7 @@
 import aiounifi
 from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER
 from .controller import UniFiController as UniFiController
-from .entity import HandlerT as HandlerT, SubscriptionT as SubscriptionT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn
+from .entity import HandlerT as HandlerT, SubscriptionT as SubscriptionT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent
 from aiounifi.models.api import ApiItemT
@@ -27,6 +27,7 @@ async def async_block_client_control_fn(api: aiounifi.Controller, obj_id: str, t
 async def async_dpi_group_control_fn(api: aiounifi.Controller, obj_id: str, target: bool) -> None: ...
 async def async_outlet_control_fn(api: aiounifi.Controller, obj_id: str, target: bool) -> None: ...
 async def async_poe_port_control_fn(api: aiounifi.Controller, obj_id: str, target: bool) -> None: ...
+async def async_wlan_control_fn(api: aiounifi.Controller, obj_id: str, target: bool) -> None: ...
 
 class UnifiSwitchEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
     control_fn: Callable[[aiounifi.Controller, str, bool], Coroutine[Any, Any, None]]
@@ -36,7 +37,7 @@ class UnifiSwitchEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
 class UnifiSwitchEntityDescription(SwitchEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT], UnifiSwitchEntityDescriptionMixin[HandlerT, ApiItemT]):
     custom_subscribe: Callable[[aiounifi.Controller], SubscriptionT] | None
     only_event_for_state_change: bool
-    def __init__(self, control_fn, is_on_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, custom_subscribe, only_event_for_state_change) -> None: ...
+    def __init__(self, control_fn, is_on_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement, custom_subscribe, only_event_for_state_change) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSwitchEntityDescription, ...]
 

homeassistant-stubs/components/unifi/update.pyi

@@ -24,7 +24,7 @@ class UnifiUpdateEntityDescriptionMixin(Generic[_HandlerT, _DataT]):
     def __init__(self, control_fn, state_fn) -> None: ...
 
 class UnifiUpdateEntityDescription(UpdateEntityDescription, UnifiEntityDescription[_HandlerT, _DataT], UnifiUpdateEntityDescriptionMixin[_HandlerT, _DataT]):
-    def __init__(self, control_fn, state_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
+    def __init__(self, control_fn, state_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, unit_of_measurement) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiUpdateEntityDescription, ...]
 

homeassistant-stubs/components/unifiprotect/entity.pyi

@@ -44,6 +44,5 @@ class EventEntityMixin(ProtectDeviceEntity):
     entity_description: ProtectEventMixin
     _event: Incomplete
     def __init__(self, *args: Any, **kwarg: Any) -> None: ...
-    def _async_event_extra_attrs(self) -> dict[str, Any]: ...
     _attr_extra_state_attributes: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...

homeassistant-stubs/components/unifiprotect/models.pyi

@@ -9,17 +9,20 @@ from typing import Any, Generic, TypeVar
 _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
 
 class ProtectRequiredKeysMixin(EntityDescription, Generic[T]):
-    ufp_required_field: str | None
-    ufp_value: str | None
+    ufp_required_field: tuple[str, ...] | str | None
+    ufp_value: tuple[str, ...] | str | None
     ufp_value_fn: Callable[[T], Any] | None
-    ufp_enabled: str | None
+    ufp_enabled: tuple[str, ...] | str | None
     ufp_perm: PermRequired | None
+    def __post_init__(self) -> None: ...
     def get_ufp_value(self, obj: T) -> Any: ...
     def get_ufp_enabled(self, obj: T) -> bool: ...
     def has_required(self, obj: T) -> bool: ...
@@ -28,7 +31,7 @@ class ProtectRequiredKeysMixin(EntityDescription, Generic[T]):
 class ProtectEventMixin(ProtectRequiredKeysMixin[T]):
     ufp_event_obj: str | None
     def get_event_obj(self, obj: T) -> Event | None: ...
-    def get_is_on(self, obj: T) -> bool: ...
+    def get_is_on(self, 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, unit_of_measurement, ufp_required_field, ufp_value, ufp_value_fn, ufp_enabled, ufp_perm, ufp_event_obj) -> None: ...
 
 class ProtectSetableKeysMixin(ProtectRequiredKeysMixin[T]):

homeassistant-stubs/components/unifiprotect/sensor.pyi

@@ -49,19 +49,16 @@ def _async_nvr_entities(data: ProtectData) -> list[ProtectDeviceEntity]: ...
 
 class ProtectDeviceSensor(ProtectDeviceEntity, SensorEntity):
     entity_description: ProtectSensorEntityDescription
-    def __init__(self, data: ProtectData, device: ProtectAdoptableDeviceModel, description: ProtectSensorEntityDescription) -> None: ...
     _attr_native_value: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
 
 class ProtectNVRSensor(ProtectNVREntity, SensorEntity):
     entity_description: ProtectSensorEntityDescription
-    def __init__(self, data: ProtectData, device: NVR, description: ProtectSensorEntityDescription) -> None: ...
     _attr_native_value: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
 
 class ProtectEventSensor(EventEntityMixin, SensorEntity):
     entity_description: ProtectSensorEventEntityDescription
-    def __init__(self, data: ProtectData, device: ProtectAdoptableDeviceModel, description: ProtectSensorEventEntityDescription) -> None: ...
     _attr_native_value: Incomplete
     _event: Incomplete
     _attr_extra_state_attributes: Incomplete

homeassistant-stubs/components/unifiprotect/utils.pyi

@@ -12,7 +12,7 @@ from typing import Any
 
 _SENTINEL: Incomplete
 
-def get_nested_attr(obj: Any, attr: str) -> Any: ...
+def get_nested_attr(obj: Any, attrs: tuple[str, ...]) -> Any: ...
 def _async_unifi_mac_from_hass(mac: str) -> str: ...
 def _async_short_mac(mac: str) -> str: ...
 async def _async_resolve(hass: HomeAssistant, host: str) -> str | None: ...

homeassistant-stubs/components/update/__init__.pyi

@@ -1,6 +1,6 @@
 from .const import ATTR_BACKUP as ATTR_BACKUP, ATTR_INSTALLED_VERSION as ATTR_INSTALLED_VERSION, ATTR_LATEST_VERSION as ATTR_LATEST_VERSION, ATTR_VERSION as ATTR_VERSION, DOMAIN as DOMAIN, SERVICE_INSTALL as SERVICE_INSTALL, SERVICE_SKIP as SERVICE_SKIP, UpdateEntityFeature as UpdateEntityFeature
 from _typeshed import Incomplete
-from homeassistant.backports.enum import StrEnum
+from enum import StrEnum
 from homeassistant.const import EntityCategory
 from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.entity import EntityDescription

homeassistant-stubs/components/uptimerobot/entity.pyi

@@ -8,6 +8,8 @@ from pyuptimerobot import UptimeRobotMonitor as UptimeRobotMonitor
 
 class UptimeRobotEntity(CoordinatorEntity[UptimeRobotDataUpdateCoordinator]):
     _attr_attribution = ATTRIBUTION
+    _attr_has_entity_name: bool
+    _attr_name: Incomplete
     entity_description: Incomplete
     _monitor: Incomplete
     _attr_device_info: Incomplete

homeassistant-stubs/components/vacuum/__init__.pyi

@@ -1,12 +1,14 @@
+import asyncio
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from enum import IntFlag
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_BATTERY_LEVEL as ATTR_BATTERY_LEVEL, ATTR_COMMAND as ATTR_COMMAND, SERVICE_TOGGLE as SERVICE_TOGGLE, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON, STATE_IDLE as STATE_IDLE, STATE_ON as STATE_ON, STATE_PAUSED as STATE_PAUSED
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE, make_entity_service_schema as make_entity_service_schema
 from homeassistant.helpers.entity import Entity as Entity, EntityDescription as EntityDescription, ToggleEntity as ToggleEntity, ToggleEntityDescription as ToggleEntityDescription
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
+from homeassistant.helpers.entity_platform import EntityPlatform as EntityPlatform
 from homeassistant.helpers.icon import icon_for_battery_level as icon_for_battery_level
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
@@ -111,6 +113,7 @@ class VacuumEntityDescription(ToggleEntityDescription):
     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, unit_of_measurement) -> None: ...
 
 class VacuumEntity(_BaseVacuum, ToggleEntity):
+    def add_to_platform_start(self, hass: HomeAssistant, platform: EntityPlatform, parallel_updates: asyncio.Semaphore | None) -> None: ...
     entity_description: VacuumEntityDescription
     _attr_status: str | None
     @property
@@ -125,8 +128,6 @@ class VacuumEntity(_BaseVacuum, ToggleEntity):
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     def start_pause(self, **kwargs: Any) -> None: ...
     async def async_start_pause(self, **kwargs: Any) -> None: ...
-    async def async_pause(self) -> None: ...
-    async def async_start(self) -> None: ...
 
 class StateVacuumEntityDescription(EntityDescription):
     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, unit_of_measurement) -> None: ...
@@ -142,6 +143,3 @@ class StateVacuumEntity(_BaseVacuum):
     async def async_start(self) -> None: ...
     def pause(self) -> None: ...
     async def async_pause(self) -> None: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-    async def async_toggle(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/vallox/__init__.pyi

@@ -55,6 +55,7 @@ class ValloxServiceHandler:
     async def async_handle(self, call: ServiceCall) -> None: ...
 
 class ValloxEntity(CoordinatorEntity[ValloxDataUpdateCoordinator]):
+    _attr_has_entity_name: bool
     _device_uuid: Incomplete
     _attr_device_info: Incomplete
     def __init__(self, name: str, coordinator: ValloxDataUpdateCoordinator) -> None: ...

homeassistant-stubs/components/vallox/binary_sensor.pyi

@@ -10,7 +10,6 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 class ValloxBinarySensorEntity(ValloxEntity, BinarySensorEntity):
     entity_description: ValloxBinarySensorEntityDescription
     _attr_entity_category: Incomplete
-    _attr_has_entity_name: bool
     _attr_unique_id: Incomplete
     def __init__(self, name: str, coordinator: ValloxDataUpdateCoordinator, description: ValloxBinarySensorEntityDescription) -> None: ...
     @property

homeassistant-stubs/components/vallox/fan.pyi

@@ -21,7 +21,7 @@ def _convert_to_int(value: StateType) -> int | None: ...
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class ValloxFanEntity(ValloxEntity, FanEntity):
-    _attr_has_entity_name: bool
+    _attr_name: Incomplete
     _attr_supported_features: Incomplete
     _client: Incomplete
     _attr_unique_id: Incomplete

homeassistant-stubs/components/vallox/number.pyi

@@ -10,7 +10,6 @@ from vallox_websocket_api import Vallox as Vallox
 
 class ValloxNumberEntity(ValloxEntity, NumberEntity):
     entity_description: ValloxNumberEntityDescription
-    _attr_has_entity_name: bool
     _attr_entity_category: Incomplete
     _attr_unique_id: Incomplete
     _client: Incomplete

homeassistant-stubs/components/vallox/sensor.pyi

@@ -12,7 +12,6 @@ from homeassistant.helpers.typing import StateType as StateType
 class ValloxSensorEntity(ValloxEntity, SensorEntity):
     entity_description: ValloxSensorEntityDescription
     _attr_entity_category: Incomplete
-    _attr_has_entity_name: bool
     _attr_unique_id: Incomplete
     def __init__(self, name: str, coordinator: ValloxDataUpdateCoordinator, description: ValloxSensorEntityDescription) -> None: ...
     @property

homeassistant-stubs/components/vallox/switch.pyi

@@ -12,7 +12,6 @@ from vallox_websocket_api import Vallox as Vallox
 class ValloxSwitchEntity(ValloxEntity, SwitchEntity):
     entity_description: ValloxSwitchEntityDescription
     _attr_entity_category: Incomplete
-    _attr_has_entity_name: bool
     _attr_unique_id: Incomplete
     _client: Incomplete
     def __init__(self, name: str, coordinator: ValloxDataUpdateCoordinator, description: ValloxSwitchEntityDescription, client: Vallox) -> None: ...

homeassistant-stubs/components/vlc_telnet/media_player.pyi

@@ -21,10 +21,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 def catch_vlc_errors(func: Callable[Concatenate[_VlcDeviceT, _P], Awaitable[None]]) -> Callable[Concatenate[_VlcDeviceT, _P], Coroutine[Any, Any, None]]: ...
 
 class VlcDevice(MediaPlayerEntity):
+    _attr_has_entity_name: bool
+    _attr_name: Incomplete
     _attr_media_content_type: Incomplete
     _attr_supported_features: Incomplete
     _config_entry: Incomplete
-    _name: Incomplete
     _volume: Incomplete
     _muted: Incomplete
     _media_position_updated_at: Incomplete
@@ -43,8 +44,6 @@ class VlcDevice(MediaPlayerEntity):
     _attr_media_album_name: Incomplete
     async def async_update(self) -> None: ...
     @property
-    def name(self) -> str: ...
-    @property
     def available(self) -> bool: ...
     @property
     def volume_level(self) -> float | None: ...

homeassistant-stubs/components/wallbox/const.pyi

@@ -1,5 +1,5 @@
 from _typeshed import Incomplete
-from homeassistant.backports.enum import StrEnum as StrEnum
+from enum import StrEnum
 
 DOMAIN: str
 BIDIRECTIONAL_MODEL_PREFIXES: Incomplete

homeassistant-stubs/components/water_heater/__init__.pyi

@@ -30,6 +30,7 @@ class WaterHeaterEntityFeature(IntFlag):
     TARGET_TEMPERATURE: int
     OPERATION_MODE: int
     AWAY_MODE: int
+    ON_OFF: int
 
 SUPPORT_TARGET_TEMPERATURE: int
 SUPPORT_OPERATION_MODE: int
@@ -97,6 +98,10 @@ class WaterHeaterEntity(Entity):
     def is_away_mode_on(self) -> bool | None: ...
     def set_temperature(self, **kwargs: Any) -> None: ...
     async def async_set_temperature(self, **kwargs: Any) -> None: ...
+    def turn_on(self, **kwargs: Any) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+    def turn_off(self, **kwargs: Any) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
     def set_operation_mode(self, operation_mode: str) -> None: ...
     async def async_set_operation_mode(self, operation_mode: str) -> None: ...
     def turn_away_mode_on(self) -> None: ...