homeassistant-stubs 2024.5.4__py3-none-any.whl → 2024.6.0__py3-none-any.whl

homeassistant-stubs/components/water_heater/__init__.pyi

@@ -1,5 +1,5 @@
+from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from collections.abc import Mapping
 from enum import IntFlag
 from functools import cached_property as cached_property
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -16,7 +16,6 @@ from typing import Any
 
 DEFAULT_MIN_TEMP: int
 DEFAULT_MAX_TEMP: int
-DOMAIN: str
 ENTITY_ID_FORMAT: Incomplete
 SCAN_INTERVAL: Incomplete
 SERVICE_SET_AWAY_MODE: str
@@ -83,7 +82,7 @@ class WaterHeaterEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
     @property
     def precision(self) -> float: ...
     @property
-    def capability_attributes(self) -> Mapping[str, Any]: ...
+    def capability_attributes(self) -> dict[str, Any]: ...
     @property
     def state_attributes(self) -> dict[str, Any]: ...
     @cached_property

homeassistant-stubs/components/water_heater/const.pyi

@@ -1,3 +1,4 @@
+DOMAIN: str
 STATE_ECO: str
 STATE_ELECTRIC: str
 STATE_PERFORMANCE: str

homeassistant-stubs/components/water_heater/group.pyi

@@ -1,6 +1,6 @@
-from .const import STATE_ECO as STATE_ECO, STATE_ELECTRIC as STATE_ELECTRIC, STATE_GAS as STATE_GAS, STATE_HEAT_PUMP as STATE_HEAT_PUMP, STATE_HIGH_DEMAND as STATE_HIGH_DEMAND, STATE_PERFORMANCE as STATE_PERFORMANCE
+from .const import DOMAIN as DOMAIN, STATE_ECO as STATE_ECO, STATE_ELECTRIC as STATE_ELECTRIC, STATE_GAS as STATE_GAS, STATE_HEAT_PUMP as STATE_HEAT_PUMP, STATE_HIGH_DEMAND as STATE_HIGH_DEMAND, STATE_PERFORMANCE as STATE_PERFORMANCE
 from homeassistant.components.group import GroupIntegrationRegistry as GroupIntegrationRegistry
-from homeassistant.const import STATE_OFF as STATE_OFF
+from homeassistant.const import STATE_OFF as STATE_OFF, STATE_ON as STATE_ON
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 
 def async_describe_on_off_states(hass: HomeAssistant, registry: GroupIntegrationRegistry) -> None: ...

homeassistant-stubs/components/weather/__init__.pyi

@@ -16,7 +16,8 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity as Coordi
 from homeassistant.util.dt import utcnow as utcnow
 from homeassistant.util.json import JsonValueType as JsonValueType
 from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM as US_CUSTOMARY_SYSTEM
-from typing import Any, Final, Generic, Literal, Required, TypeVar, TypedDict
+from typing import Any, Final, Generic, Literal, Required, TypedDict
+from typing_extensions import TypeVar
 
 _LOGGER: Incomplete
 ATTR_CONDITION_CLASS: str
@@ -64,10 +65,10 @@ SCAN_INTERVAL: Incomplete
 ROUNDING_PRECISION: int
 LEGACY_SERVICE_GET_FORECAST: Final[str]
 SERVICE_GET_FORECASTS: Final[str]
-_ObservationUpdateCoordinatorT = TypeVar('_ObservationUpdateCoordinatorT', bound='DataUpdateCoordinator[Any]')
-_DailyForecastUpdateCoordinatorT = TypeVar('_DailyForecastUpdateCoordinatorT', bound='TimestampDataUpdateCoordinator[Any]')
-_HourlyForecastUpdateCoordinatorT = TypeVar('_HourlyForecastUpdateCoordinatorT', bound='TimestampDataUpdateCoordinator[Any]')
-_TwiceDailyForecastUpdateCoordinatorT = TypeVar('_TwiceDailyForecastUpdateCoordinatorT', bound='TimestampDataUpdateCoordinator[Any]')
+_ObservationUpdateCoordinatorT = TypeVar('_ObservationUpdateCoordinatorT', bound=DataUpdateCoordinator[Any], default=DataUpdateCoordinator[dict[str, Any]])
+_DailyForecastUpdateCoordinatorT = TypeVar('_DailyForecastUpdateCoordinatorT', bound=TimestampDataUpdateCoordinator[Any], default=TimestampDataUpdateCoordinator[None])
+_HourlyForecastUpdateCoordinatorT = TypeVar('_HourlyForecastUpdateCoordinatorT', bound=TimestampDataUpdateCoordinator[Any], default=_DailyForecastUpdateCoordinatorT)
+_TwiceDailyForecastUpdateCoordinatorT = TypeVar('_TwiceDailyForecastUpdateCoordinatorT', bound=TimestampDataUpdateCoordinator[Any], default=_DailyForecastUpdateCoordinatorT)
 
 def round_temperature(temperature: float | None, precision: float) -> float | None: ...
 
