homeassistant-stubs 2024.4.4__py3-none-any.whl → 2024.5.0__py3-none-any.whl

homeassistant-stubs/components/ring/switch.pyi

@@ -0,0 +1,31 @@
+from . import RingData as RingData
+from .const import DOMAIN as DOMAIN
+from .coordinator import RingDataCoordinator as RingDataCoordinator
+from .entity import RingEntity as RingEntity, exception_wrap as exception_wrap
+from _typeshed import Incomplete
+from homeassistant.components.switch import SwitchEntity as SwitchEntity
+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 ring_doorbell import RingStickUpCam
+from typing import Any
+
+_LOGGER: Incomplete
+SKIP_UPDATES_DELAY: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class BaseRingSwitch(RingEntity[RingStickUpCam], SwitchEntity):
+    _device_type: Incomplete
+    _attr_unique_id: Incomplete
+    def __init__(self, device: RingStickUpCam, coordinator: RingDataCoordinator, device_type: str) -> None: ...
+
+class SirenSwitch(BaseRingSwitch):
+    _attr_translation_key: str
+    _no_updates_until: Incomplete
+    _attr_is_on: Incomplete
+    def __init__(self, device: RingStickUpCam, coordinator: RingDataCoordinator) -> None: ...
+    def _handle_coordinator_update(self) -> None: ...
+    def _set_switch(self, new_state: int) -> None: ...
+    def turn_on(self, **kwargs: Any) -> None: ...
+    def turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/romy/binary_sensor.pyi

@@ -0,0 +1,19 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import RomyVacuumCoordinator as RomyVacuumCoordinator
+from .entity import RomyEntity as RomyEntity
+from _typeshed import Incomplete
+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.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+BINARY_SENSORS: list[BinarySensorEntityDescription]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RomyBinarySensor(RomyEntity, BinarySensorEntity):
+    entity_description: BinarySensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: RomyVacuumCoordinator, entity_description: BinarySensorEntityDescription) -> None: ...
+    @property
+    def is_on(self) -> bool: ...

homeassistant-stubs/components/romy/config_flow.pyi

@@ -1,16 +1,16 @@
 from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
-from homeassistant import config_entries as config_entries
 from homeassistant.components import zeroconf as zeroconf
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD
 
-class RomyConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
+class RomyConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
     host: str
     password: str
     robot_name_given_by_user: str
     def __init__(self) -> None: ...
