homeassistant-stubs 2025.9.0b0__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/components/roborock/config_flow.pyi

@@ -1,5 +1,5 @@
 from . import RoborockConfigEntry as RoborockConfigEntry
-from .const import CONF_BASE_URL as CONF_BASE_URL, CONF_ENTRY_CODE as CONF_ENTRY_CODE, CONF_USER_DATA as CONF_USER_DATA, DEFAULT_DRAWABLES as DEFAULT_DRAWABLES, DOMAIN as DOMAIN, DRAWABLES as DRAWABLES
+from .const import CONF_BASE_URL as CONF_BASE_URL, CONF_ENTRY_CODE as CONF_ENTRY_CODE, CONF_SHOW_BACKGROUND as CONF_SHOW_BACKGROUND, CONF_USER_DATA as CONF_USER_DATA, DEFAULT_DRAWABLES as DEFAULT_DRAWABLES, DOMAIN as DOMAIN, DRAWABLES as DRAWABLES
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlowWithReload as OptionsFlowWithReload, SOURCE_REAUTH as SOURCE_REAUTH
@@ -7,7 +7,7 @@ from homeassistant.const import CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import callback as callback
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo as DhcpServiceInfo
-from roborock.containers import UserData as UserData
+from roborock.data import UserData as UserData
 from roborock.web_api import RoborockApiClient
 from typing import Any
 
@@ -25,7 +25,7 @@ class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
     async def async_step_dhcp(self, discovery_info: DhcpServiceInfo) -> ConfigFlowResult: ...
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
     async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    def _create_entry(self, client: RoborockApiClient, username: str, user_data: UserData) -> ConfigFlowResult: ...
+    async def _create_entry(self, client: RoborockApiClient, username: str, user_data: UserData) -> ConfigFlowResult: ...
     @staticmethod
     @callback
     def async_get_options_flow(config_entry: RoborockConfigEntry) -> RoborockOptionsFlowHandler: ...

homeassistant-stubs/components/roborock/const.pyi

@@ -5,6 +5,7 @@ DOMAIN: str
 CONF_ENTRY_CODE: str
 CONF_BASE_URL: str
 CONF_USER_DATA: str
+CONF_SHOW_BACKGROUND: str
 DRAWABLES: str
 DEFAULT_DRAWABLES: Incomplete
 PLATFORMS: Incomplete

homeassistant-stubs/components/roborock/coordinator.pyi

@@ -1,83 +1,110 @@
-from .const import A01_UPDATE_INTERVAL as A01_UPDATE_INTERVAL, 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_SCALE as MAP_SCALE, MAP_SLEEP as MAP_SLEEP, V1_CLOUD_IN_CLEANING_INTERVAL as V1_CLOUD_IN_CLEANING_INTERVAL, V1_CLOUD_NOT_CLEANING_INTERVAL as V1_CLOUD_NOT_CLEANING_INTERVAL, V1_LOCAL_IN_CLEANING_INTERVAL as V1_LOCAL_IN_CLEANING_INTERVAL, V1_LOCAL_NOT_CLEANING_INTERVAL as V1_LOCAL_NOT_CLEANING_INTERVAL
-from .models import RoborockA01HassDeviceInfo as RoborockA01HassDeviceInfo, RoborockHassDeviceInfo as RoborockHassDeviceInfo, RoborockMapInfo as RoborockMapInfo
-from .roborock_storage import RoborockMapStorage as RoborockMapStorage
+from .const import A01_UPDATE_INTERVAL as A01_UPDATE_INTERVAL, DOMAIN as DOMAIN, IMAGE_CACHE_INTERVAL as IMAGE_CACHE_INTERVAL, V1_CLOUD_IN_CLEANING_INTERVAL as V1_CLOUD_IN_CLEANING_INTERVAL, V1_CLOUD_NOT_CLEANING_INTERVAL as V1_CLOUD_NOT_CLEANING_INTERVAL, V1_LOCAL_IN_CLEANING_INTERVAL as V1_LOCAL_IN_CLEANING_INTERVAL, V1_LOCAL_NOT_CLEANING_INTERVAL as V1_LOCAL_NOT_CLEANING_INTERVAL
+from .models import DeviceState as DeviceState
 from _typeshed import Incomplete
 from dataclasses import dataclass
