homeassistant-stubs 2024.7.3__py3-none-any.whl → 2024.8.0__py3-none-any.whl

homeassistant-stubs/components/ring/sensor.pyi

@@ -34,6 +34,6 @@ class RingSensorEntityDescription(SensorEntityDescription, Generic[RingDeviceT])
     value_fn: Callable[[RingDeviceT], StateType] = ...
     exists_fn: Callable[[RingGeneric], bool] = ...
     extra_state_attributes_fn: Callable[[RingDeviceT], dict[str, Any] | None] = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn, exists_fn, extra_state_attributes_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn=..., exists_fn=..., extra_state_attributes_fn=...) -> None: ...
 
 SENSOR_TYPES: tuple[RingSensorEntityDescription[Any], ...]

homeassistant-stubs/components/rituals_perfume_genie/binary_sensor.pyi

@@ -15,7 +15,7 @@ from pyrituals import Diffuser as Diffuser
 class RitualsBinarySensorEntityDescription(BinarySensorEntityDescription):
     is_on_fn: Callable[[Diffuser], bool]
     has_fn: Callable[[Diffuser], bool]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, is_on_fn, has_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., is_on_fn, has_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/rituals_perfume_genie/number.pyi

@@ -15,7 +15,7 @@ from typing import Any
 class RitualsNumberEntityDescription(NumberEntityDescription):
     value_fn: Callable[[Diffuser], int]
     set_value_fn: Callable[[Diffuser, int], Awaitable[Any]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, value_fn, set_value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., max_value=..., min_value=..., mode=..., native_max_value=..., native_min_value=..., native_step=..., native_unit_of_measurement=..., step=..., value_fn, set_value_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/rituals_perfume_genie/select.pyi

@@ -15,7 +15,7 @@ from pyrituals import Diffuser as Diffuser
 class RitualsSelectEntityDescription(SelectEntityDescription):
     current_fn: Callable[[Diffuser], str]
     select_fn: Callable[[Diffuser, str], Awaitable[None]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, options, current_fn, select_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., options=..., current_fn, select_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/rituals_perfume_genie/sensor.pyi

@@ -15,7 +15,7 @@ from pyrituals import Diffuser as Diffuser
 class RitualsSensorEntityDescription(SensorEntityDescription):
     has_fn: Callable[[Diffuser], bool] = ...
     value_fn: Callable[[Diffuser], int | str]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, has_fn, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., has_fn=..., value_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/rituals_perfume_genie/switch.pyi

@@ -16,7 +16,7 @@ class RitualsSwitchEntityDescription(SwitchEntityDescription):
     is_on_fn: Callable[[Diffuser], bool]
     turn_on_fn: Callable[[Diffuser], Awaitable[None]]
     turn_off_fn: Callable[[Diffuser], Awaitable[None]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, is_on_fn, turn_on_fn, turn_off_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., is_on_fn, turn_on_fn, turn_off_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/roborock/__init__.pyi

@@ -13,6 +13,7 @@ from typing import Any
 
 SCAN_INTERVAL: Incomplete
 _LOGGER: Incomplete
+RoborockConfigEntry = ConfigEntry[RoborockCoordinators]
 
 @dataclass
 class RoborockCoordinators:
@@ -21,10 +22,10 @@ class RoborockCoordinators:
     def values(self) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]: ...
     def __init__(self, v1, a01) -> None: ...
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+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: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+async def async_unload_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> bool: ...
+async def update_listener(hass: HomeAssistant, entry: RoborockConfigEntry) -> None: ...

homeassistant-stubs/components/roborock/binary_sensor.pyi

@@ -1,26 +1,23 @@
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.util import slugify as slugify
 from roborock.roborock_typing import DeviceProp as DeviceProp
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockBinarySensorDescription(BinarySensorEntityDescription):
     value_fn: Callable[[DeviceProp], bool | int | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., value_fn) -> None: ...
 
 BINARY_SENSOR_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: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockBinarySensorEntity(RoborockCoordinatedEntityV1, BinarySensorEntity):
     entity_description: RoborockBinarySensorDescription

homeassistant-stubs/components/roborock/button.pyi

@@ -1,26 +1,23 @@
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device 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.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.util import slugify as slugify
 from roborock.roborock_typing import RoborockCommand
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockButtonDescription(ButtonEntityDescription):
     command: RoborockCommand
     param: list | dict | None
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, command, param) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., command, param) -> None: ...
 
 CONSUMABLE_BUTTON_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: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockButtonEntity(RoborockEntityV1, ButtonEntity):
     entity_description: RoborockButtonDescription