@@ -220,7 +221,7 @@ class CoordinatorWeatherEntity(CoordinatorEntity[_ObservationUpdateCoordinatorT]
     forecast_coordinators: Incomplete
     forecast_valid: Incomplete
     unsub_forecast: Incomplete
-    def __init__(self, observation_coordinator: _ObservationUpdateCoordinatorT, *, context: Any = None, daily_coordinator: _DailyForecastUpdateCoordinatorT | None = None, hourly_coordinator: _DailyForecastUpdateCoordinatorT | None = None, twice_daily_coordinator: _DailyForecastUpdateCoordinatorT | None = None, daily_forecast_valid: timedelta | None = None, hourly_forecast_valid: timedelta | None = None, twice_daily_forecast_valid: timedelta | None = None) -> None: ...
+    def __init__(self, observation_coordinator: _ObservationUpdateCoordinatorT, *, context: Any = None, daily_coordinator: _DailyForecastUpdateCoordinatorT | None = None, hourly_coordinator: _HourlyForecastUpdateCoordinatorT | None = None, twice_daily_coordinator: _TwiceDailyForecastUpdateCoordinatorT | None = None, daily_forecast_valid: timedelta | None = None, hourly_forecast_valid: timedelta | None = None, twice_daily_forecast_valid: timedelta | None = None) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     def _remove_forecast_listener(self, forecast_type: Literal['daily', 'hourly', 'twice_daily']) -> None: ...
     def _async_subscription_started(self, forecast_type: Literal['daily', 'hourly', 'twice_daily']) -> None: ...
@@ -238,6 +239,6 @@ class CoordinatorWeatherEntity(CoordinatorEntity[_ObservationUpdateCoordinatorT]
     async def async_forecast_hourly(self) -> list[Forecast] | None: ...
     async def async_forecast_twice_daily(self) -> list[Forecast] | None: ...
 
-class SingleCoordinatorWeatherEntity(CoordinatorWeatherEntity[_ObservationUpdateCoordinatorT, TimestampDataUpdateCoordinator[None], TimestampDataUpdateCoordinator[None], TimestampDataUpdateCoordinator[None]]):
+class SingleCoordinatorWeatherEntity(CoordinatorWeatherEntity[_ObservationUpdateCoordinatorT, TimestampDataUpdateCoordinator[None]]):
     def __init__(self, coordinator: _ObservationUpdateCoordinatorT, context: Any = None) -> None: ...
     def _handle_coordinator_update(self) -> None: ...

homeassistant-stubs/components/weather/group.pyi

@@ -1,3 +1,4 @@
+from .const import DOMAIN as DOMAIN
 from homeassistant.components.group import GroupIntegrationRegistry as GroupIntegrationRegistry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 

homeassistant-stubs/components/weather/intent.pyi

@@ -10,5 +10,7 @@ async def async_setup_intents(hass: HomeAssistant) -> None: ...
 
 class GetWeatherIntent(intent.IntentHandler):
     intent_type = INTENT_GET_WEATHER
+    description: str
     slot_schema: Incomplete
+    platforms: Incomplete
     async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse: ...

homeassistant-stubs/components/webostv/media_player.pyi

@@ -3,7 +3,7 @@ from .const import ATTR_PAYLOAD as ATTR_PAYLOAD, ATTR_SOUND_OUTPUT as ATTR_SOUND
 from .triggers.turn_on import async_get_turn_on_trigger as async_get_turn_on_trigger
 from _typeshed import Incomplete
 from aiowebostv import WebOsClient as WebOsClient
-from collections.abc import Awaitable, Callable as Callable, Coroutine
+from collections.abc import Awaitable as Awaitable, Callable as Callable, Coroutine
 from homeassistant import util as util
 from homeassistant.components.media_player import MediaPlayerDeviceClass as MediaPlayerDeviceClass, MediaPlayerEntity as MediaPlayerEntity, MediaPlayerEntityFeature as MediaPlayerEntityFeature, MediaPlayerState as MediaPlayerState, MediaType as MediaType
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -16,7 +16,7 @@ from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_d
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
 from homeassistant.helpers.trigger import PluggableAction as PluggableAction
-from typing import Any, Concatenate, ParamSpec, TypeVar
+from typing import Any, Concatenate
 
 _LOGGER: Incomplete
 SUPPORT_WEBOSTV: Incomplete
@@ -26,9 +26,6 @@ MIN_TIME_BETWEEN_FORCED_SCANS: Incomplete
 SCAN_INTERVAL: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
-_T = TypeVar('_T', bound='LgWebOSMediaPlayerEntity')
-_P = ParamSpec('_P')
-
 def cmd(func: Callable[Concatenate[_T, _P], Awaitable[None]]) -> Callable[Concatenate[_T, _P], Coroutine[Any, Any, None]]: ...
 
 class LgWebOSMediaPlayerEntity(RestoreEntity, MediaPlayerEntity):

homeassistant-stubs/components/websocket_api/commands.pyi

@@ -12,9 +12,9 @@ from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, S
 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 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.json import ExtendedJSONEncoder as ExtendedJSONEncoder, JSON_DUMP as JSON_DUMP, find_paths_unserializable_data as find_paths_unserializable_data, json_bytes as json_bytes, json_fragment as json_fragment
 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
+from homeassistant.loader import 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
 from homeassistant.setup import async_get_loaded_integrations as async_get_loaded_integrations, async_get_setup_timings as async_get_setup_timings
 from homeassistant.util.json import format_unserializable_data as format_unserializable_data
 from typing import Any
@@ -24,8 +24,8 @@ _LOGGER: Incomplete
 
 def async_register_commands(hass: HomeAssistant, async_reg: Callable[[HomeAssistant, const.WebSocketCommandHandler], None]) -> None: ...
 def pong_message(iden: int) -> dict[str, Any]: ...
-def _forward_events_check_permissions(send_message: Callable[[bytes | str | dict[str, Any] | Callable[[], str]], None], user: User, msg_id: int, event: Event) -> None: ...
-def _forward_events_unconditional(send_message: Callable[[bytes | str | dict[str, Any] | Callable[[], str]], None], msg_id: int, event: Event) -> None: ...
+def _forward_events_check_permissions(send_message: Callable[[bytes | str | dict[str, Any]], None], user: User, message_id_as_bytes: bytes, event: Event) -> None: ...
+def _forward_events_unconditional(send_message: Callable[[bytes | str | dict[str, Any]], None], message_id_as_bytes: bytes, event: Event) -> None: ...
 def handle_subscribe_events(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]) -> None: ...
 def handle_subscribe_bootstrap_integrations(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]) -> None: ...
 def handle_unsubscribe_events(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]) -> None: ...