+from datetime import datetime
 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.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue, async_delete_issue as async_delete_issue
 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, 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
-from vacuum_map_parser_base.map_data import MapData as MapData
+from roborock import B01Props
+from roborock.data import HomeDataScene as HomeDataScene
+from roborock.devices.device import RoborockDevice as RoborockDevice
+from roborock.devices.traits.a01 import DyadApi as DyadApi, ZeoApi as ZeoApi
+from roborock.devices.traits.b01 import Q7PropertiesApi as Q7PropertiesApi
+from roborock.devices.traits.v1 import PropertiesApi as PropertiesApi
+from roborock.roborock_message import RoborockB01Props, RoborockDyadDataProtocol, RoborockZeoProtocol
+from typing import Any, TypeVar
 
 SCAN_INTERVAL: Incomplete
+MIN_UNAVAILABLE_DURATION: Incomplete
 _LOGGER: Incomplete
 
 @dataclass
 class RoborockCoordinators:
     v1: list[RoborockDataUpdateCoordinator]
     a01: list[RoborockDataUpdateCoordinatorA01]
-    def values(self) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]: ...
+    b01: list[RoborockDataUpdateCoordinatorB01]
+    def values(self) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | RoborockDataUpdateCoordinatorB01]: ...
 type RoborockConfigEntry = ConfigEntry[RoborockCoordinators]
 
-class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
+class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceState]):
     config_entry: RoborockConfigEntry
-    roborock_device_info: Incomplete
-    api: RoborockLocalClientV1 | RoborockMqttClientV1
-    cloud_api: Incomplete
+    _device: Incomplete
+    properties_api: Incomplete
     device_info: Incomplete
-    current_map: int | None
-    maps: dict[int, RoborockMapInfo]
-    _home_data_rooms: Incomplete
-    map_storage: Incomplete
-    _user_data: Incomplete
-    _api_client: Incomplete
-    _is_cloud_api: bool
-    map_parser: Incomplete
     last_update_state: str | 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], api_client: RoborockApiClient, user_data: UserData) -> None: ...
+    _last_home_update_attempt: datetime
+    last_home_update: datetime | None
+    _last_update_success_time: datetime | None
+    _has_connected_locally: bool
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: RoborockDevice, properties_api: PropertiesApi) -> None: ...
     @cached_property
     def dock_device_info(self) -> DeviceInfo: ...
-    def parse_map_data_v1(self, map_bytes: bytes) -> tuple[bytes | None, MapData | None]: ...
     async def _async_setup(self) -> None: ...
     async def update_map(self) -> None: ...
     update_interval: Incomplete
     async def _verify_api(self) -> None: ...
-    async def async_shutdown(self) -> None: ...
     async def _update_device_prop(self) -> None: ...
-    async def _async_update_data(self) -> DeviceProp: ...
-    def _set_current_map(self) -> None: ...
-    async def set_current_map_rooms(self) -> None: ...
+    async def _async_update_data(self) -> DeviceState: ...
+    def _should_suppress_update_failure(self) -> bool: ...
     async def get_routines(self) -> list[HomeDataScene]: ...
     async def execute_routines(self, routine_id: int) -> None: ...
     @cached_property
     def duid(self) -> str: ...
     @cached_property
     def duid_slug(self) -> str: ...
-    async def refresh_coordinator_map(self) -> None: ...
+    @property
+    def device(self) -> RoborockDevice: ...
 
-class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]]):
+async def _refresh_traits(traits: list[Any]) -> None: ...
+_V = TypeVar('_V', bound=RoborockDyadDataProtocol | RoborockZeoProtocol)
+
+class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[_V, StateType]]):
     config_entry: RoborockConfigEntry
+    _device: Incomplete
+    device_info: Incomplete
+    request_protocols: list[_V]
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: RoborockDevice) -> None: ...
+    @cached_property
+    def duid(self) -> str: ...
+    @cached_property
+    def duid_slug(self) -> str: ...
+    @property
+    def device(self) -> RoborockDevice: ...
+
+class RoborockWashingMachineUpdateCoordinator(RoborockDataUpdateCoordinatorA01[RoborockZeoProtocol]):
     api: Incomplete