-    async def async_step_user(self, user_input: dict[str, str] | None = None) -> config_entries.ConfigFlowResult: ...
-    async def async_step_password(self, user_input: dict[str, str] | None = None) -> config_entries.ConfigFlowResult: ...
-    async def async_step_zeroconf(self, discovery_info: zeroconf.ZeroconfServiceInfo) -> config_entries.ConfigFlowResult: ...
-    async def async_step_zeroconf_confirm(self, user_input: dict[str, str] | None = None) -> config_entries.ConfigFlowResult: ...
-    async def _async_step_finish_config(self) -> config_entries.ConfigFlowResult: ...
+    async def async_step_user(self, user_input: dict[str, str] | None = None) -> ConfigFlowResult: ...
+    async def async_step_password(self, user_input: dict[str, str] | None = None) -> ConfigFlowResult: ...
+    async def async_step_zeroconf(self, discovery_info: zeroconf.ZeroconfServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_zeroconf_confirm(self, user_input: dict[str, str] | None = None) -> ConfigFlowResult: ...
+    async def _async_step_finish_config(self) -> ConfigFlowResult: ...

homeassistant-stubs/components/romy/sensor.pyi

@@ -0,0 +1,20 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import RomyVacuumCoordinator as RomyVacuumCoordinator
+from .entity import RomyEntity as RomyEntity
+from _typeshed import Incomplete
+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 AREA_SQUARE_METERS as AREA_SQUARE_METERS, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfLength as UnitOfLength, UnitOfTime as UnitOfTime
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+SENSORS: list[SensorEntityDescription]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RomySensor(RomyEntity, SensorEntity):
+    entity_description: SensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: RomyVacuumCoordinator, entity_description: SensorEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> int: ...

homeassistant-stubs/components/sensor/__init__.pyi

@@ -80,7 +80,7 @@ class SensorEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
     def unit_of_measurement(self) -> str | None: ...
     @property
     def state(self) -> Any: ...
-    def _suggested_precision_or_none(self) -> int | None: ...
+    def _display_precision_or_none(self) -> int | None: ...
     def _update_suggested_precision(self) -> None: ...
     def _custom_unit_or_undef(self, primary_key: str, secondary_key: str) -> str | None | UndefinedType: ...
     def async_registry_entry_updated(self) -> None: ...

homeassistant-stubs/components/sensor/significant_change.pyi

@@ -4,5 +4,5 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.helpers.significant_change import check_absolute_change as check_absolute_change, check_percentage_change as check_percentage_change, check_valid_float as check_valid_float
 from typing import Any
 
-def _absolute_and_relative_change(old_state: int | float | None, new_state: int | float | None, absolute_change: int | float, percentage_change: int | float) -> bool: ...
+def _absolute_and_relative_change(old_state: float | None, new_state: float | None, absolute_change: float, percentage_change: float) -> bool: ...
 def async_check_significant_change(hass: HomeAssistant, old_state: str, old_attrs: dict, new_state: str, new_attrs: dict, **kwargs: Any) -> bool | None: ...

homeassistant-stubs/components/shelly/__init__.pyi

@@ -1,12 +1,10 @@
 from .const import BLOCK_EXPECTED_SLEEP_PERIOD as BLOCK_EXPECTED_SLEEP_PERIOD, BLOCK_WRONG_SLEEP_PERIOD as BLOCK_WRONG_SLEEP_PERIOD, CONF_COAP_PORT as CONF_COAP_PORT, CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DATA_CONFIG_ENTRY as DATA_CONFIG_ENTRY, DOMAIN as DOMAIN, FIRMWARE_UNSUPPORTED_ISSUE_ID as FIRMWARE_UNSUPPORTED_ISSUE_ID, LOGGER as LOGGER, MODELS_WITH_WRONG_SLEEP_PERIOD as MODELS_WITH_WRONG_SLEEP_PERIOD, PUSH_UPDATE_ISSUE_ID as PUSH_UPDATE_ISSUE_ID
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyEntryData as ShellyEntryData, ShellyRestCoordinator as ShellyRestCoordinator, ShellyRpcCoordinator as ShellyRpcCoordinator, ShellyRpcPollingCoordinator as ShellyRpcPollingCoordinator, get_entry_data as get_entry_data
-from .utils import async_create_issue_unsupported_firmware as async_create_issue_unsupported_firmware, async_shutdown_device as async_shutdown_device, get_block_device_sleep_period as get_block_device_sleep_period, get_coap_context as get_coap_context, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_rpc_device_wakeup_period as get_rpc_device_wakeup_period, get_ws_context as get_ws_context
+from .utils import async_create_issue_unsupported_firmware as async_create_issue_unsupported_firmware, get_coap_context as get_coap_context, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_ws_context as get_ws_context
 from _typeshed import Incomplete
-from aioshelly.block_device import BlockUpdateType as BlockUpdateType
-from aioshelly.rpc_device import RpcUpdateType as RpcUpdateType
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME, Platform as Platform
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, format_mac as format_mac

homeassistant-stubs/components/shelly/coordinator.pyi

@@ -1,13 +1,13 @@
 from .bluetooth import async_connect_scanner as async_connect_scanner
 from .const import ATTR_CHANNEL as ATTR_CHANNEL, ATTR_CLICK_TYPE as ATTR_CLICK_TYPE, ATTR_DEVICE as ATTR_DEVICE, ATTR_GENERATION as ATTR_GENERATION, BATTERY_DEVICES_WITH_PERMANENT_CONNECTION as BATTERY_DEVICES_WITH_PERMANENT_CONNECTION, BLEScannerMode as BLEScannerMode, CONF_BLE_SCANNER_MODE as CONF_BLE_SCANNER_MODE, CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DATA_CONFIG_ENTRY as DATA_CONFIG_ENTRY, DOMAIN as DOMAIN, DUAL_MODE_LIGHT_MODELS as DUAL_MODE_LIGHT_MODELS, ENTRY_RELOAD_COOLDOWN as ENTRY_RELOAD_COOLDOWN, EVENT_SHELLY_CLICK as EVENT_SHELLY_CLICK, INPUTS_EVENTS_DICT as INPUTS_EVENTS_DICT, LOGGER as LOGGER, MAX_PUSH_UPDATE_FAILURES as MAX_PUSH_UPDATE_FAILURES, MODELS_SUPPORTING_LIGHT_EFFECTS as MODELS_SUPPORTING_LIGHT_EFFECTS, OTA_BEGIN as OTA_BEGIN, OTA_ERROR as OTA_ERROR, OTA_PROGRESS as OTA_PROGRESS, OTA_SUCCESS as OTA_SUCCESS, PUSH_UPDATE_ISSUE_ID as PUSH_UPDATE_ISSUE_ID, REST_SENSORS_UPDATE_INTERVAL as REST_SENSORS_UPDATE_INTERVAL, RPC_INPUTS_EVENTS_TYPES as RPC_INPUTS_EVENTS_TYPES, RPC_RECONNECT_INTERVAL as RPC_RECONNECT_INTERVAL, RPC_SENSORS_POLLING_INTERVAL as RPC_SENSORS_POLLING_INTERVAL, SHBTN_MODELS as SHBTN_MODELS, SLEEP_PERIOD_MULTIPLIER as SLEEP_PERIOD_MULTIPLIER, UPDATE_PERIOD_MULTIPLIER as UPDATE_PERIOD_MULTIPLIER
-from .utils import async_shutdown_device as async_shutdown_device, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_rpc_device_wakeup_period as get_rpc_device_wakeup_period, update_device_fw_info as update_device_fw_info
+from .utils import async_create_issue_unsupported_firmware as async_create_issue_unsupported_firmware, async_shutdown_device as async_shutdown_device, get_block_device_sleep_period as get_block_device_sleep_period, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_rpc_device_wakeup_period as get_rpc_device_wakeup_period, update_device_fw_info as update_device_fw_info
 from _typeshed import Incomplete
 from aioshelly.block_device import BlockDevice, BlockUpdateType
 from aioshelly.rpc_device import RpcDevice, RpcUpdateType
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
-from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_HOST as CONF_HOST, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
+from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_HOST as CONF_HOST, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.debounce import Debouncer as Debouncer
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC
@@ -19,11 +19,10 @@ _DeviceT = TypeVar('_DeviceT', bound='BlockDevice|RpcDevice')
 @dataclass
 class ShellyEntryData:
     block: ShellyBlockCoordinator | None = ...
-    device: BlockDevice | RpcDevice | None = ...
     rest: ShellyRestCoordinator | None = ...
     rpc: ShellyRpcCoordinator | None = ...
     rpc_poll: ShellyRpcPollingCoordinator | None = ...
-    def __init__(self, block, device, rest, rpc, rpc_poll) -> None: ...
+    def __init__(self, block, rest, rpc, rpc_poll) -> None: ...
 
 def get_entry_data(hass: HomeAssistant) -> dict[str, ShellyEntryData]: ...
 
@@ -31,6 +30,7 @@ class ShellyCoordinatorBase(DataUpdateCoordinator[None], Generic[_DeviceT]):
     entry: Incomplete
     device: Incomplete
     device_id: Incomplete
+    _pending_platforms: Incomplete
     _debounced_reload: Incomplete
     def __init__(self, hass: HomeAssistant, entry: ConfigEntry, device: _DeviceT, update_interval: float) -> None: ...
     @property
@@ -41,7 +41,8 @@ class ShellyCoordinatorBase(DataUpdateCoordinator[None], Generic[_DeviceT]):
     def sw_version(self) -> str: ...
     @property
     def sleep_period(self) -> int: ...
-    def async_setup(self) -> None: ...
+    def async_setup(self, pending_platforms: list[Platform] | None = None) -> None: ...
+    async def _async_device_connect_task(self) -> bool: ...
     async def _async_reload_entry(self) -> None: ...
     last_update_success: bool
     async def async_shutdown_device_and_start_reauth(self) -> None: ...
@@ -60,7 +61,7 @@ class ShellyBlockCoordinator(ShellyCoordinatorBase[BlockDevice]):
     def _async_device_updates_handler(self) -> None: ...
     async def _async_update_data(self) -> None: ...
     def _async_handle_update(self, device_: BlockDevice, update_type: BlockUpdateType) -> None: ...
-    def async_setup(self) -> None: ...
+    def async_setup(self, pending_platforms: list[Platform] | None = None) -> None: ...
     def shutdown(self) -> None: ...
     def _handle_ha_stop(self, _event: Event) -> None: ...
 
@@ -91,7 +92,7 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
     async def _async_run_connected_events(self) -> None: ...
     async def _async_connect_ble_scanner(self) -> None: ...
     def _async_handle_update(self, device_: RpcDevice, update_type: RpcUpdateType) -> None: ...
-    def async_setup(self) -> None: ...
+    def async_setup(self, pending_platforms: list[Platform] | None = None) -> None: ...
     async def shutdown(self) -> None: ...
     async def _handle_ha_stop(self, _event: Event) -> None: ...
 

homeassistant-stubs/components/shelly/switch.pyi

@@ -1,7 +1,7 @@
 from .const import DOMAIN as DOMAIN, GAS_VALVE_OPEN_STATES as GAS_VALVE_OPEN_STATES
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyRpcCoordinator as ShellyRpcCoordinator, get_entry_data as get_entry_data
 from .entity import BlockEntityDescription as BlockEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, ShellyBlockEntity as ShellyBlockEntity, ShellyRpcEntity as ShellyRpcEntity, async_setup_block_attribute_entities as async_setup_block_attribute_entities
-from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, is_block_channel_type_light as is_block_channel_type_light, is_rpc_channel_type_light as is_rpc_channel_type_light, is_rpc_thermostat_internal_actuator as is_rpc_thermostat_internal_actuator
+from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, is_block_channel_type_light as is_block_channel_type_light, is_rpc_channel_type_light as is_rpc_channel_type_light, is_rpc_thermostat_internal_actuator as is_rpc_thermostat_internal_actuator, is_rpc_thermostat_mode as is_rpc_thermostat_mode
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from dataclasses import dataclass

homeassistant-stubs/components/shelly/utils.pyi

@@ -59,3 +59,4 @@ def is_rpc_wifi_stations_disabled(config: dict[str, Any], _status: dict[str, Any
 def get_http_port(data: MappingProxyType[str, Any]) -> int: ...
 async def async_shutdown_device(device: BlockDevice | RpcDevice) -> None: ...
 def async_remove_shelly_rpc_entities(hass: HomeAssistant, domain: str, mac: str, keys: list[str]) -> None: ...
+def is_rpc_thermostat_mode(ident: int, status: dict[str, Any]) -> bool: ...

homeassistant-stubs/components/starlink/coordinator.pyi

@@ -20,6 +20,7 @@ class StarlinkUpdateCoordinator(DataUpdateCoordinator[StarlinkData]):
     channel_context: Incomplete
     timezone: Incomplete
     def __init__(self, hass: HomeAssistant, name: str, url: str) -> None: ...
+    def _get_starlink_data(self) -> StarlinkData: ...
     async def _async_update_data(self) -> StarlinkData: ...
     async def async_stow_starlink(self, stow: bool) -> None: ...
     async def async_reboot_starlink(self) -> None: ...

homeassistant-stubs/components/statistics/sensor.pyi

@@ -5,9 +5,9 @@ from datetime import datetime, timedelta
 from homeassistant.components.recorder import get_instance as get_instance, history as history
 from homeassistant.components.sensor import DEVICE_CLASS_STATE_CLASSES as DEVICE_CLASS_STATE_CLASSES, PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorStateClass as SensorStateClass
 from homeassistant.const import ATTR_DEVICE_CLASS as ATTR_DEVICE_CLASS, ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME, CONF_UNIQUE_ID as CONF_UNIQUE_ID, PERCENTAGE as PERCENTAGE, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.event import EventStateChangedData as EventStateChangedData, async_track_point_in_utc_time as async_track_point_in_utc_time, async_track_state_change_event as async_track_state_change_event
+from homeassistant.helpers.event import async_track_point_in_utc_time as async_track_point_in_utc_time, async_track_state_change_event as async_track_state_change_event
 from homeassistant.helpers.reload import async_setup_reload_service as async_setup_reload_service
 from homeassistant.helpers.start import async_at_start as async_at_start
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, StateType as StateType

homeassistant-stubs/components/switch/light.pyi

@@ -1,9 +1,9 @@
 from _typeshed import Incomplete
 from homeassistant.components.light import ColorMode as ColorMode, LightEntity as LightEntity, PLATFORM_SCHEMA as PLATFORM_SCHEMA
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON, STATE_ON as STATE_ON, STATE_UNAVAILABLE as STATE_UNAVAILABLE
-from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.event import EventStateChangedData as EventStateChangedData, async_track_state_change_event as async_track_state_change_event
+from homeassistant.helpers.event import async_track_state_change_event as async_track_state_change_event
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from typing import Any
 

homeassistant-stubs/components/synology_dsm/__init__.pyi

@@ -1,4 +1,4 @@
-from .common import SynoApi as SynoApi
+from .common import SynoApi as SynoApi, raise_config_entry_auth_error as raise_config_entry_auth_error
 from .const import DEFAULT_VERIFY_SSL as DEFAULT_VERIFY_SSL, DOMAIN as DOMAIN, EXCEPTION_DETAILS as EXCEPTION_DETAILS, EXCEPTION_UNKNOWN as EXCEPTION_UNKNOWN, PLATFORMS as PLATFORMS, SYNOLOGY_AUTH_FAILED_EXCEPTIONS as SYNOLOGY_AUTH_FAILED_EXCEPTIONS, SYNOLOGY_CONNECTION_EXCEPTIONS as SYNOLOGY_CONNECTION_EXCEPTIONS
 from .coordinator import SynologyDSMCameraUpdateCoordinator as SynologyDSMCameraUpdateCoordinator, SynologyDSMCentralUpdateCoordinator as SynologyDSMCentralUpdateCoordinator, SynologyDSMSwitchUpdateCoordinator as SynologyDSMSwitchUpdateCoordinator
 from .models import SynologyDSMData as SynologyDSMData
@@ -7,7 +7,7 @@ from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_MAC as CONF_MAC, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers import device_registry as dr
 from synology_dsm.api.surveillance_station.camera import SynoCamera as SynoCamera
 

homeassistant-stubs/components/synology_dsm/camera.pyi

@@ -17,7 +17,8 @@ _LOGGER: Incomplete
 
 @dataclass(frozen=True, kw_only=True)
 class SynologyDSMCameraEntityDescription(CameraEntityDescription, SynologyDSMEntityDescription):
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_key) -> None: ...
+    camera_id: int
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, api_key, camera_id) -> None: ...
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
@@ -25,7 +26,7 @@ class SynoDSMCamera(SynologyDSMBaseEntity[SynologyDSMCameraUpdateCoordinator], C
     _attr_supported_features: Incomplete
     entity_description: SynologyDSMCameraEntityDescription
     snapshot_quality: Incomplete
-    def __init__(self, api: SynoApi, coordinator: SynologyDSMCameraUpdateCoordinator, camera_id: str) -> None: ...
+    def __init__(self, api: SynoApi, coordinator: SynologyDSMCameraUpdateCoordinator, camera_id: int) -> None: ...
     @property
     def camera_data(self) -> SynoCamera: ...
     @property

homeassistant-stubs/components/synology_dsm/common.pyi

@@ -1,20 +1,22 @@
-from .const import CONF_DEVICE_TOKEN as CONF_DEVICE_TOKEN, SYNOLOGY_CONNECTION_EXCEPTIONS as SYNOLOGY_CONNECTION_EXCEPTIONS
+from .const import CONF_DEVICE_TOKEN as CONF_DEVICE_TOKEN, EXCEPTION_DETAILS as EXCEPTION_DETAILS, EXCEPTION_UNKNOWN as EXCEPTION_UNKNOWN, SYNOLOGY_CONNECTION_EXCEPTIONS as SYNOLOGY_CONNECTION_EXCEPTIONS
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_SSL as CONF_SSL, CONF_TIMEOUT as CONF_TIMEOUT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from synology_dsm import SynologyDSM
 from synology_dsm.api.dsm.network import SynoDSMNetwork as SynoDSMNetwork
 from synology_dsm.api.photos import SynoPhotos as SynoPhotos
 
 LOGGER: Incomplete
 
 class SynoApi:
+    dsm: SynologyDSM
     _hass: Incomplete
     _entry: Incomplete
     config_url: Incomplete
-    dsm: Incomplete
     information: Incomplete
     network: Incomplete
     security: Incomplete
@@ -44,3 +46,5 @@ class SynoApi:
     async def async_unload(self) -> None: ...
     async def async_update(self) -> None: ...
     async def _update(self) -> None: ...
+
+def raise_config_entry_auth_error(err: Exception) -> None: ...

homeassistant-stubs/components/synology_dsm/coordinator.pyi

@@ -1,5 +1,5 @@
-from .common import SynoApi as SynoApi
-from .const import DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, SIGNAL_CAMERA_SOURCE_CHANGED as SIGNAL_CAMERA_SOURCE_CHANGED, SYNOLOGY_CONNECTION_EXCEPTIONS as SYNOLOGY_CONNECTION_EXCEPTIONS
+from .common import SynoApi as SynoApi, raise_config_entry_auth_error as raise_config_entry_auth_error
+from .const import DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, SIGNAL_CAMERA_SOURCE_CHANGED as SIGNAL_CAMERA_SOURCE_CHANGED, SYNOLOGY_AUTH_FAILED_EXCEPTIONS as SYNOLOGY_AUTH_FAILED_EXCEPTIONS, SYNOLOGY_CONNECTION_EXCEPTIONS as SYNOLOGY_CONNECTION_EXCEPTIONS
 from _typeshed import Incomplete
 from datetime import timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -28,6 +28,6 @@ class SynologyDSMCentralUpdateCoordinator(SynologyDSMUpdateCoordinator[None]):
     def __init__(self, hass: HomeAssistant, entry: ConfigEntry, api: SynoApi) -> None: ...
     async def _async_update_data(self) -> None: ...
 
-class SynologyDSMCameraUpdateCoordinator(SynologyDSMUpdateCoordinator[dict[str, dict[str, SynoCamera]]]):
+class SynologyDSMCameraUpdateCoordinator(SynologyDSMUpdateCoordinator[dict[str, dict[int, SynoCamera]]]):
     def __init__(self, hass: HomeAssistant, entry: ConfigEntry, api: SynoApi) -> None: ...
-    async def _async_update_data(self) -> dict[str, dict[str, SynoCamera]]: ...
+    async def _async_update_data(self) -> dict[str, dict[int, SynoCamera]]: ...

homeassistant-stubs/components/synology_dsm/diagnostics.pyi

@@ -5,7 +5,6 @@ from homeassistant.components.diagnostics import async_redact_data as async_reda
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
-from synology_dsm.api.surveillance_station.camera import SynoCamera as SynoCamera
 from typing import Any
 
 TO_REDACT: Incomplete

homeassistant-stubs/components/system_log/__init__.pyi

@@ -1,12 +1,12 @@
 import logging
 import re
-import traceback
 from _typeshed import Incomplete
 from collections import OrderedDict
 from homeassistant.components import websocket_api as websocket_api
 from homeassistant.const import EVENT_HOMEASSISTANT_CLOSE as EVENT_HOMEASSISTANT_CLOSE
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.helpers.typing import ConfigType as ConfigType
+from types import FrameType
 from typing import Any
 
 KeyType: Incomplete
@@ -26,7 +26,7 @@ CONFIG_SCHEMA: Incomplete
 SERVICE_CLEAR_SCHEMA: Incomplete
 SERVICE_WRITE_SCHEMA: Incomplete
 
-def _figure_out_source(record: logging.LogRecord, paths_re: re.Pattern[str], extracted_tb: traceback.StackSummary | None = None) -> tuple[str, int]: ...
+def _figure_out_source(record: logging.LogRecord, paths_re: re.Pattern[str], extracted_tb: list[tuple[FrameType, int]] | None = None) -> tuple[str, int]: ...
 def _safe_get_message(record: logging.LogRecord) -> str: ...
 
 class LogEntry:
@@ -40,7 +40,7 @@ class LogEntry:
     source: Incomplete
     count: int
     key: Incomplete
-    def __init__(self, record: logging.LogRecord, paths_re: re.Pattern, figure_out_source: bool = False) -> None: ...
+    def __init__(self, record: logging.LogRecord, paths_re: re.Pattern, formatter: logging.Formatter | None = None, figure_out_source: bool = False) -> None: ...
     def to_dict(self) -> dict[str, Any]: ...
 
 class DedupStore(OrderedDict[KeyType, LogEntry]):

homeassistant-stubs/components/threshold/binary_sensor.pyi

@@ -3,10 +3,10 @@ from _typeshed import Incomplete
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, PLATFORM_SCHEMA as PLATFORM_SCHEMA
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
-from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.event import EventStateChangedData as EventStateChangedData, async_track_state_change_event as async_track_state_change_event
+from homeassistant.helpers.event import async_track_state_change_event as async_track_state_change_event
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from typing import Any
 

homeassistant-stubs/components/tibber/sensor.pyi

@@ -16,6 +16,7 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity as Coordi
 from homeassistant.util import Throttle as Throttle
 from typing import Any
 
+FIVE_YEARS: Incomplete
 _LOGGER: Incomplete
 ICON: str
 SCAN_INTERVAL: Incomplete

homeassistant-stubs/components/tplink/coordinator.pyi

@@ -1,5 +1,6 @@
 from _typeshed import Incomplete
 from datetime import timedelta
+from homeassistant import config_entries as config_entries
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.debounce import Debouncer as Debouncer
@@ -10,6 +11,7 @@ _LOGGER: Incomplete
 REQUEST_REFRESH_DELAY: float
 
 class TPLinkDataUpdateCoordinator(DataUpdateCoordinator[None]):
+    config_entry: config_entries.ConfigEntry
     device: Incomplete
     def __init__(self, hass: HomeAssistant, device: SmartDevice, update_interval: timedelta) -> None: ...
     async def _async_update_data(self) -> None: ...

homeassistant-stubs/components/tplink/entity.pyi

@@ -2,6 +2,7 @@ from .const import DOMAIN as DOMAIN
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Coroutine
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from kasa import SmartDevice as SmartDevice

homeassistant-stubs/components/tplink/light.pyi

@@ -40,7 +40,7 @@ class TPLinkSmartBulb(CoordinatedTPLinkEntity, LightEntity):
     def __init__(self, device: SmartBulb, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
     def _async_extract_brightness_transition(self, **kwargs: Any) -> tuple[int | None, int | None]: ...
     async def _async_set_hsv(self, hs_color: tuple[int, int], brightness: int | None, transition: int | None) -> None: ...
-    async def _async_set_color_temp(self, color_temp: float | int, brightness: int | None, transition: int | None) -> None: ...
+    async def _async_set_color_temp(self, color_temp: float, brightness: int | None, transition: int | None) -> None: ...
     async def _async_turn_on_with_brightness(self, brightness: int | None, transition: int | None) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/tplink/sensor.pyi

@@ -8,7 +8,7 @@ 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
+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
 
@@ -30,5 +30,6 @@ class SmartPlugSensor(CoordinatedTPLinkEntity, SensorEntity):
     _attr_translation_placeholders: Incomplete
     _attr_translation_key: Incomplete
     def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkSensorEntityDescription, has_parent: bool = False) -> None: ...
-    @property
-    def native_value(self) -> float | None: ...
+    _attr_native_value: Incomplete
+    def _async_update_attrs(self) -> None: ...
+    def _handle_coordinator_update(self) -> None: ...

homeassistant-stubs/components/tplink_omada/switch.pyi

@@ -24,28 +24,27 @@ def _get_switch_port_base_name(port: OmadaSwitchPortDetails) -> str: ...
 @dataclass(frozen=True, kw_only=True)
 class OmadaDevicePortSwitchEntityDescription(SwitchEntityDescription, Generic[TCoordinator, TDevice, TPort]):
     exists_func: Callable[[TDevice, TPort], bool] = ...
-    coordinator_update_func: Callable[[TCoordinator, TDevice, int | str], TPort | None] = ...
-    set_func: Callable[[OmadaSiteClient, TDevice, TPort, bool], Awaitable[TPort]]
+    coordinator_update_func: Callable[[TCoordinator, TDevice, TPort], TPort | None]
+    set_func: Callable[[OmadaSiteClient, TDevice, TPort, bool], Awaitable[TPort | None]]
     update_func: Callable[[TPort], bool]
-    refresh_after_set: 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, exists_func, coordinator_update_func, set_func, update_func, refresh_after_set) -> 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, exists_func, coordinator_update_func, set_func, update_func) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class OmadaSwitchPortSwitchEntityDescription(OmadaDevicePortSwitchEntityDescription[OmadaSwitchPortCoordinator, OmadaSwitch, OmadaSwitchPortDetails]):
-    coordinator_update_func: Callable[[OmadaSwitchPortCoordinator, OmadaSwitch, int | str], OmadaSwitchPortDetails | 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, exists_func, coordinator_update_func, set_func, update_func, refresh_after_set) -> None: ...
+    coordinator_update_func: Callable[[OmadaSwitchPortCoordinator, OmadaSwitch, OmadaSwitchPortDetails], OmadaSwitchPortDetails | 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, exists_func, coordinator_update_func, set_func, update_func) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class OmadaGatewayPortConfigSwitchEntityDescription(OmadaDevicePortSwitchEntityDescription[OmadaGatewayCoordinator, OmadaGateway, OmadaGatewayPortConfig]):
-    coordinator_update_func: Callable[[OmadaGatewayCoordinator, OmadaGateway, int | str], OmadaGatewayPortConfig | 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, exists_func, coordinator_update_func, set_func, update_func, refresh_after_set) -> None: ...
+    coordinator_update_func: Callable[[OmadaGatewayCoordinator, OmadaGateway, OmadaGatewayPortConfig], OmadaGatewayPortConfig | 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, exists_func, coordinator_update_func, set_func, update_func) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class OmadaGatewayPortStatusSwitchEntityDescription(OmadaDevicePortSwitchEntityDescription[OmadaGatewayCoordinator, OmadaGateway, OmadaGatewayPortStatus]):
-    coordinator_update_func: Callable[[OmadaGatewayCoordinator, OmadaGateway, int | str], OmadaGatewayPortStatus] = ...
-    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, exists_func, coordinator_update_func, set_func, update_func, refresh_after_set) -> None: ...
+    coordinator_update_func: Callable[[OmadaGatewayCoordinator, OmadaGateway, OmadaGatewayPortStatus], OmadaGatewayPortStatus] = ...
+    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, exists_func, coordinator_update_func, set_func, update_func) -> None: ...
 