homeassistant-stubs/components/roborock/coordinator.pyi

@@ -1,11 +1,13 @@
 from .const import DOMAIN as DOMAIN
 from .models import RoborockA01HassDeviceInfo as RoborockA01HassDeviceInfo, RoborockHassDeviceInfo as RoborockHassDeviceInfo, RoborockMapInfo as RoborockMapInfo
 from _typeshed import Incomplete
+from functools import cached_property as cached_property
 from homeassistant.const import ATTR_CONNECTIONS as ATTR_CONNECTIONS
 from homeassistant.core import HomeAssistant as HomeAssistant
 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 roborock import HomeDataRoom as HomeDataRoom
 from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, NetworkInfo as NetworkInfo
 from roborock.roborock_message import RoborockDyadDataProtocol, RoborockZeoProtocol
@@ -32,8 +34,10 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
     def _set_current_map(self) -> None: ...
     async def get_maps(self) -> None: ...
     async def get_rooms(self) -> None: ...
-    @property
+    @cached_property
     def duid(self) -> str: ...
+    @cached_property
+    def duid_slug(self) -> str: ...
 
 class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]]):
     api: Incomplete
@@ -43,5 +47,7 @@ class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[RoborockDyadDa
     def __init__(self, hass: HomeAssistant, device: HomeDataDevice, product_info: HomeDataProduct, api: RoborockClientA01) -> None: ...
     async def _async_update_data(self) -> dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]: ...
     async def release(self) -> None: ...
-    @property
+    @cached_property
     def duid(self) -> str: ...
+    @cached_property
+    def duid_slug(self) -> str: ...

homeassistant-stubs/components/roborock/diagnostics.pyi

@@ -1,8 +1,6 @@
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics.util import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_UNIQUE_ID as CONF_UNIQUE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
@@ -10,4 +8,4 @@ from typing import Any
 TO_REDACT_CONFIG: Incomplete
 TO_REDACT_COORD: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: RoborockConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/roborock/image.pyi

@@ -1,11 +1,10 @@
-from . import RoborockCoordinators as RoborockCoordinators
+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_SLEEP as MAP_SLEEP
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from datetime import datetime
 from homeassistant.components.image import ImageEntity as ImageEntity
-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
@@ -13,7 +12,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 from homeassistant.util import slugify as slugify
 from vacuum_map_parser_base.config.drawable import Drawable as Drawable
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
     _attr_has_entity_name: bool

homeassistant-stubs/components/roborock/number.pyi

@@ -1,16 +1,13 @@
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockEntityV1 as RoborockEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
 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.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.util import slugify as slugify
 from roborock.command_cache import CacheableAttribute
 from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
 from typing import Any
