homeassistant-stubs 2025.2.4__py3-none-any.whl → 2025.3.0__py3-none-any.whl

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,10 +1,8 @@
 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
@@ -14,19 +12,12 @@ 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]) -> 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]) -> 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]) -> 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,6 +2,7 @@ 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
@@ -21,8 +22,15 @@ from roborock.version_a01_apis import RoborockClientA01 as RoborockClientA01
 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 +39,7 @@ 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: ...
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: HomeDataDevice, device_networking: NetworkInfo, product_info: HomeDataProduct, cloud_api: RoborockMqttClientV1, home_data_rooms: list[HomeDataRoom]) -> None: ...
     async def _async_setup(self) -> None: ...
     async def _verify_api(self) -> None: ...
     async def async_shutdown(self) -> None: ...
@@ -45,11 +53,12 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
     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,11 @@ 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: ...
+_LOGGER: Incomplete
+
+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/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

homeassistant-stubs/components/roborock/switch.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.switch import SwitchEntity as SwitchEntity, Switch
 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
 from roborock.command_cache import CacheableAttribute
 from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
 from typing import Any
@@ -23,7 +23,7 @@ class RoborockSwitchDescription(SwitchEntityDescription):
 
 SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription]
 
-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 RoborockSwitch(RoborockEntityV1, SwitchEntity):
     entity_description: RoborockSwitchDescription

homeassistant-stubs/components/roborock/time.pyi

@@ -1,6 +1,6 @@
 import datetime
-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
@@ -10,7 +10,7 @@ from homeassistant.components.time import TimeEntity as TimeEntity, TimeEntityDe
 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
 from roborock.command_cache import CacheableAttribute
 from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
 from typing import Any
@@ -25,7 +25,7 @@ class RoborockTimeDescription(TimeEntityDescription):
 
 TIME_DESCRIPTIONS: list[RoborockTimeDescription]
 
-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 RoborockTimeEntity(RoborockEntityV1, TimeEntity):
     entity_description: RoborockTimeDescription

homeassistant-stubs/components/roborock/vacuum.pyi

@@ -1,6 +1,5 @@
-from . import RoborockConfigEntry as RoborockConfigEntry
 from .const import DOMAIN as DOMAIN, GET_MAPS_SERVICE_NAME as GET_MAPS_SERVICE_NAME, GET_VACUUM_CURRENT_POSITION_SERVICE_NAME as GET_VACUUM_CURRENT_POSITION_SERVICE_NAME, SET_VACUUM_GOTO_POSITION_SERVICE_NAME as SET_VACUUM_GOTO_POSITION_SERVICE_NAME
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from .image import ColorsPalette as ColorsPalette, ImageConfig as ImageConfig, RoborockMapDataParser as RoborockMapDataParser, Sizes as Sizes
 from _typeshed import Incomplete
@@ -8,12 +7,12 @@ from homeassistant.components.vacuum import StateVacuumEntity as StateVacuumEnti
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import entity_platform as entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
 STATE_CODE_TO_STATE: 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 RoborockVacuum(RoborockCoordinatedEntityV1, StateVacuumEntity):
     _attr_icon: str

homeassistant-stubs/components/roku/__init__.pyi

@@ -1,12 +1,9 @@
-from .const import CONF_PLAY_MEDIA_APP_ID as CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID as DEFAULT_PLAY_MEDIA_APP_ID
-from .coordinator import RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
+from .coordinator import RokuConfigEntry as RokuConfigEntry, RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_HOST as CONF_HOST, Platform as Platform
+from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 
 PLATFORMS: Incomplete
-type RokuConfigEntry = ConfigEntry[RokuDataUpdateCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: RokuConfigEntry) -> bool: ...

homeassistant-stubs/components/roku/binary_sensor.pyi

@@ -1,11 +1,11 @@
-from . import RokuConfigEntry as RokuConfigEntry
+from .coordinator import RokuConfigEntry as RokuConfigEntry
 from .entity import RokuEntity as RokuEntity
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.binary_sensor import 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 rokuecp.models import Device as RokuDevice
 
 PARALLEL_UPDATES: int
@@ -16,7 +16,7 @@ class RokuBinarySensorEntityDescription(BinarySensorEntityDescription):
 
 BINARY_SENSORS: tuple[RokuBinarySensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RokuBinarySensorEntity(RokuEntity, BinarySensorEntity):
     entity_description: RokuBinarySensorEntityDescription

homeassistant-stubs/components/roku/config_flow.pyi

@@ -1,5 +1,5 @@
-from . import RokuConfigEntry as RokuConfigEntry
 from .const import CONF_PLAY_MEDIA_APP_ID as CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID as DEFAULT_PLAY_MEDIA_APP_ID, DOMAIN as DOMAIN
+from .coordinator import RokuConfigEntry as RokuConfigEntry
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_RECONFIGURE as SOURCE_RECONFIGURE
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_NAME as CONF_NAME

homeassistant-stubs/components/roku/coordinator.pyi

@@ -1,6 +1,8 @@
-from .const import DOMAIN as DOMAIN
+from .const import CONF_PLAY_MEDIA_APP_ID as CONF_PLAY_MEDIA_APP_ID, DEFAULT_PLAY_MEDIA_APP_ID as DEFAULT_PLAY_MEDIA_APP_ID, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from datetime import datetime
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_HOST as CONF_HOST
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.debounce import Debouncer as Debouncer
@@ -12,12 +14,14 @@ from rokuecp.models import Device
 REQUEST_REFRESH_DELAY: float
 SCAN_INTERVAL: Incomplete
 _LOGGER: Incomplete
+type RokuConfigEntry = ConfigEntry[RokuDataUpdateCoordinator]
 
 class RokuDataUpdateCoordinator(DataUpdateCoordinator[Device]):
+    config_entry: RokuConfigEntry
     last_full_update: datetime | None
     roku: Roku
     device_id: Incomplete
     play_media_app_id: Incomplete
     full_update_interval: Incomplete
-    def __init__(self, hass: HomeAssistant, *, host: str, device_id: str, play_media_app_id: str) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: RokuConfigEntry) -> None: ...
     async def _async_update_data(self) -> Device: ...