-def _wan_connect_disconnect(client: OmadaSiteClient, device: OmadaDevice, port: OmadaGatewayPortStatus, enable: bool, ipv6: bool) -> Awaitable[OmadaGatewayPortStatus]: ...
+async def _wan_connect_disconnect(client: OmadaSiteClient, device: OmadaDevice, port: OmadaGatewayPortStatus, enable: bool, ipv6: bool) -> None: ...
 
 SWITCH_PORT_DETAILS_SWITCHES: list[OmadaSwitchPortSwitchEntityDescription]
 GATEWAY_PORT_STATUS_SWITCHES: list[OmadaGatewayPortStatusSwitchEntityDescription]
@@ -53,13 +52,12 @@ GATEWAY_PORT_CONFIG_SWITCHES: list[OmadaGatewayPortConfigSwitchEntityDescription
 
 class OmadaDevicePortSwitchEntity(OmadaDeviceEntity[TCoordinator], SwitchEntity, Generic[TCoordinator, TDevice, TPort]):
     _attr_has_entity_name: bool
-    _port_details: TPort | None
     entity_description: OmadaDevicePortSwitchEntityDescription[TCoordinator, TDevice, TPort]
     _device: Incomplete
-    _port_id: Incomplete
+    _port_details: Incomplete
     _attr_unique_id: Incomplete
     _attr_translation_placeholders: Incomplete
-    def __init__(self, coordinator: TCoordinator, device: TDevice, port_id: int | str, entity_description: OmadaDevicePortSwitchEntityDescription[TCoordinator, TDevice, TPort], port_name: str | None = None) -> None: ...
+    def __init__(self, coordinator: TCoordinator, device: TDevice, port_details: TPort, port_id: str, entity_description: OmadaDevicePortSwitchEntityDescription[TCoordinator, TDevice, TPort], port_name: str | None = None) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     _attr_is_on: Incomplete
     async def _async_turn_on_off(self, enable: bool) -> None: ...

homeassistant-stubs/components/trend/binary_sensor.pyi

@@ -5,10 +5,10 @@ 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.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, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
 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 EventStateChangedData as EventStateChangedData, async_track_state_change_event as async_track_state_change_event
+from homeassistant.helpers.event import async_track_state_change_event as async_track_state_change_event
 from homeassistant.helpers.reload import async_setup_reload_service as async_setup_reload_service
 from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType

homeassistant-stubs/components/unifi/__init__.pyi

@@ -8,6 +8,7 @@ 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, callback as callback
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
 from homeassistant.helpers.storage import Store as Store
 from homeassistant.helpers.typing import ConfigType as ConfigType
 
@@ -19,6 +20,7 @@ CONFIG_SCHEMA: Incomplete
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: ConfigEntry, device_entry: DeviceEntry) -> bool: ...
 
 class UnifiWirelessClients:
     hass: Incomplete