@@ -21,11 +18,11 @@ _LOGGER: Incomplete
 class RoborockNumberDescription(NumberEntityDescription):
     cache_key: CacheableAttribute
     update_value: Callable[[AttributeCache, float], Coroutine[Any, Any, None]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, cache_key, update_value) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., max_value=..., min_value=..., mode=..., native_max_value=..., native_min_value=..., native_step=..., native_unit_of_measurement=..., step=..., cache_key, update_value) -> None: ...
 
 NUMBER_DESCRIPTIONS: list[RoborockNumberDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockNumberEntity(RoborockEntityV1, NumberEntity):
     entity_description: RoborockNumberDescription

homeassistant-stubs/components/roborock/select.pyi

@@ -1,16 +1,13 @@
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
-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.util import slugify as slugify
 from roborock.containers import Status as Status
 from roborock.roborock_message import RoborockDataProtocol
 from roborock.roborock_typing import RoborockCommand
@@ -22,11 +19,11 @@ class RoborockSelectDescription(SelectEntityDescription):
     options_lambda: Callable[[Status], list[str] | None]
     parameter_lambda: Callable[[str, Status], list[int]]
     protocol_listener: RoborockDataProtocol | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, options, api_command, value_fn, options_lambda, parameter_lambda, protocol_listener) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., options=..., api_command, value_fn, options_lambda, parameter_lambda, protocol_listener=...) -> None: ...
 
 SELECT_DESCRIPTIONS: list[RoborockSelectDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
     entity_description: RoborockSelectDescription
@@ -35,3 +32,12 @@ class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
     async def async_select_option(self, option: str) -> None: ...
     @property
     def current_option(self) -> str | None: ...
+
+class RoborockCurrentMapSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
+    _attr_entity_category: Incomplete
+    _attr_translation_key: str
+    async def async_select_option(self, option: str) -> None: ...
+    @property
+    def options(self) -> list[str]: ...
+    @property
+    def current_option(self) -> str | None: ...

homeassistant-stubs/components/roborock/sensor.pyi

@@ -1,38 +1,35 @@
 import datetime
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
 from .device import RoborockCoordinatedEntityA01 as RoborockCoordinatedEntityA01, RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import AREA_SQUARE_METERS as AREA_SQUARE_METERS, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, 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 homeassistant.util import slugify as slugify
-from roborock.roborock_message import RoborockDataProtocol, RoborockDyadDataProtocol
+from roborock.roborock_message import RoborockDataProtocol, RoborockDyadDataProtocol, RoborockZeoProtocol
 from roborock.roborock_typing import DeviceProp as DeviceProp
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSensorDescription(SensorEntityDescription):
     value_fn: Callable[[DeviceProp], StateType | datetime.datetime]
     protocol_listener: RoborockDataProtocol | None = ...
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn, protocol_listener) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn, protocol_listener=...) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSensorDescriptionA01(SensorEntityDescription):
-    data_protocol: RoborockDyadDataProtocol
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, data_protocol) -> None: ...
+    data_protocol: RoborockDyadDataProtocol | RoborockZeoProtocol
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., data_protocol) -> None: ...
 
 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: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
     entity_description: RoborockSensorDescription

homeassistant-stubs/components/roborock/switch.pyi

@@ -1,16 +1,13 @@
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockEntityV1 as RoborockEntityV1
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
 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.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.util import slugify as slugify
 from roborock.command_cache import CacheableAttribute
 from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
 from typing import Any
@@ -22,11 +19,11 @@ class RoborockSwitchDescription(SwitchEntityDescription):
     cache_key: CacheableAttribute
     update_value: Callable[[AttributeCache, bool], Coroutine[Any, Any, None]]
     attribute: str
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, cache_key, update_value, attribute) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., cache_key, update_value, attribute) -> None: ...
 
 SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockSwitch(RoborockEntityV1, SwitchEntity):
     entity_description: RoborockSwitchDescription

homeassistant-stubs/components/roborock/time.pyi

@@ -1,6 +1,5 @@
 import datetime
-from . import RoborockCoordinators as RoborockCoordinators
-from .const import DOMAIN as DOMAIN
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockEntityV1 as RoborockEntityV1
 from _typeshed import Incomplete
@@ -8,11 +7,9 @@ from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from datetime import time
 from homeassistant.components.time import TimeEntity as TimeEntity, TimeEntityDescription as TimeEntityDescription
-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.util import slugify as slugify
 from roborock.command_cache import CacheableAttribute
 from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
 from typing import Any
@@ -24,11 +21,11 @@ class RoborockTimeDescription(TimeEntityDescription):
     cache_key: CacheableAttribute
     update_value: Callable[[AttributeCache, datetime.time], Coroutine[Any, Any, None]]
     get_value: Callable[[AttributeCache], datetime.time]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, cache_key, update_value, get_value) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., cache_key, update_value, get_value) -> None: ...
 
 TIME_DESCRIPTIONS: list[RoborockTimeDescription]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockTimeEntity(RoborockEntityV1, TimeEntity):
     entity_description: RoborockTimeDescription

homeassistant-stubs/components/roborock/vacuum.pyi

@@ -1,19 +1,17 @@
-from . import RoborockCoordinators as RoborockCoordinators
+from . import RoborockConfigEntry as RoborockConfigEntry
 from .const import DOMAIN as DOMAIN, GET_MAPS_SERVICE_NAME as GET_MAPS_SERVICE_NAME
 from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .device import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
 from homeassistant.components.vacuum import STATE_CLEANING as STATE_CLEANING, STATE_DOCKED as STATE_DOCKED, STATE_ERROR as STATE_ERROR, STATE_IDLE as STATE_IDLE, STATE_PAUSED as STATE_PAUSED, STATE_RETURNING as STATE_RETURNING, StateVacuumEntity as StateVacuumEntity, VacuumEntityFeature as VacuumEntityFeature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse
 from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.util import slugify as slugify
 from typing import Any
 
 STATE_CODE_TO_STATE: 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: RoborockConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RoborockVacuum(RoborockCoordinatedEntityV1, StateVacuumEntity):
     _attr_icon: str