+    request_protocols: list[RoborockZeoProtocol]
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: RoborockDevice, api: ZeoApi) -> None: ...
+    async def _async_update_data(self) -> dict[RoborockZeoProtocol, StateType]: ...
+
+class RoborockWetDryVacUpdateCoordinator(RoborockDataUpdateCoordinatorA01[RoborockDyadDataProtocol]):
+    api: Incomplete
+    request_protocols: list[RoborockDyadDataProtocol]
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: RoborockDevice, api: DyadApi) -> None: ...
+    async def _async_update_data(self) -> dict[RoborockDyadDataProtocol, StateType]: ...
+
+class RoborockDataUpdateCoordinatorB01(DataUpdateCoordinator[B01Props]):
+    config_entry: RoborockConfigEntry
+    _device: Incomplete
     device_info: Incomplete
-    request_protocols: list[RoborockDyadDataProtocol | RoborockZeoProtocol]
-    roborock_device_info: Incomplete
-    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: ...
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: RoborockDevice) -> None: ...
     @cached_property
     def duid(self) -> str: ...
     @cached_property
     def duid_slug(self) -> str: ...
+    @property
+    def device(self) -> RoborockDevice: ...
+
+class RoborockB01Q7UpdateCoordinator(RoborockDataUpdateCoordinatorB01):
+    api: Incomplete
+    request_protocols: list[RoborockB01Props]
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: RoborockDevice, api: Q7PropertiesApi) -> None: ...
+    async def _async_update_data(self) -> B01Props: ...

homeassistant-stubs/components/roborock/diagnostics.pyi

@@ -5,7 +5,7 @@ from homeassistant.const import CONF_UNIQUE_ID as CONF_UNIQUE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
 
+_LOGGER: Incomplete
 TO_REDACT_CONFIG: Incomplete
-TO_REDACT_COORD: Incomplete
 
 async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: RoborockConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/roborock/entity.pyi

@@ -1,55 +1,38 @@
 from .const import DOMAIN as DOMAIN
-from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
+from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01, RoborockDataUpdateCoordinatorB01 as RoborockDataUpdateCoordinatorB01
 from _typeshed import Incomplete
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity import Entity as Entity
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from roborock.api import RoborockClient as RoborockClient
-from roborock.command_cache import CacheableAttribute as CacheableAttribute
-from roborock.containers import Consumable as Consumable, Status
-from roborock.roborock_message import RoborockDataProtocol
+from roborock.data import Status as Status
+from roborock.devices.traits.v1.command import CommandTrait as CommandTrait
 from roborock.roborock_typing import RoborockCommand
-from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache, RoborockClientV1 as RoborockClientV1
-from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1 as RoborockMqttClientV1
-from roborock.version_a01_apis import RoborockClientA01 as RoborockClientA01
 from typing import Any
 
 class RoborockEntity(Entity):
     _attr_has_entity_name: bool
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
-    _api: Incomplete
-    def __init__(self, unique_id: str, device_info: DeviceInfo, api: RoborockClient) -> None: ...
+    def __init__(self, unique_id: str, device_info: DeviceInfo) -> None: ...
 
 class RoborockEntityV1(RoborockEntity):
-    _api: RoborockClientV1
-    def __init__(self, unique_id: str, device_info: DeviceInfo, api: RoborockClientV1) -> None: ...
-    def get_cache(self, attribute: CacheableAttribute) -> AttributeCache: ...
-    @classmethod
-    async def _send_command(cls, command: RoborockCommand | str, api: RoborockClientV1, params: dict[str, Any] | list[Any] | int | None = None) -> dict: ...
+    _api: Incomplete
+    def __init__(self, unique_id: str, device_info: DeviceInfo, api: CommandTrait) -> None: ...
     async def send(self, command: RoborockCommand | str, params: dict[str, Any] | list[Any] | int | None = None) -> dict: ...
-    @property
-    def api(self) -> RoborockClientV1: ...
-
-class RoborockEntityA01(RoborockEntity):
-    _api: RoborockClientA01
-    def __init__(self, unique_id: str, device_info: DeviceInfo, api: RoborockClientA01) -> None: ...
 
 class RoborockCoordinatedEntityV1(RoborockEntityV1, CoordinatorEntity[RoborockDataUpdateCoordinator]):
     _attr_has_entity_name: bool
     _attr_unique_id: Incomplete