homeassistant-stubs/components/roku/diagnostics.pyi

@@ -1,4 +1,4 @@
-from . import RokuConfigEntry as RokuConfigEntry
+from .coordinator import RokuConfigEntry as RokuConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
 

homeassistant-stubs/components/roku/entity.pyi

@@ -1,5 +1,5 @@
-from . import RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
 from .const import DOMAIN as DOMAIN
+from .coordinator import RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity import EntityDescription as EntityDescription

homeassistant-stubs/components/roku/media_player.pyi

@@ -1,8 +1,7 @@
 import datetime as dt
-from . import RokuConfigEntry as RokuConfigEntry
 from .browse_media import async_browse_media as async_browse_media
 from .const import ATTR_ARTIST_NAME as ATTR_ARTIST_NAME, ATTR_CONTENT_ID as ATTR_CONTENT_ID, ATTR_FORMAT as ATTR_FORMAT, ATTR_KEYWORD as ATTR_KEYWORD, ATTR_MEDIA_TYPE as ATTR_MEDIA_TYPE, ATTR_THUMBNAIL as ATTR_THUMBNAIL, SERVICE_SEARCH as SERVICE_SEARCH
-from .coordinator import RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
+from .coordinator import RokuConfigEntry as RokuConfigEntry, RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
 from .entity import RokuEntity as RokuEntity
 from .helpers import format_channel_name as format_channel_name, roku_exception_handler as roku_exception_handler
 from _typeshed import Incomplete
@@ -12,7 +11,7 @@ from homeassistant.components.stream import FORMAT_CONTENT_TYPE as FORMAT_CONTEN
 from homeassistant.const import ATTR_NAME as ATTR_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers import entity_platform as entity_platform
-from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import VolDictType as VolDictType
 from typing import Any
 
@@ -24,7 +23,7 @@ ATTRS_TO_PLAY_ON_ROKU_AUDIO_PARAMS: Incomplete
 SEARCH_SCHEMA: VolDictType
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RokuMediaPlayer(RokuEntity, MediaPlayerEntity):
     _attr_name: Incomplete

homeassistant-stubs/components/roku/remote.pyi

@@ -1,16 +1,16 @@
-from . import RokuConfigEntry as RokuConfigEntry
+from .coordinator import RokuConfigEntry as RokuConfigEntry
 from .entity import RokuEntity as RokuEntity
 from .helpers import roku_exception_handler as roku_exception_handler
 from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.components.remote import ATTR_NUM_REPEATS as ATTR_NUM_REPEATS, RemoteEntity as RemoteEntity
 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 typing import Any
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RokuRemote(RokuEntity, RemoteEntity):
     _attr_name: Incomplete

homeassistant-stubs/components/roku/select.pyi

@@ -1,4 +1,4 @@
-from . import RokuConfigEntry as RokuConfigEntry
+from .coordinator import RokuConfigEntry as RokuConfigEntry
 from .entity import RokuEntity as RokuEntity
 from .helpers import format_channel_name as format_channel_name, roku_exception_handler as roku_exception_handler
 from _typeshed import Incomplete
@@ -6,7 +6,7 @@ from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
 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 rokuecp import Roku as Roku
 from rokuecp.models import Device as RokuDevice
 
@@ -28,7 +28,7 @@ class RokuSelectEntityDescription(SelectEntityDescription):
 ENTITIES: tuple[RokuSelectEntityDescription, ...]
 CHANNEL_ENTITY: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RokuSelectEntity(RokuEntity, SelectEntity):
     entity_description: RokuSelectEntityDescription

homeassistant-stubs/components/roku/sensor.pyi

@@ -1,11 +1,11 @@
-from . import RokuConfigEntry as RokuConfigEntry
+from .coordinator import RokuConfigEntry as RokuConfigEntry
 from .entity import RokuEntity as RokuEntity
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
 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 rokuecp.models import Device as RokuDevice
 
 PARALLEL_UPDATES: int
@@ -16,7 +16,7 @@ class RokuSensorEntityDescription(SensorEntityDescription):
 
 SENSORS: tuple[RokuSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: RokuConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class RokuSensorEntity(RokuEntity, SensorEntity):
     entity_description: RokuSensorEntityDescription