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

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(self) -> None: ...
     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/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: ...

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

@@ -1,45 +1,40 @@
 import aiounifi
 from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_SITE_ID as CONF_SITE_ID, PLATFORMS as PLATFORMS
 from .config import UnifiConfig as UnifiConfig
+from .entity_helper import UnifiEntityHelper as UnifiEntityHelper
 from .entity_loader import UnifiEntityLoader as UnifiEntityLoader
 from .websocket import UnifiWebsocket as UnifiWebsocket
 from _typeshed import Incomplete
 from datetime import datetime
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry, DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 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
-
-CHECK_HEARTBEAT_INTERVAL: Incomplete
 
 class UnifiHub:
     hass: Incomplete
     api: Incomplete
     config: Incomplete
     entity_loader: Incomplete
+    _entity_helper: Incomplete
     websocket: Incomplete
     site: Incomplete
     is_admin: bool
-    _cancel_heartbeat_check: Incomplete
-    _heartbeat_time: Incomplete
-    poe_command_queue: Incomplete
-    _cancel_poe_command: Incomplete
     def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, api: aiounifi.Controller) -> None: ...
     @staticmethod
     def get_hub(hass: HomeAssistant, config_entry: ConfigEntry) -> UnifiHub: ...
     @property
     def available(self) -> bool: ...
     @property
+    def signal_heartbeat_missed(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: ...
+    @property
     def signal_reachable(self) -> str: ...
     @property
     def signal_options_update(self) -> str: ...
-    @property
-    def signal_heartbeat_missed(self) -> str: ...
     async def initialize(self) -> None: ...
-    def async_heartbeat(self, unique_id: str, heartbeat_expire_time: datetime | None = None) -> None: ...
-    def _async_check_for_stale(self, *_: datetime) -> None: ...
-    def async_queue_poe_port_command(self, device_id: str, port_idx: int, poe_mode: str) -> None: ...
     @property
     def device_info(self) -> DeviceInfo: ...
     def async_update_device_registry(self) -> DeviceEntry: ...

homeassistant-stubs/components/unifi/sensor.pyi

@@ -13,7 +13,7 @@ from datetime import date, datetime
 from decimal import Decimal
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass, UnitOfTemperature as UnitOfTemperature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import EntityCategory as EntityCategory, UnitOfDataRate as UnitOfDataRate, UnitOfPower as UnitOfPower
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfDataRate as UnitOfDataRate, UnitOfPower as UnitOfPower
 from homeassistant.core import Event as core_Event, 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
@@ -29,6 +29,7 @@ def async_device_uptime_value_fn(hub: UnifiHub, device: Device) -> datetime | No
 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: ...
 def async_device_outlet_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def device_system_stats_supported_fn(stat_index: int, hub: UnifiHub, obj_id: str) -> bool: ...
 def async_client_is_connected_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_device_state_value_fn(hub: UnifiHub, device: Device) -> str: ...
 

homeassistant-stubs/components/velbus/cover.pyi

@@ -12,7 +12,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 
 class VelbusCover(VelbusEntity, CoverEntity):
     _channel: VelbusBlind
+    _assumed_closed: bool
     _attr_supported_features: Incomplete
+    _attr_assumed_state: bool
     def __init__(self, channel: VelbusBlind) -> None: ...
     @property
     def is_closed(self) -> bool | None: ...

homeassistant-stubs/components/weather/significant_change.pyi

@@ -7,5 +7,5 @@ from typing import Any
 SIGNIFICANT_ATTRIBUTES: set[str]
 VALID_CARDINAL_DIRECTIONS: list[str]
 
-def _cardinal_to_degrees(value: str | int | float | None) -> int | float | None: ...
+def _cardinal_to_degrees(value: str | float | None) -> int | float | None: ...
 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/websocket_api/commands.pyi

@@ -7,11 +7,11 @@ from homeassistant.auth.models import User as User
 from homeassistant.auth.permissions.const import POLICY_READ as POLICY_READ
 from homeassistant.auth.permissions.events import SUBSCRIBE_ALLOWLIST as SUBSCRIBE_ALLOWLIST
 from homeassistant.const import EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, MATCH_ALL as MATCH_ALL, SIGNAL_BOOTSTRAP_INTEGRATIONS as SIGNAL_BOOTSTRAP_INTEGRATIONS
-from homeassistant.core import Context as Context, Event as Event, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, State as State, callback as callback
+from homeassistant.core import Context as Context, Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, State as State, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceNotFound as ServiceNotFound, ServiceValidationError as ServiceValidationError, TemplateError as TemplateError, Unauthorized as Unauthorized
 from homeassistant.helpers import entity as entity, template as template
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
-from homeassistant.helpers.event import EventStateChangedData as EventStateChangedData, TrackTemplate as TrackTemplate, TrackTemplateResult as TrackTemplateResult, async_track_template_result as async_track_template_result
+from homeassistant.helpers.event import TrackTemplate as TrackTemplate, TrackTemplateResult as TrackTemplateResult, async_track_template_result as async_track_template_result
 from homeassistant.helpers.json import ExtendedJSONEncoder as ExtendedJSONEncoder, JSON_DUMP as JSON_DUMP, find_paths_unserializable_data as find_paths_unserializable_data, json_bytes as json_bytes
 from homeassistant.helpers.service import async_get_all_descriptions as async_get_all_descriptions
 from homeassistant.loader import Integration as Integration, IntegrationNotFound as IntegrationNotFound, async_get_integration as async_get_integration, async_get_integration_descriptions as async_get_integration_descriptions, async_get_integrations as async_get_integrations

homeassistant-stubs/components/websocket_api/messages.pyi

@@ -1,8 +1,7 @@
 from . import const as const
 from _typeshed import Incomplete
 from homeassistant.const import COMPRESSED_STATE_ATTRIBUTES as COMPRESSED_STATE_ATTRIBUTES, COMPRESSED_STATE_CONTEXT as COMPRESSED_STATE_CONTEXT, COMPRESSED_STATE_LAST_CHANGED as COMPRESSED_STATE_LAST_CHANGED, COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
-from homeassistant.core import Event as Event, State as State
-from homeassistant.helpers.event import EventStateChangedData as EventStateChangedData
+from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData, State as State
 from homeassistant.helpers.json import JSON_DUMP as JSON_DUMP, find_paths_unserializable_data as find_paths_unserializable_data, json_bytes as json_bytes
 from homeassistant.util.json import format_unserializable_data as format_unserializable_data
 from typing import Any, Final

homeassistant-stubs/components/websocket_api/sensor.pyi

@@ -8,13 +8,9 @@ from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfo
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 
 class APICount(SensorEntity):
-    count: int
+    _attr_name: str
+    _attr_native_unit_of_measurement: str
+    _attr_native_value: int
     def __init__(self) -> None: ...
     async def async_added_to_hass(self) -> None: ...
-    @property
-    def name(self) -> str: ...
-    @property
-    def native_value(self) -> int: ...
-    @property
-    def native_unit_of_measurement(self) -> str: ...
     def _update_count(self) -> None: ...