-    listener_requests: Incomplete
-    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, listener_request: list[RoborockDataProtocol] | RoborockDataProtocol | None = None, is_dock_entity: bool = False) -> None: ...
-    async def async_added_to_hass(self) -> None: ...
-    async def async_will_remove_from_hass(self) -> None: ...
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, is_dock_entity: bool = False) -> None: ...
     @property
     def _device_status(self) -> Status: ...
-    @property
-    def cloud_api(self) -> RoborockMqttClientV1: ...
     async def send(self, command: RoborockCommand | str, params: dict[str, Any] | list[Any] | int | None = None) -> dict: ...
-    def _update_from_listener(self, value: Status | Consumable) -> None: ...
 
-class RoborockCoordinatedEntityA01(RoborockEntityA01, CoordinatorEntity[RoborockDataUpdateCoordinatorA01]):
+class RoborockCoordinatedEntityA01(RoborockEntity, CoordinatorEntity[RoborockDataUpdateCoordinatorA01]):
     _attr_unique_id: Incomplete
     def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinatorA01) -> None: ...
+
+class RoborockCoordinatedEntityB01(RoborockEntity, CoordinatorEntity[RoborockDataUpdateCoordinatorB01]):
+    _attr_unique_id: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinatorB01) -> None: ...

homeassistant-stubs/components/roborock/image.pyi

@@ -6,7 +6,10 @@ 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
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from roborock.devices.traits.v1.home import HomeTrait as HomeTrait
+from roborock.devices.traits.v1.map_content import MapContent as MapContent
 
 _LOGGER: Incomplete
 PARALLEL_UPDATES: int
@@ -18,13 +21,14 @@ class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
     image_last_updated: datetime
     _attr_name: str
     config_entry: Incomplete
+    _home_trait: Incomplete
     map_flag: Incomplete
-    cached_map: bytes
+    cached_map: bytes | None
     _attr_entity_category: Incomplete
-    def __init__(self, config_entry: ConfigEntry, unique_id: str, coordinator: RoborockDataUpdateCoordinator, map_flag: int, map_name: str) -> None: ...
-    @property
-    def is_selected(self) -> bool: ...
+    def __init__(self, config_entry: ConfigEntry, coordinator: RoborockDataUpdateCoordinator, home_trait: HomeTrait, map_flag: int, map_name: str) -> None: ...
     _attr_image_last_updated: Incomplete
     async def async_added_to_hass(self) -> None: ...
+    @property
+    def _map_content(self) -> MapContent | None: ...
     def _handle_coordinator_update(self) -> None: ...
     async def async_image(self) -> bytes | None: ...

homeassistant-stubs/components/roborock/models.pyi

@@ -1,16 +1,24 @@
+from _typeshed import Incomplete
 from dataclasses import dataclass
 from datetime import datetime
-from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, NetworkInfo as NetworkInfo
-from roborock.roborock_typing import DeviceProp as DeviceProp
+from roborock.data import CleanSummaryWithDetail as CleanSummaryWithDetail, Consumable as Consumable, DnDTimer as DnDTimer, HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, NetworkInfo as NetworkInfo, Status as Status
 from typing import Any
 from vacuum_map_parser_base.map_data import MapData as MapData
 
+_LOGGER: Incomplete
+
+@dataclass
+class DeviceState:
+    status: Status
+    dnd_timer: DnDTimer
+    consumable: Consumable
+    clean_summary: CleanSummaryWithDetail
+
 @dataclass
 class RoborockHassDeviceInfo:
     device: HomeDataDevice
     network_info: NetworkInfo
     product: HomeDataProduct
-    props: DeviceProp
     def as_dict(self) -> dict[str, dict[str, Any]]: ...
 
 @dataclass
@@ -23,9 +31,6 @@ class RoborockA01HassDeviceInfo:
 class RoborockMapInfo:
     flag: int
     name: str
-    rooms: dict[int, str]
     image: bytes | None
     last_updated: datetime
     map_data: MapData | None
-    @property
-    def current_room(self) -> str | None: ...

homeassistant-stubs/components/roborock/number.pyi

@@ -9,8 +9,7 @@ from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 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 roborock.devices.traits.v1 import PropertiesApi as PropertiesApi
 from typing import Any
 
 _LOGGER: Incomplete