homeassistant-stubs/components/roku/binary_sensor.pyi

@@ -12,7 +12,7 @@ from rokuecp.models import Device as RokuDevice
 @dataclass(frozen=True, kw_only=True)
 class RokuBinarySensorEntityDescription(BinarySensorEntityDescription):
     value_fn: Callable[[RokuDevice], bool | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., value_fn) -> None: ...
 
 BINARY_SENSORS: tuple[RokuBinarySensorEntityDescription, ...]
 

homeassistant-stubs/components/roku/browse_media.pyi

@@ -1,6 +1,7 @@
 from .coordinator import RokuDataUpdateCoordinator as RokuDataUpdateCoordinator
 from .helpers import format_channel_name as format_channel_name
 from _typeshed import Incomplete
+from collections.abc import Callable
 from homeassistant.components import media_source as media_source
 from homeassistant.components.media_player import BrowseError as BrowseError, BrowseMedia as BrowseMedia, MediaClass as MediaClass, MediaType as MediaType
 from homeassistant.core import HomeAssistant as HomeAssistant
@@ -10,7 +11,7 @@ CONTENT_TYPE_MEDIA_CLASS: Incomplete
 CONTAINER_TYPES_SPECIFIC_MEDIA_CLASS: Incomplete
 PLAYABLE_MEDIA_TYPES: Incomplete
 EXPANDABLE_MEDIA_TYPES: Incomplete
-GetBrowseImageUrlType: Incomplete
+GetBrowseImageUrlType = Callable[[str, str, str | None], str | None]
 
 def get_thumbnail_url_full(coordinator: RokuDataUpdateCoordinator, is_internal: bool, get_browse_image_url: GetBrowseImageUrlType, media_content_type: str, media_content_id: str, media_image_id: str | None = None) -> str | None: ...
 async def async_browse_media(hass: HomeAssistant, coordinator: RokuDataUpdateCoordinator, get_browse_image_url: GetBrowseImageUrlType, media_content_id: str | None, media_content_type: str | None) -> BrowseMedia: ...

homeassistant-stubs/components/roku/select.pyi

@@ -24,7 +24,7 @@ class RokuSelectEntityDescription(SelectEntityDescription):
     options_fn: Callable[[RokuDevice], list[str]]
     value_fn: Callable[[RokuDevice], str | None]
     set_fn: Callable[[RokuDevice, Roku, str], Awaitable[None]]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, options, options_fn, value_fn, set_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., options=..., options_fn, value_fn, set_fn) -> None: ...
 
 ENTITIES: tuple[RokuSelectEntityDescription, ...]
 CHANNEL_ENTITY: Incomplete

homeassistant-stubs/components/roku/sensor.pyi

@@ -13,7 +13,7 @@ from rokuecp.models import Device as RokuDevice
 @dataclass(frozen=True, kw_only=True)
 class RokuSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[RokuDevice], str | None]
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+    def __init__(self, *, key, device_class=..., entity_category=..., entity_registry_enabled_default=..., entity_registry_visible_default=..., force_update=..., icon=..., has_entity_name=..., name=..., translation_key=..., translation_placeholders=..., unit_of_measurement=..., last_reset=..., native_unit_of_measurement=..., options=..., state_class=..., suggested_display_precision=..., suggested_unit_of_measurement=..., value_fn) -> None: ...
 
 SENSORS: tuple[RokuSensorEntityDescription, ...]
 

homeassistant-stubs/components/samsungtv/bridge.pyi

@@ -71,7 +71,7 @@ class SamsungTVLegacyBridge(SamsungTVBridge):
     def _is_on(self) -> bool: ...
     async def async_try_connect(self) -> str: ...
     def _try_connect(self) -> str: ...
-    async def async_device_info(self) -> None: ...
+    async def async_device_info(self) -> dict[str, Any] | None: ...
     def _notify_reauth_callback(self) -> None: ...
     auth_failed: bool
     def _get_remote(self) -> Remote: ...