@@ -33,7 +33,7 @@ async def handle_call_service(hass: HomeAssistant, connection: ActiveConnection,
 def _async_get_allowed_states(hass: HomeAssistant, connection: ActiveConnection) -> list[State]: ...
 def handle_get_states(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]) -> None: ...
 def _send_handle_get_states_response(connection: ActiveConnection, msg_id: int, serialized_states: list[bytes]) -> None: ...
-def _forward_entity_changes(send_message: Callable[[str | bytes | dict[str, Any] | Callable[[], str]], None], entity_ids: set[str], user: User, msg_id: int, event: Event[EventStateChangedData]) -> None: ...
+def _forward_entity_changes(send_message: Callable[[str | bytes | dict[str, Any]], None], entity_ids: set[str], user: User, message_id_as_bytes: bytes, event: Event[EventStateChangedData]) -> None: ...
 def handle_subscribe_entities(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]) -> None: ...
 def _send_handle_entities_init_response(connection: ActiveConnection, msg_id: int, serialized_states: list[bytes]) -> None: ...
 async def _async_get_all_descriptions_json(hass: HomeAssistant) -> bytes: ...

homeassistant-stubs/components/websocket_api/connection.pyi

@@ -12,8 +12,8 @@ from homeassistant.util.json import JsonValueType as JsonValueType
 from typing import Any
 
 current_connection: Incomplete
-MessageHandler: Incomplete
-BinaryHandler: Incomplete
+MessageHandler = Callable[[HomeAssistant, ActiveConnection, dict[str, Any]], None]
+BinaryHandler = Callable[[HomeAssistant, ActiveConnection, bytes], None]
 
 class ActiveConnection:
     __slots__: Incomplete

homeassistant-stubs/components/websocket_api/const.pyi

@@ -1,15 +1,17 @@
 from .connection import ActiveConnection as ActiveConnection
 from _typeshed import Incomplete