@@ -18,8 +17,9 @@ PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockNumberDescription(NumberEntityDescription):
-    cache_key: CacheableAttribute
-    update_value: Callable[[AttributeCache, float], Coroutine[Any, Any, None]]
+    trait: Callable[[PropertiesApi], Any | None]
+    get_value: Callable[[Any], float]
+    set_value: Callable[[Any, float], Coroutine[Any, Any, None]]
 
 NUMBER_DESCRIPTIONS: list[RoborockNumberDescription]
 
@@ -27,7 +27,8 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEnt
 
 class RoborockNumberEntity(RoborockEntityV1, NumberEntity):
     entity_description: RoborockNumberDescription
-    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockNumberDescription) -> None: ...
+    _trait: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockNumberDescription, trait: Any) -> None: ...
     @property
     def native_value(self) -> float | None: ...
     async def async_set_native_value(self, value: float) -> None: ...

homeassistant-stubs/components/roborock/roborock_storage.pyi

@@ -1,23 +1,28 @@
-from .const import DOMAIN as DOMAIN, MAP_FILENAME_SUFFIX as MAP_FILENAME_SUFFIX
+from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.storage import Store as Store
 from pathlib import Path
+from roborock.devices.cache import Cache, CacheData
+from typing import Any
 
 _LOGGER: Incomplete
 STORAGE_PATH: Incomplete
 MAPS_PATH: str
+CACHE_VERSION: int
 
 def _storage_path_prefix(hass: HomeAssistant, entry_id: str) -> Path: ...
+async def async_cleanup_map_storage(hass: HomeAssistant, entry_id: str) -> None: ...
 
-class RoborockMapStorage:
-    _hass: Incomplete
-    _path_prefix: Incomplete
-    _write_queue: dict[int, bytes]
-    def __init__(self, hass: HomeAssistant, entry_id: str, device_id_slug: str) -> None: ...
-    async def async_load_map(self, map_flag: int) -> bytes | None: ...
-    def _load_map(self, filename: Path) -> bytes | None: ...
-    async def async_save_map(self, map_flag: int, content: bytes) -> None: ...
-    async def flush(self) -> None: ...
-    def _save_map(self, filename: Path, content: bytes) -> None: ...
+class StoreImpl(Store[dict[str, Any]]):
+    def __init__(self, hass: HomeAssistant, entry_id: str) -> None: ...
+    async def _async_migrate_func(self, old_major_version: int, old_minor_version: int, old_data: dict[str, Any]) -> dict[str, Any]: ...
 
-async def async_remove_map_storage(hass: HomeAssistant, entry_id: str) -> None: ...
+class CacheStore(Cache):
+    _cache_store: Incomplete
+    _cache_data: CacheData | None
+    def __init__(self, hass: HomeAssistant, entry_id: str) -> None: ...
+    async def get(self) -> CacheData: ...
+    async def set(self, value: CacheData) -> None: ...
+    async def flush(self) -> None: ...
+    async def async_remove(self) -> None: ...

homeassistant-stubs/components/roborock/select.pyi

@@ -1,4 +1,4 @@
-from .const import MAP_SLEEP as MAP_SLEEP
+from .const import DOMAIN as DOMAIN, MAP_SLEEP as MAP_SLEEP
 from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
@@ -7,19 +7,21 @@ 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.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from roborock.roborock_message import RoborockDataProtocol
-from roborock.roborock_typing import DeviceProp as DeviceProp, RoborockCommand
+from roborock.devices.traits.v1 import PropertiesApi as PropertiesApi
+from roborock.devices.traits.v1.home import HomeTrait as HomeTrait
+from roborock.devices.traits.v1.maps import MapsTrait as MapsTrait
+from roborock.roborock_typing import RoborockCommand
 
 PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSelectDescription(SelectEntityDescription):
     api_command: RoborockCommand
-    value_fn: Callable[[DeviceProp], str | None]
-    options_lambda: Callable[[DeviceProp], list[str] | None]
-    parameter_lambda: Callable[[str, DeviceProp], list[int]]
-    protocol_listener: RoborockDataProtocol | None = ...
+    value_fn: Callable[[PropertiesApi], str | None]
+    options_lambda: Callable[[PropertiesApi], list[str] | None]
+    parameter_lambda: Callable[[str, PropertiesApi], list[int]]
     is_dock_entity: bool = ...
 
 SELECT_DESCRIPTIONS: list[RoborockSelectDescription]