homeassistant-stubs/components/unifi/button.pyi

@@ -1,5 +1,5 @@
 import aiounifi
-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_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
 from .hub import UnifiHub as UnifiHub
 from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
@@ -14,6 +14,7 @@ from typing import Any
 
 async def async_restart_device_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
 async def async_power_cycle_port_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
+async def async_regenerate_password_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class UnifiButtonEntityDescription(ButtonEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):

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

@@ -0,0 +1,44 @@
+import aiounifi
+from _typeshed import Incomplete
+from datetime import datetime
+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.event import async_call_later as async_call_later, async_track_time_interval as async_track_time_interval
+
+class UnifiEntityHelper:
+    hass: Incomplete
+    api: Incomplete
+    _device_command: Incomplete
+    _heartbeat: Incomplete
+    def __init__(self, hass: HomeAssistant, api: aiounifi.Controller) -> None: ...
+    def reset(self) -> None: ...
+    def initialize(self) -> None: ...
+    @property
+    def signal_heartbeat(self) -> str: ...
+    def update_heartbeat(self, unique_id: str, heartbeat_expire_time: datetime) -> None: ...
+    def remove_heartbeat(self, unique_id: str) -> None: ...
+    def queue_poe_port_command(self, device_id: str, port_idx: int, poe_mode: str) -> None: ...
+
+class UnifiEntityHeartbeat:
+    CHECK_HEARTBEAT_INTERVAL: Incomplete
+    hass: Incomplete
+    _cancel_heartbeat_check: Incomplete
+    _heartbeat_time: Incomplete
+    def __init__(self, hass: HomeAssistant) -> None: ...
+    def reset(self) -> None: ...
+    def initialize(self) -> None: ...
+    @property
+    def signal(self) -> str: ...
+    def update(self, unique_id: str, heartbeat_expire_time: datetime) -> None: ...
+    def remove(self, unique_id: str) -> None: ...
+    def _check_for_stale(self, *_: datetime) -> None: ...
+
+class UnifiDeviceCommand:
+    COMMAND_DELAY: int
+    hass: Incomplete
+    api: Incomplete
+    _command_queue: Incomplete
+    _cancel_command: Incomplete
+    def __init__(self, hass: HomeAssistant, api: aiounifi.Controller) -> None: ...
+    def reset(self) -> None: ...
+    def queue_poe_command(self, device_id: str, port_idx: int, poe_mode: str) -> None: ...