+from collections.abc import Awaitable, Callable
 from homeassistant.core import HomeAssistant as HomeAssistant
-from typing import Final
+from typing import Any, Final
 
-WebSocketCommandHandler: Incomplete
-AsyncWebSocketCommandHandler: Incomplete
+WebSocketCommandHandler = Callable[[HomeAssistant, ActiveConnection, dict[str, Any]], None]
+AsyncWebSocketCommandHandler = Callable[[HomeAssistant, ActiveConnection, dict[str, Any]], Awaitable[None]]
 DOMAIN: Final[str]
 URL: Final[str]
 PENDING_MSG_PEAK: Final[int]
 PENDING_MSG_PEAK_TIME: Final[int]
 MAX_PENDING_MSG: Final[int]
+PENDING_MSG_MAX_FORCE_READY: Final[int]
 ERR_ID_REUSE: Final[str]
 ERR_INVALID_FORMAT: Final[str]
 ERR_NOT_ALLOWED: Final[str]

homeassistant-stubs/components/websocket_api/http.pyi

@@ -2,7 +2,7 @@ import datetime as dt
 import logging
 from .auth import AUTH_REQUIRED_MESSAGE as AUTH_REQUIRED_MESSAGE, AuthPhase as AuthPhase
 from .connection import ActiveConnection as ActiveConnection
-from .const import DATA_CONNECTIONS as DATA_CONNECTIONS, MAX_PENDING_MSG as MAX_PENDING_MSG, PENDING_MSG_PEAK as PENDING_MSG_PEAK, PENDING_MSG_PEAK_TIME as PENDING_MSG_PEAK_TIME, SIGNAL_WEBSOCKET_CONNECTED as SIGNAL_WEBSOCKET_CONNECTED, SIGNAL_WEBSOCKET_DISCONNECTED as SIGNAL_WEBSOCKET_DISCONNECTED, URL as URL
+from .const import DATA_CONNECTIONS as DATA_CONNECTIONS, MAX_PENDING_MSG as MAX_PENDING_MSG, PENDING_MSG_MAX_FORCE_READY as PENDING_MSG_MAX_FORCE_READY, PENDING_MSG_PEAK as PENDING_MSG_PEAK, PENDING_MSG_PEAK_TIME as PENDING_MSG_PEAK_TIME, SIGNAL_WEBSOCKET_CONNECTED as SIGNAL_WEBSOCKET_CONNECTED, SIGNAL_WEBSOCKET_DISCONNECTED as SIGNAL_WEBSOCKET_DISCONNECTED, URL as URL
 from .error import Disconnect as Disconnect
 from .messages import message_to_json_bytes as message_to_json_bytes
 from .util import describe_request as describe_request
@@ -32,6 +32,7 @@ class WebSocketAdapter(logging.LoggerAdapter):
 class WebSocketHandler:
     __slots__: Incomplete
     _hass: Incomplete
+    _loop: Incomplete
     _request: Incomplete
     _wsock: Incomplete
     _handle_task: Incomplete
@@ -43,6 +44,7 @@ class WebSocketHandler:
     _connection: Incomplete
     _message_queue: Incomplete
     _ready_future: Incomplete
+    _release_ready_queue_size: int
     def __init__(self, hass: HomeAssistant, request: web.Request) -> None: ...
     def __repr__(self) -> str: ...
     @property
@@ -50,6 +52,7 @@ class WebSocketHandler:
     async def _writer(self, send_bytes_text: Callable[[bytes], Coroutine[Any, Any, None]]) -> None: ...
     def _cancel_peak_checker(self) -> None: ...
     def _send_message(self, message: str | bytes | dict[str, Any]) -> None: ...
+    def _release_ready_future_or_reschedule(self) -> None: ...
     def _check_write_peak(self, _utc_time: dt.datetime) -> None: ...
     def _cancel(self) -> None: ...
     def _async_handle_hass_stop(self, event: Event) -> None: ...

homeassistant-stubs/components/websocket_api/messages.pyi

@@ -1,7 +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, EventStateChangedData as EventStateChangedData, State as State
+from homeassistant.core import CompressedState as CompressedState, Event as Event, EventStateChangedData as EventStateChangedData
 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
@@ -21,11 +21,10 @@ def result_message(iden: int, result: Any = None) -> dict[str, Any]: ...
 def construct_result_message(iden: int, payload: bytes) -> bytes: ...
 def error_message(iden: int | None, code: str, message: str, translation_key: str | None = None, translation_domain: str | None = None, translation_placeholders: dict[str, Any] | None = None) -> dict[str, Any]: ...
 def event_message(iden: int, event: Any) -> dict[str, Any]: ...