@@ -37,6 +39,11 @@ class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
 class RoborockCurrentMapSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
     _attr_entity_category: Incomplete
     _attr_translation_key: str
+    _home_trait: Incomplete
+    _maps_trait: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, home_trait: HomeTrait, maps_trait: MapsTrait) -> None: ...
+    @property
+    def _available_map_names(self) -> dict[int, str]: ...
     async def async_select_option(self, option: str) -> None: ...
     @property
     def options(self) -> list[str]: ...

homeassistant-stubs/components/roborock/sensor.pyi

@@ -1,6 +1,7 @@
 import datetime
-from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
-from .entity import RoborockCoordinatedEntityA01 as RoborockCoordinatedEntityA01, RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1, RoborockEntity as RoborockEntity
+from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01, RoborockDataUpdateCoordinatorB01 as RoborockDataUpdateCoordinatorB01
+from .entity import RoborockCoordinatedEntityA01 as RoborockCoordinatedEntityA01, RoborockCoordinatedEntityB01 as RoborockCoordinatedEntityB01, RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1, RoborockEntity as RoborockEntity
+from .models import DeviceState as DeviceState
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
@@ -9,25 +10,30 @@ from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as
 from homeassistant.core import HomeAssistant as HomeAssistant
 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
+from roborock.data import B01Props as B01Props
+from roborock.roborock_message import RoborockDyadDataProtocol, RoborockZeoProtocol
 
+_LOGGER: Incomplete
 PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSensorDescription(SensorEntityDescription):
-    value_fn: Callable[[DeviceProp], StateType | datetime.datetime]
-    protocol_listener: RoborockDataProtocol | None = ...
+    value_fn: Callable[[DeviceState], StateType | datetime.datetime]
     is_dock_entity: bool = ...
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSensorDescriptionA01(SensorEntityDescription):
     data_protocol: RoborockDyadDataProtocol | RoborockZeoProtocol
 
-def _dock_error_value_fn(properties: DeviceProp) -> str | None: ...
+@dataclass(frozen=True, kw_only=True)
+class RoborockSensorDescriptionB01(SensorEntityDescription):
+    value_fn: Callable[[B01Props], StateType]
+
+def _dock_error_value_fn(state: DeviceState) -> str | None: ...
 
 SENSOR_DESCRIPTIONS: Incomplete
 A01_SENSOR_DESCRIPTIONS: list[RoborockSensorDescriptionA01]
+Q7_B01_SENSOR_DESCRIPTIONS: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
@@ -41,6 +47,8 @@ class RoborockCurrentRoom(RoborockCoordinatedEntityV1, SensorEntity):
     _attr_device_class: Incomplete
     _attr_translation_key: str
     _attr_entity_category: Incomplete
+    _home_trait: Incomplete
+    _map_content_trait: Incomplete
     def __init__(self, coordinator: RoborockDataUpdateCoordinator) -> None: ...
     @property
     def options(self) -> list[str]: ...
@@ -52,3 +60,9 @@ class RoborockSensorEntityA01(RoborockCoordinatedEntityA01, SensorEntity):
     def __init__(self, coordinator: RoborockDataUpdateCoordinatorA01, description: RoborockSensorDescriptionA01) -> None: ...
     @property
     def native_value(self) -> StateType: ...
+
+class RoborockSensorEntityB01(RoborockCoordinatedEntityB01, SensorEntity):
+    entity_description: RoborockSensorDescriptionB01
+    def __init__(self, coordinator: RoborockDataUpdateCoordinatorB01, description: RoborockSensorDescriptionB01) -> None: ...
+    @property
+    def native_value(self) -> StateType: ...

homeassistant-stubs/components/roborock/switch.pyi

@@ -2,15 +2,15 @@ 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
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 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 AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from roborock.command_cache import CacheableAttribute
-from roborock.version_1_apis.roborock_client_v1 import AttributeCache as AttributeCache
+from roborock.devices.traits.v1 import PropertiesApi as PropertiesApi
+from roborock.devices.traits.v1.common import RoborockSwitchBase as RoborockSwitchBase
 from typing import Any
 
 _LOGGER: Incomplete
@@ -18,9 +18,7 @@ PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSwitchDescription(SwitchEntityDescription):
-    cache_key: CacheableAttribute
-    update_value: Callable[[AttributeCache, bool], Coroutine[Any, Any, None]]
-    attribute: str
+    trait: Callable[[PropertiesApi], RoborockSwitchBase | None]
     is_dock_entity: bool = ...
 
 SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription]
