homeassistant-stubs 2025.2.5__py3-none-any.whl → 2025.3.0b0__py3-none-any.whl

homeassistant-stubs/components/ring/binary_sensor.pyi

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.const import Platform as Platform
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.event import async_call_at as async_call_at
 from ring_doorbell import RingCapability, RingEvent as RingEvent
 from typing import Any, Generic
@@ -20,7 +20,7 @@ class RingBinarySensorEntityDescription(BinarySensorEntityDescription, RingEntit
 
 BINARY_SENSOR_TYPES: tuple[RingBinarySensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingBinarySensor(RingBaseEntity[RingListenCoordinator, RingDeviceT], BinarySensorEntity):
     _active_alert: RingEvent | None

homeassistant-stubs/components/ring/button.pyi

@@ -4,13 +4,13 @@ from .entity import RingEntity as RingEntity, exception_wrap as exception_wrap
 from _typeshed import Incomplete
 from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from ring_doorbell import RingOther
 
 PARALLEL_UPDATES: int
 BUTTON_DESCRIPTION: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingDoorButton(RingEntity[RingOther], ButtonEntity):
     entity_description: Incomplete

homeassistant-stubs/components/ring/camera.pyi

@@ -1,4 +1,5 @@
 from . import RingConfigEntry as RingConfigEntry
+from .const import DOMAIN as DOMAIN
 from .coordinator import RingDataCoordinator as RingDataCoordinator
 from .entity import RingDeviceT as RingDeviceT, RingEntity as RingEntity, exception_wrap as exception_wrap
 from _typeshed import Incomplete
@@ -10,7 +11,7 @@ from homeassistant.components.camera import Camera as Camera, CameraEntityDescri
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.aiohttp_client import async_aiohttp_proxy_stream as async_aiohttp_proxy_stream
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from ring_doorbell import RingDoorBell
 from ring_doorbell.webrtcstream import RingWebRtcMessage as RingWebRtcMessage
 from typing import Any, Generic
@@ -28,7 +29,7 @@ class RingCameraEntityDescription(CameraEntityDescription, Generic[RingDeviceT])
 
 CAMERA_DESCRIPTIONS: tuple[RingCameraEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingCam(RingEntity[RingDoorBell], Camera):
     entity_description: Incomplete

homeassistant-stubs/components/ring/coordinator.pyi

@@ -1,4 +1,4 @@
-from .const import SCAN_INTERVAL as SCAN_INTERVAL
+from .const import DOMAIN as DOMAIN, SCAN_INTERVAL as SCAN_INTERVAL
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
@@ -19,13 +19,12 @@ class RingData:
     listen_coordinator: RingListenCoordinator
 type RingConfigEntry = ConfigEntry[RingData]
 
-async def _call_api[*_Ts, _R](hass: HomeAssistant, target: Callable[[*_Ts], Coroutine[Any, Any, _R]], *args: *_Ts, msg_suffix: str = '') -> _R: ...
-
 class RingDataCoordinator(DataUpdateCoordinator[RingDevices]):
     config_entry: RingConfigEntry
     ring_api: Ring
     first_call: bool
     def __init__(self, hass: HomeAssistant, config_entry: RingConfigEntry, ring_api: Ring) -> None: ...
+    async def _call_api[*_Ts, _R](self, target: Callable[[*_Ts], Coroutine[Any, Any, _R]], *args: *_Ts) -> _R: ...
     async def _async_update_data(self) -> RingDevices: ...
 
 class RingListenCoordinator(BaseDataUpdateCoordinatorProtocol):

homeassistant-stubs/components/ring/entity.pyi

@@ -18,6 +18,7 @@ from typing import Any, Concatenate, Generic, TypeVar
 
 RingDeviceT = TypeVar('RingDeviceT', bound=RingGeneric, default=RingGeneric)
 _RingCoordinatorT = TypeVar('_RingCoordinatorT', bound=RingDataCoordinator | RingListenCoordinator)
+_LOGGER: Incomplete
 
 @dataclass(slots=True)
 class DeprecatedInfo:

homeassistant-stubs/components/ring/event.pyi

@@ -5,7 +5,7 @@ from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.event import EventDeviceClass as EventDeviceClass, EventEntity as EventEntity, EventEntityDescription as EventEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from ring_doorbell import RingCapability, RingEvent as RingAlert
 from typing import Generic
 
@@ -17,7 +17,7 @@ class RingEventEntityDescription(EventEntityDescription, Generic[RingDeviceT]):
 
 EVENT_DESCRIPTIONS: tuple[RingEventEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingEvent(RingBaseEntity[RingListenCoordinator, RingDeviceT], EventEntity):
     entity_description: RingEventEntityDescription[RingDeviceT]

homeassistant-stubs/components/ring/light.pyi

@@ -5,7 +5,7 @@ from _typeshed import Incomplete
 from enum import StrEnum
 from homeassistant.components.light import ColorMode as ColorMode, LightEntity as LightEntity
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from ring_doorbell import RingStickUpCam
 from typing import Any
 
@@ -17,7 +17,7 @@ class OnOffState(StrEnum):
     ON = ...
     OFF = ...
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingLight(RingEntity[RingStickUpCam], LightEntity):
     _attr_color_mode: Incomplete

homeassistant-stubs/components/ring/number.pyi

@@ -6,14 +6,14 @@ from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription, NumberMode as NumberMode
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
 from ring_doorbell import RingGeneric as RingGeneric
 from typing import Any, Generic
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class RingNumberEntityDescription(NumberEntityDescription, Generic[RingDeviceT]):

homeassistant-stubs/components/ring/sensor.pyi

@@ -7,14 +7,14 @@ from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, Platform as Platform, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
 from ring_doorbell import RingEventKind, RingGeneric
 from typing import Any, Generic
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingSensor(RingEntity[RingDeviceT], SensorEntity):
     entity_description: RingSensorEntityDescription[RingDeviceT]

homeassistant-stubs/components/ring/siren.pyi

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.siren import ATTR_TONE as ATTR_TONE, SirenEntity as SirenEntity, SirenEntityDescription as SirenEntityDescription, SirenEntityFeature as SirenEntityFeature, SirenTurnOnServiceParameters as SirenTurnOnServiceParameters
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from ring_doorbell import RingGeneric as RingGeneric
 from typing import Any, Generic
 
@@ -24,7 +24,7 @@ class RingSirenEntityDescription(SirenEntityDescription, RingEntityDescription,
 
 SIRENS: tuple[RingSirenEntityDescription[Any], ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingSiren(RingEntity[RingDeviceT], SirenEntity):
     entity_description: RingSirenEntityDescription[RingDeviceT]

homeassistant-stubs/components/ring/switch.pyi

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any, Generic, Self
 
 _LOGGER: Incomplete
@@ -24,7 +24,7 @@ class RingSwitchEntityDescription(SwitchEntityDescription, RingEntityDescription
 
 SWITCHES: Sequence[RingSwitchEntityDescription[Any]]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RingConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RingSwitch(RingEntity[RingDeviceT], SwitchEntity):
     entity_description: RingSwitchEntityDescription[RingDeviceT]

homeassistant-stubs/components/rituals_perfume_genie/binary_sensor.pyi

@@ -8,7 +8,7 @@ from homeassistant.components.binary_sensor import BinarySensorDeviceClass as Bi
 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
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from pyrituals import Diffuser as Diffuser
 
 @dataclass(frozen=True, kw_only=True)
@@ -18,7 +18,7 @@ class RitualsBinarySensorEntityDescription(BinarySensorEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RitualsBinarySensorEntity(DiffuserEntity, BinarySensorEntity):
     entity_description: RitualsBinarySensorEntityDescription

homeassistant-stubs/components/rituals_perfume_genie/coordinator.pyi

@@ -1,6 +1,7 @@
 from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from datetime import timedelta
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
 from pyrituals import Diffuser as Diffuser
@@ -8,6 +9,7 @@ from pyrituals import Diffuser as Diffuser
 _LOGGER: Incomplete
 
 class RitualsDataUpdateCoordinator(DataUpdateCoordinator[None]):
+    config_entry: ConfigEntry
     diffuser: Incomplete
-    def __init__(self, hass: HomeAssistant, diffuser: Diffuser, update_interval: timedelta) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, diffuser: Diffuser, update_interval: timedelta) -> None: ...
     async def _async_update_data(self) -> None: ...

homeassistant-stubs/components/rituals_perfume_genie/number.pyi

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from pyrituals import Diffuser as Diffuser
 from typing import Any
 
@@ -18,7 +18,7 @@ class RitualsNumberEntityDescription(NumberEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RitualsNumberEntity(DiffuserEntity, NumberEntity):
     entity_description: RitualsNumberEntityDescription

homeassistant-stubs/components/rituals_perfume_genie/select.pyi

@@ -8,7 +8,7 @@ from homeassistant.components.select import SelectEntity as SelectEntity, Select
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, UnitOfArea as UnitOfArea
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from pyrituals import Diffuser as Diffuser
 
 @dataclass(frozen=True, kw_only=True)
@@ -18,7 +18,7 @@ class RitualsSelectEntityDescription(SelectEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RitualsSelectEntity(DiffuserEntity, SelectEntity):
     entity_description: RitualsSelectEntityDescription

homeassistant-stubs/components/rituals_perfume_genie/sensor.pyi

@@ -8,7 +8,7 @@ from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceCla
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from pyrituals import Diffuser as Diffuser
 
 @dataclass(frozen=True, kw_only=True)
@@ -18,7 +18,7 @@ class RitualsSensorEntityDescription(SensorEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RitualsSensorEntity(DiffuserEntity, SensorEntity):
     entity_description: RitualsSensorEntityDescription

homeassistant-stubs/components/rituals_perfume_genie/switch.pyi

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from pyrituals import Diffuser as Diffuser
 from typing import Any
 
@@ -19,7 +19,7 @@ class RitualsSwitchEntityDescription(SwitchEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RitualsSwitchEntity(DiffuserEntity, SwitchEntity):
     entity_description: RitualsSwitchEntityDescription

homeassistant-stubs/components/roborock/__init__.pyi

@@ -1,32 +1,24 @@
 from .const import CONF_BASE_URL as CONF_BASE_URL, CONF_USER_DATA as CONF_USER_DATA, DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockCoordinators as RoborockCoordinators, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
 from .roborock_storage import async_remove_map_storage as async_remove_map_storage
 from _typeshed import Incomplete
 from collections.abc import Coroutine
-from dataclasses import dataclass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_USERNAME as CONF_USERNAME, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from roborock import HomeDataRoom as HomeDataRoom
 from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, UserData
+from roborock.web_api import RoborockApiClient
 from typing import Any
 
 SCAN_INTERVAL: Incomplete
 _LOGGER: Incomplete
-type RoborockConfigEntry = ConfigEntry[RoborockCoordinators]
-
-@dataclass
-class RoborockCoordinators:
-    v1: list[RoborockDataUpdateCoordinator]
-    a01: list[RoborockDataUpdateCoordinatorA01]
-    def values(self) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]: ...
 
 async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> bool: ...
-def build_setup_functions(hass: HomeAssistant, device_map: dict[str, HomeDataDevice], user_data: UserData, product_info: dict[str, HomeDataProduct], home_data_rooms: list[HomeDataRoom]) -> list[Coroutine[Any, Any, RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None]]: ...
-async def setup_device(hass: HomeAssistant, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct, home_data_rooms: list[HomeDataRoom]) -> RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None: ...
-async def setup_device_v1(hass: HomeAssistant, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct, home_data_rooms: list[HomeDataRoom]) -> RoborockDataUpdateCoordinator | None: ...
-async def setup_device_a01(hass: HomeAssistant, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct) -> RoborockDataUpdateCoordinatorA01 | None: ...
+def build_setup_functions(hass: HomeAssistant, entry: RoborockConfigEntry, device_map: dict[str, HomeDataDevice], user_data: UserData, product_info: dict[str, HomeDataProduct], home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient) -> list[Coroutine[Any, Any, RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None]]: ...
+async def setup_device(hass: HomeAssistant, entry: RoborockConfigEntry, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct, home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient) -> RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None: ...
+async def setup_device_v1(hass: HomeAssistant, entry: RoborockConfigEntry, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct, home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient) -> RoborockDataUpdateCoordinator | None: ...
+async def setup_device_a01(hass: HomeAssistant, entry: RoborockConfigEntry, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct) -> RoborockDataUpdateCoordinatorA01 | None: ...
 async def async_unload_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> bool: ...
 async def update_listener(hass: HomeAssistant, entry: RoborockConfigEntry) -> None: ...
 async def async_remove_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> None: ...

homeassistant-stubs/components/roborock/binary_sensor.pyi

@@ -1,5 +1,4 @@
-from . import RoborockConfigEntry as RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
@@ -7,7 +6,7 @@ from dataclasses import dataclass
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from roborock.roborock_typing import DeviceProp as DeviceProp
 
 @dataclass(frozen=True, kw_only=True)
@@ -16,7 +15,7 @@ class RoborockBinarySensorDescription(BinarySensorEntityDescription):
 
 BINARY_SENSOR_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RoborockBinarySensorEntity(RoborockCoordinatedEntityV1, BinarySensorEntity):
     entity_description: RoborockBinarySensorDescription

homeassistant-stubs/components/roborock/button.pyi

@@ -1,12 +1,11 @@
-from . import RoborockConfigEntry as RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockEntityV1 as RoborockEntityV1
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from roborock.roborock_typing import RoborockCommand
 
 @dataclass(frozen=True, kw_only=True)
@@ -16,7 +15,7 @@ class RoborockButtonDescription(ButtonEntityDescription):
 
 CONSUMABLE_BUTTON_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RoborockButtonEntity(RoborockEntityV1, ButtonEntity):
     entity_description: RoborockButtonDescription

homeassistant-stubs/components/roborock/coordinator.pyi

@@ -2,27 +2,37 @@ from .const import DOMAIN as DOMAIN
 from .models import RoborockA01HassDeviceInfo as RoborockA01HassDeviceInfo, RoborockHassDeviceInfo as RoborockHassDeviceInfo, RoborockMapInfo as RoborockMapInfo
 from .roborock_storage import RoborockMapStorage as RoborockMapStorage
 from _typeshed import Incomplete
+from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_CONNECTIONS as ATTR_CONNECTIONS
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.typing import StateType as StateType
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 from homeassistant.util import slugify as slugify
 from propcache.api import cached_property
 from roborock import HomeDataRoom as HomeDataRoom
-from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, NetworkInfo as NetworkInfo
+from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, HomeDataScene as HomeDataScene, NetworkInfo as NetworkInfo, UserData as UserData
 from roborock.roborock_message import RoborockDyadDataProtocol, RoborockZeoProtocol
 from roborock.roborock_typing import DeviceProp
 from roborock.version_1_apis.roborock_local_client_v1 import RoborockLocalClientV1
 from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1 as RoborockMqttClientV1
 from roborock.version_a01_apis import RoborockClientA01 as RoborockClientA01
+from roborock.web_api import RoborockApiClient as RoborockApiClient
 
 SCAN_INTERVAL: Incomplete
 _LOGGER: Incomplete
 
+@dataclass
+class RoborockCoordinators:
+    v1: list[RoborockDataUpdateCoordinator]
+    a01: list[RoborockDataUpdateCoordinatorA01]
+    def values(self) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]: ...
+type RoborockConfigEntry = ConfigEntry[RoborockCoordinators]
+
 class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
-    config_entry: ConfigEntry
+    config_entry: RoborockConfigEntry
     roborock_device_info: Incomplete
     api: RoborockLocalClientV1 | RoborockMqttClientV1
     cloud_api: Incomplete
@@ -31,7 +41,9 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
     maps: dict[int, RoborockMapInfo]
     _home_data_rooms: Incomplete
     map_storage: Incomplete
-    def __init__(self, hass: HomeAssistant, device: HomeDataDevice, device_networking: NetworkInfo, product_info: HomeDataProduct, cloud_api: RoborockMqttClientV1, home_data_rooms: list[HomeDataRoom]) -> None: ...
+    _user_data: Incomplete
+    _api_client: Incomplete
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: HomeDataDevice, device_networking: NetworkInfo, product_info: HomeDataProduct, cloud_api: RoborockMqttClientV1, home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient, user_data: UserData) -> None: ...
     async def _async_setup(self) -> None: ...
     async def _verify_api(self) -> None: ...
     async def async_shutdown(self) -> None: ...
@@ -39,17 +51,20 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
     async def _async_update_data(self) -> DeviceProp: ...
     def _set_current_map(self) -> None: ...
     async def set_current_map_rooms(self) -> None: ...
+    async def get_scenes(self) -> list[HomeDataScene]: ...
+    async def execute_scene(self, scene_id: int) -> None: ...
     @cached_property
     def duid(self) -> str: ...
     @cached_property
     def duid_slug(self) -> str: ...
 
 class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]]):
+    config_entry: RoborockConfigEntry
     api: Incomplete
     device_info: Incomplete
     request_protocols: list[RoborockDyadDataProtocol | RoborockZeoProtocol]
     roborock_device_info: Incomplete
-    def __init__(self, hass: HomeAssistant, device: HomeDataDevice, product_info: HomeDataProduct, api: RoborockClientA01) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: HomeDataDevice, product_info: HomeDataProduct, api: RoborockClientA01) -> None: ...
     async def _async_update_data(self) -> dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]: ...
     async def async_shutdown(self) -> None: ...
     @cached_property

homeassistant-stubs/components/roborock/diagnostics.pyi

@@ -1,4 +1,4 @@
-from . import RoborockConfigEntry as RoborockConfigEntry
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
 from homeassistant.const import CONF_UNIQUE_ID as CONF_UNIQUE_ID

homeassistant-stubs/components/roborock/image.pyi

@@ -1,6 +1,5 @@
-from . import RoborockConfigEntry as RoborockConfigEntry
 from .const import DEFAULT_DRAWABLES as DEFAULT_DRAWABLES, DOMAIN as DOMAIN, DRAWABLES as DRAWABLES, IMAGE_CACHE_INTERVAL as IMAGE_CACHE_INTERVAL, MAP_FILE_FORMAT as MAP_FILE_FORMAT, MAP_SLEEP as MAP_SLEEP
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
@@ -10,9 +9,9 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
     _attr_has_entity_name: bool

homeassistant-stubs/components/roborock/number.pyi

@@ -1,5 +1,5 @@
-from . import DOMAIN as DOMAIN, RoborockConfigEntry as RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .const import DOMAIN as DOMAIN
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockEntityV1 as RoborockEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
@@ -8,7 +8,7 @@ from homeassistant.components.number import NumberEntity as NumberEntity, Number
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from roborock.command_cache import CacheableAttribute
 from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
 from typing import Any
@@ -22,7 +22,7 @@ class RoborockNumberDescription(NumberEntityDescription):
 
 NUMBER_DESCRIPTIONS: list[RoborockNumberDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RoborockNumberEntity(RoborockEntityV1, NumberEntity):
     entity_description: RoborockNumberDescription

homeassistant-stubs/components/roborock/scene.pyi

@@ -0,0 +1,18 @@
+from . import RoborockConfigEntry as RoborockConfigEntry
+from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .entity import RoborockEntity as RoborockEntity
+from _typeshed import Incomplete
+from homeassistant.components.scene import Scene as SceneEntity
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity import EntityDescription as EntityDescription
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from typing import Any
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class RoborockSceneEntity(RoborockEntity, SceneEntity):
+    entity_description: EntityDescription
+    _scene_id: Incomplete
+    _coordinator: Incomplete
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator, entity_description: EntityDescription) -> None: ...
+    async def async_activate(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/roborock/select.pyi

@@ -1,6 +1,5 @@
-from . import RoborockConfigEntry as RoborockConfigEntry
 from .const import MAP_SLEEP as MAP_SLEEP
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
@@ -8,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from roborock.containers import Status as Status
 from roborock.roborock_message import RoborockDataProtocol
 from roborock.roborock_typing import RoborockCommand
@@ -23,7 +22,7 @@ class RoborockSelectDescription(SelectEntityDescription):
 
 SELECT_DESCRIPTIONS: list[RoborockSelectDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
     entity_description: RoborockSelectDescription

homeassistant-stubs/components/roborock/sensor.pyi

@@ -1,6 +1,5 @@
 import datetime
-from . import RoborockConfigEntry as RoborockConfigEntry
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
 from .entity import RoborockCoordinatedEntityA01 as RoborockCoordinatedEntityA01, RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
@@ -8,7 +7,7 @@ from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfArea as UnitOfArea, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
 from roborock.roborock_message import RoborockDataProtocol, RoborockDyadDataProtocol, RoborockZeoProtocol
 from roborock.roborock_typing import DeviceProp as DeviceProp
@@ -27,7 +26,7 @@ def _dock_error_value_fn(properties: DeviceProp) -> str | None: ...
 SENSOR_DESCRIPTIONS: Incomplete
 A01_SENSOR_DESCRIPTIONS: list[RoborockSensorDescriptionA01]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
     entity_description: RoborockSensorDescription