-def cached_event_message(iden: int, event: Event) -> bytes: ...
+def cached_event_message(message_id_as_bytes: bytes, event: Event) -> bytes: ...
 def _partial_cached_event_message(event: Event) -> bytes: ...
-def cached_state_diff_message(iden: int, event: Event[EventStateChangedData]) -> bytes: ...
+def cached_state_diff_message(message_id_as_bytes: bytes, event: Event[EventStateChangedData]) -> bytes: ...
 def _partial_cached_state_diff_message(event: Event[EventStateChangedData]) -> bytes: ...
-def _state_diff_event(event: Event[EventStateChangedData]) -> dict: ...
-def _state_diff(old_state: State, new_state: State) -> dict[str, dict[str, dict[str, dict[str, str | list[str]]]]]: ...
+def _state_diff_event(event: Event[EventStateChangedData]) -> dict[str, list[str] | dict[str, CompressedState] | dict[str, dict[str, dict[str, str | list[str]]]]]: ...
 def _message_to_json_bytes_or_none(message: dict[str, Any]) -> bytes | None: ...
 def message_to_json_bytes(message: dict[str, Any]) -> bytes: ...

homeassistant-stubs/components/wemo/__init__.pyi

@@ -1,7 +1,7 @@
 import pywemo
 from .const import DOMAIN as DOMAIN
+from .coordinator import DeviceCoordinator as DeviceCoordinator, async_register_device as async_register_device
 from .models import WemoConfigEntryData as WemoConfigEntryData, WemoData as WemoData, async_wemo_data as async_wemo_data
-from .wemo_device import DeviceCoordinator as DeviceCoordinator, async_register_device as async_register_device
 from _typeshed import Incomplete
 from collections.abc import Callable, Coroutine, Sequence
 from datetime import datetime

homeassistant-stubs/components/wemo/binary_sensor.pyi

@@ -1,6 +1,6 @@
 from . import async_wemo_dispatcher_connect as async_wemo_dispatcher_connect
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from .entity import WemoBinaryStateEntity as WemoBinaryStateEntity, WemoEntity as WemoEntity
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
 from homeassistant.components.binary_sensor import BinarySensorEntity as BinarySensorEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/components/wemo/config_flow.pyi

@@ -1,6 +1,6 @@
 import voluptuous as vol
 from .const import DOMAIN as DOMAIN
-from .wemo_device import Options as Options, OptionsValidationError as OptionsValidationError
+from .coordinator import Options as Options, OptionsValidationError as OptionsValidationError
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback

homeassistant-stubs/components/wemo/device_trigger.pyi

@@ -1,5 +1,5 @@
 from .const import WEMO_SUBSCRIPTION_EVENT as WEMO_SUBSCRIPTION_EVENT
-from .wemo_device import async_get_coordinator as async_get_coordinator
+from .coordinator import async_get_coordinator as async_get_coordinator
 from _typeshed import Incomplete
 from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA as DEVICE_TRIGGER_BASE_SCHEMA
 from homeassistant.const import CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_DOMAIN as CONF_DOMAIN, CONF_PLATFORM as CONF_PLATFORM, CONF_TYPE as CONF_TYPE

homeassistant-stubs/components/wemo/entity.pyi

@@ -1,4 +1,4 @@
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from _typeshed import Incomplete
 from collections.abc import Generator
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo

homeassistant-stubs/components/wemo/fan.pyi

@@ -1,7 +1,7 @@
 from . import async_wemo_dispatcher_connect as async_wemo_dispatcher_connect
 from .const import SERVICE_RESET_FILTER_LIFE as SERVICE_RESET_FILTER_LIFE, SERVICE_SET_HUMIDITY as SERVICE_SET_HUMIDITY
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from .entity import WemoBinaryStateEntity as WemoBinaryStateEntity
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
 from _typeshed import Incomplete
 from homeassistant.components.fan import FanEntity as FanEntity, FanEntityFeature as FanEntityFeature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/wemo/light.pyi

@@ -1,6 +1,6 @@
 from . import async_wemo_dispatcher_connect as async_wemo_dispatcher_connect
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from .entity import WemoBinaryStateEntity as WemoBinaryStateEntity, WemoEntity as WemoEntity
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
 from _typeshed import Incomplete
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP as ATTR_COLOR_TEMP, ATTR_HS_COLOR as ATTR_HS_COLOR, ATTR_TRANSITION as ATTR_TRANSITION, ColorMode as ColorMode, LightEntity as LightEntity, LightEntityFeature as LightEntityFeature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/wemo/models.pyi