@@ -29,7 +27,8 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEnt
 
 class RoborockSwitch(RoborockEntityV1, SwitchEntity):
     entity_description: RoborockSwitchDescription
-    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockSwitchDescription) -> None: ...
+    _trait: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockSwitchDescription, trait: RoborockSwitchBase) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     @property

homeassistant-stubs/components/roborock/time.pyi

@@ -11,8 +11,6 @@ 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 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
 
 _LOGGER: Incomplete
@@ -20,9 +18,9 @@ PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockTimeDescription(TimeEntityDescription):
-    cache_key: CacheableAttribute
-    update_value: Callable[[AttributeCache, datetime.time], Coroutine[Any, Any, None]]
-    get_value: Callable[[AttributeCache], datetime.time]
+    trait: Callable[[Any], Any | None]
+    get_value: Callable[[Any], datetime.time]
+    update_value: Callable[[Any, datetime.time], Coroutine[Any, Any, None]]
 
 TIME_DESCRIPTIONS: list[RoborockTimeDescription]
 
@@ -30,7 +28,8 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEnt
 
 class RoborockTimeEntity(RoborockEntityV1, TimeEntity):
     entity_description: RoborockTimeDescription
-    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockTimeDescription) -> None: ...
+    _trait: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockTimeDescription, trait: Any) -> None: ...
     @property
     def native_value(self) -> time | None: ...
     async def async_set_value(self, value: time) -> None: ...

homeassistant-stubs/components/roborock/vacuum.pyi

@@ -9,6 +9,7 @@ from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
+_LOGGER: Incomplete
 STATE_CODE_TO_STATE: Incomplete
 PARALLEL_UPDATES: int
 
@@ -19,9 +20,10 @@ class RoborockVacuum(RoborockCoordinatedEntityV1, StateVacuumEntity):
     _attr_supported_features: Incomplete
     _attr_translation_key = DOMAIN
     _attr_name: Incomplete
-    _attr_fan_speed_list: Incomplete
     def __init__(self, coordinator: RoborockDataUpdateCoordinator) -> None: ...
     @property
+    def fan_speed_list(self) -> list[str]: ...
+    @property
     def activity(self) -> VacuumActivity | None: ...
     @property
     def fan_speed(self) -> str | None: ...

homeassistant-stubs/components/route_b_smart_meter/__init__.pyi

@@ -0,0 +1,10 @@
+from .coordinator import BRouteConfigEntry as BRouteConfigEntry, BRouteUpdateCoordinator as BRouteUpdateCoordinator
+from _typeshed import Incomplete
+from homeassistant.const import Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant
+
+_LOGGER: Incomplete
+PLATFORMS: list[Platform]
+
+async def async_setup_entry(hass: HomeAssistant, entry: BRouteConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: BRouteConfigEntry) -> bool: ...

homeassistant-stubs/components/route_b_smart_meter/config_flow.pyi

@@ -0,0 +1,22 @@
+from .const import DOMAIN as DOMAIN, ENTRY_TITLE as ENTRY_TITLE
+from _typeshed import Incomplete
+from homeassistant.components.usb import get_serial_by_id as get_serial_by_id, human_readable_device_name as human_readable_device_name
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.const import CONF_DEVICE as CONF_DEVICE, CONF_ID as CONF_ID, CONF_PASSWORD as CONF_PASSWORD
+from homeassistant.core import callback as callback
+from homeassistant.helpers.service_info.usb import UsbServiceInfo as UsbServiceInfo
+from serial.tools.list_ports_common import ListPortInfo as ListPortInfo
+from typing import Any
+
+_LOGGER: Incomplete
+
+def _validate_input(device: str, id: str, password: str) -> None: ...
+def _human_readable_device_name(port: UsbServiceInfo | ListPortInfo) -> str: ...
+
+class BRouteConfigFlow(ConfigFlow, domain=DOMAIN):
+    VERSION: int
+    device: UsbServiceInfo | None
+    @callback
+    def _get_discovered_device_id_and_name(self, device_options: dict[str, ListPortInfo]) -> tuple[str | None, str | None]: ...
+    async def _get_usb_devices(self) -> dict[str, ListPortInfo]: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...