@@ -1,14 +1,14 @@
 import pywemo
 from . import HostPortTuple as HostPortTuple, WemoDiscovery as WemoDiscovery, WemoDispatcher as WemoDispatcher
 from .const import DOMAIN as DOMAIN
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from collections.abc import Sequence
 from dataclasses import dataclass
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 
 @dataclass
 class WemoConfigEntryData:
-    device_coordinators: dict[str, 'DeviceCoordinator']
+    device_coordinators: dict[str, DeviceCoordinator]
     discovery: WemoDiscovery
     dispatcher: WemoDispatcher
     def __init__(self, device_coordinators, discovery, dispatcher) -> None: ...
@@ -16,7 +16,7 @@ class WemoConfigEntryData:
 @dataclass
 class WemoData:
     discovery_enabled: bool
-    static_config: Sequence['HostPortTuple']
+    static_config: Sequence[HostPortTuple]
     registry: pywemo.SubscriptionRegistry
     config_entry_data: WemoConfigEntryData = ...
     def __init__(self, discovery_enabled, static_config, registry, config_entry_data) -> None: ...

homeassistant-stubs/components/wemo/sensor.pyi

@@ -1,6 +1,6 @@
 from . import async_wemo_dispatcher_connect as async_wemo_dispatcher_connect
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from .entity import WemoEntity as WemoEntity
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass

homeassistant-stubs/components/wemo/switch.pyi

@@ -1,6 +1,6 @@
 from . import async_wemo_dispatcher_connect as async_wemo_dispatcher_connect
+from .coordinator import DeviceCoordinator as DeviceCoordinator
 from .entity import WemoBinaryStateEntity as WemoBinaryStateEntity
-from .wemo_device import DeviceCoordinator as DeviceCoordinator
 from _typeshed import Incomplete
 from homeassistant.components.switch import SwitchEntity as SwitchEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/withings/__init__.pyi

@@ -19,6 +19,7 @@ PLATFORMS: Incomplete
 SUBSCRIBE_DELAY: Incomplete
 UNSUBSCRIBE_DELAY: Incomplete
 CONF_CLOUDHOOK_URL: str
+WithingsConfigEntry = ConfigEntry[WithingsData]
 
 @dataclass(slots=True)
 class WithingsData:
@@ -33,8 +34,8 @@ class WithingsData:
     def __post_init__(self) -> None: ...
     def __init__(self, client, measurement_coordinator, sleep_coordinator, bed_presence_coordinator, goals_coordinator, activity_coordinator, workout_coordinator, coordinators) -> None: ...
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: WithingsConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: WithingsConfigEntry) -> bool: ...
 async def async_subscribe_webhooks(client: WithingsClient, webhook_url: str) -> None: ...
 
 class WithingsWebhookManager:
@@ -42,14 +43,14 @@ class WithingsWebhookManager:
     _register_lock: Incomplete
     hass: Incomplete
     entry: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+    def __init__(self, hass: HomeAssistant, entry: WithingsConfigEntry) -> None: ...
     @property
     def withings_data(self) -> WithingsData: ...
     async def unregister_webhook(self, _: Any) -> None: ...
     async def register_webhook(self, _: Any) -> None: ...
 
 async def async_unsubscribe_webhooks(client: WithingsClient) -> None: ...
-async def _async_cloudhook_generate_url(hass: HomeAssistant, entry: ConfigEntry) -> str: ...
-async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+async def _async_cloudhook_generate_url(hass: HomeAssistant, entry: WithingsConfigEntry) -> str: ...
+async def async_remove_entry(hass: HomeAssistant, entry: WithingsConfigEntry) -> None: ...
 def json_message_response(message: str, message_code: int) -> Response: ...
 def get_webhook_handler(withings_data: WithingsData) -> Callable[[HomeAssistant, str, Request], Awaitable[Response | None]]: ...

homeassistant-stubs/components/withings/binary_sensor.pyi

@@ -1,15 +1,15 @@
+from . import WithingsConfigEntry as WithingsConfigEntry
 from .const import DOMAIN as DOMAIN
 from .coordinator import WithingsBedPresenceDataUpdateCoordinator as WithingsBedPresenceDataUpdateCoordinator
 from .entity import WithingsEntity as WithingsEntity
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: WithingsConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class WithingsBinarySensor(WithingsEntity, BinarySensorEntity):
     _attr_translation_key: str

homeassistant-stubs/components/withings/calendar.pyi

@@ -1,4 +1,4 @@
-from . import DOMAIN as DOMAIN, WithingsData as WithingsData
+from . import DOMAIN as DOMAIN, WithingsConfigEntry as WithingsConfigEntry
 from .coordinator import WithingsWorkoutDataUpdateCoordinator as WithingsWorkoutDataUpdateCoordinator
 from .entity import WithingsEntity as WithingsEntity
 from _typeshed import Incomplete
@@ -6,12 +6,11 @@ from aiowithings import WithingsClient as WithingsClient, WorkoutCategory as Wor
 from collections.abc import Callable as Callable
 from datetime import datetime
 from homeassistant.components.calendar import CalendarEntity as CalendarEntity, CalendarEvent as CalendarEvent
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: WithingsConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 def get_event_name(category: WorkoutCategory) -> str: ...
 
 class WithingsWorkoutCalendarEntity(WithingsEntity[WithingsWorkoutDataUpdateCoordinator], CalendarEntity):

homeassistant-stubs/components/withings/coordinator.pyi

@@ -1,20 +1,18 @@
 import abc
+from . import WithingsConfigEntry as WithingsConfigEntry
 from .const import LOGGER as LOGGER
 from _typeshed import Incomplete
 from abc import abstractmethod
 from aiowithings import Activity, Goals, MeasurementType, NotificationCategory, SleepSummary, WithingsClient as WithingsClient, Workout
 from datetime import datetime, timedelta
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from typing import TypeVar
 
-_T = TypeVar('_T')
 UPDATE_INTERVAL: Incomplete
 
-class WithingsDataUpdateCoordinator(DataUpdateCoordinator[_T], metaclass=abc.ABCMeta):
-    config_entry: ConfigEntry
+class WithingsDataUpdateCoordinator(DataUpdateCoordinator[_DataT], metaclass=abc.ABCMeta):
+    config_entry: WithingsConfigEntry
     _default_update_interval: timedelta | None
     _last_valid_update: datetime | None
     webhooks_connected: bool
@@ -26,9 +24,9 @@ class WithingsDataUpdateCoordinator(DataUpdateCoordinator[_T], metaclass=abc.ABC
     update_interval: Incomplete
     def webhook_subscription_listener(self, connected: bool) -> None: ...
     async def async_webhook_data_updated(self, notification_category: NotificationCategory) -> None: ...
-    async def _async_update_data(self) -> _T: ...
+    async def _async_update_data(self) -> _DataT: ...
     @abstractmethod
-    async def _internal_update_data(self) -> _T: ...
+    async def _internal_update_data(self) -> _DataT: ...
 
 class WithingsMeasurementDataUpdateCoordinator(WithingsDataUpdateCoordinator[dict[MeasurementType, float]]):
     coordinator_name: str

homeassistant-stubs/components/withings/diagnostics.pyi

@@ -1,8 +1,6 @@
-from . import CONF_CLOUDHOOK_URL as CONF_CLOUDHOOK_URL, WithingsData as WithingsData
-from .const import DOMAIN as DOMAIN
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from . import CONF_CLOUDHOOK_URL as CONF_CLOUDHOOK_URL, WithingsConfigEntry as WithingsConfigEntry
 from homeassistant.const import CONF_WEBHOOK_ID as CONF_WEBHOOK_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: WithingsConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/withings/entity.pyi

@@ -3,9 +3,6 @@ from .coordinator import WithingsDataUpdateCoordinator as WithingsDataUpdateCoor
 from _typeshed import Incomplete
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from typing import TypeVar
-
-_T = TypeVar('_T', bound=WithingsDataUpdateCoordinator)
 
 class WithingsEntity(CoordinatorEntity[_T]):
     _attr_has_entity_name: bool

homeassistant-stubs/components/withings/sensor.pyi

@@ -1,4 +1,4 @@
-from . import WithingsData as WithingsData
+from . import WithingsConfigEntry as WithingsConfigEntry
 from .const import DOMAIN as DOMAIN, LOGGER as LOGGER, SCORE_POINTS as SCORE_POINTS, UOM_BEATS_PER_MINUTE as UOM_BEATS_PER_MINUTE, UOM_BREATHS_PER_MINUTE as UOM_BREATHS_PER_MINUTE, UOM_FREQUENCY as UOM_FREQUENCY, UOM_MMHG as UOM_MMHG
 from .coordinator import WithingsActivityDataUpdateCoordinator as WithingsActivityDataUpdateCoordinator, WithingsDataUpdateCoordinator as WithingsDataUpdateCoordinator, WithingsGoalsDataUpdateCoordinator as WithingsGoalsDataUpdateCoordinator, WithingsMeasurementDataUpdateCoordinator as WithingsMeasurementDataUpdateCoordinator, WithingsSleepDataUpdateCoordinator as WithingsSleepDataUpdateCoordinator, WithingsWorkoutDataUpdateCoordinator as WithingsWorkoutDataUpdateCoordinator
 from .entity import WithingsEntity as WithingsEntity
@@ -8,12 +8,10 @@ from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from datetime import datetime
 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 PERCENTAGE as PERCENTAGE, Platform as Platform, UnitOfLength as UnitOfLength, UnitOfMass as UnitOfMass, UnitOfSpeed as UnitOfSpeed, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
-from typing import Generic, TypeVar
 
 @dataclass(frozen=True, kw_only=True)
 class WithingsMeasurementSensorEntityDescription(SensorEntityDescription):
@@ -55,11 +53,9 @@ _WORKOUT_CATEGORY: Incomplete
 WORKOUT_SENSORS: Incomplete
 
 def get_current_goals(goals: Goals) -> set[str]: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
-_T = TypeVar('_T', bound=WithingsDataUpdateCoordinator)
-_ED = TypeVar('_ED', bound=SensorEntityDescription)
+async def async_setup_entry(hass: HomeAssistant, entry: WithingsConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class WithingsSensor(WithingsEntity[_T], SensorEntity, Generic[_T, _ED]):
+class WithingsSensor(WithingsEntity[_T], SensorEntity):
     entity_description: _ED
     def __init__(self, coordinator: _T, entity_description: _ED) -> None: ...
 

homeassistant-stubs/components/wled/__init__.pyi

@@ -1,4 +1,4 @@
-from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
+from .const import LOGGER as LOGGER
 from .coordinator import WLEDDataUpdateCoordinator as WLEDDataUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -6,7 +6,8 @@ from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 
 PLATFORMS: Incomplete
+WLEDConfigEntry = ConfigEntry[WLEDDataUpdateCoordinator]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: WLEDConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: WLEDConfigEntry) -> bool: ...
 async def async_reload_entry(hass: HomeAssistant, entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/wled/binary_sensor.pyi

@@ -1,14 +1,13 @@
-from .const import DOMAIN as DOMAIN
+from . import WLEDConfigEntry as WLEDConfigEntry
 from .coordinator import WLEDDataUpdateCoordinator as WLEDDataUpdateCoordinator
-from .models import WLEDEntity as WLEDEntity
+from .entity import WLEDEntity as WLEDEntity
 from _typeshed import Incomplete
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: WLEDConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class WLEDUpdateBinarySensor(WLEDEntity, BinarySensorEntity):
     _attr_entity_category: Incomplete

homeassistant-stubs/components/wled/button.pyi

@@ -1,15 +1,14 @@
-from .const import DOMAIN as DOMAIN
+from . import WLEDConfigEntry as WLEDConfigEntry
 from .coordinator import WLEDDataUpdateCoordinator as WLEDDataUpdateCoordinator
+from .entity import WLEDEntity as WLEDEntity
 from .helpers import wled_exception_handler as wled_exception_handler
-from .models import WLEDEntity as WLEDEntity
 from _typeshed import Incomplete
 from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: WLEDConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class WLEDRestartButton(WLEDEntity, ButtonEntity):
     _attr_device_class: Incomplete

homeassistant-stubs/components/wled/diagnostics.pyi

@@ -1,8 +1,6 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import WLEDDataUpdateCoordinator as WLEDDataUpdateCoordinator
+from . import WLEDConfigEntry as WLEDConfigEntry
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: WLEDConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/wled/helpers.pyi

@@ -1,9 +1,6 @@
-from .models import WLEDEntity as WLEDEntity
+from .entity import WLEDEntity as WLEDEntity
 from collections.abc import Callable as Callable, Coroutine
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
-from typing import Any, Concatenate, ParamSpec, TypeVar
-
-_WLEDEntityT = TypeVar('_WLEDEntityT', bound=WLEDEntity)
-_P = ParamSpec('_P')
+from typing import Any, Concatenate
 
 def wled_exception_handler(func: Callable[Concatenate[_WLEDEntityT, _P], Coroutine[Any, Any, Any]]) -> Callable[Concatenate[_WLEDEntityT, _P], Coroutine[Any, Any, None]]: ...