homeassistant-stubs 2025.3.3__py3-none-any.whl → 2025.4.0__py3-none-any.whl

homeassistant-stubs/components/reolink/binary_sensor.pyi

@@ -1,4 +1,4 @@
-from .entity import ReolinkChannelCoordinatorEntity as ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription as ReolinkChannelEntityDescription
+from .entity import ReolinkChannelCoordinatorEntity as ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription as ReolinkChannelEntityDescription, ReolinkEntityDescription as ReolinkEntityDescription
 from .util import ReolinkConfigEntry as ReolinkConfigEntry, ReolinkData as ReolinkData
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
@@ -16,8 +16,15 @@ PARALLEL_UPDATES: int
 class ReolinkBinarySensorEntityDescription(BinarySensorEntityDescription, ReolinkChannelEntityDescription):
     value: Callable[[Host, int], bool]
 
+@dataclass(frozen=True, kw_only=True)
+class ReolinkSmartAIBinarySensorEntityDescription(BinarySensorEntityDescription, ReolinkEntityDescription):
+    smart_type: str
+    value: Callable[[Host, int, int], bool]
+    supported: Callable[[Host, int, int], bool] = ...
+
 BINARY_PUSH_SENSORS: Incomplete
 BINARY_SENSORS: Incomplete
+BINARY_SMART_AI_SENSORS: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ReolinkConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
@@ -31,3 +38,12 @@ class ReolinkBinarySensorEntity(ReolinkChannelCoordinatorEntity, BinarySensorEnt
 class ReolinkPushBinarySensorEntity(ReolinkBinarySensorEntity):
     async def async_added_to_hass(self) -> None: ...
     async def _async_handle_event(self, event: str) -> None: ...
+
+class ReolinkSmartAIBinarySensorEntity(ReolinkChannelCoordinatorEntity, BinarySensorEntity):
+    entity_description: ReolinkSmartAIBinarySensorEntityDescription
+    _attr_unique_id: Incomplete
+    _location: Incomplete
+    _attr_translation_placeholders: Incomplete
+    def __init__(self, reolink_data: ReolinkData, channel: int, location: int, entity_description: ReolinkSmartAIBinarySensorEntityDescription) -> None: ...
+    @property
+    def is_on(self) -> bool: ...

homeassistant-stubs/components/reolink/config_flow.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_SUPPORTS_PRIVACY_MODE as CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS as CONF_USE_HTTPS, DOMAIN as DOMAIN
+from .const import CONF_BC_PORT as CONF_BC_PORT, CONF_SUPPORTS_PRIVACY_MODE as CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS as CONF_USE_HTTPS, DOMAIN as DOMAIN
 from .exceptions import PasswordIncompatible as PasswordIncompatible, ReolinkException as ReolinkException, ReolinkWebhookException as ReolinkWebhookException, UserNotAdmin as UserNotAdmin
 from .host import ReolinkHost as ReolinkHost
 from .util import ReolinkConfigEntry as ReolinkConfigEntry, is_connected as is_connected

homeassistant-stubs/components/reolink/const.pyi

@@ -1,3 +1,4 @@
 DOMAIN: str
 CONF_USE_HTTPS: str
+CONF_BC_PORT: str
 CONF_SUPPORTS_PRIVACY_MODE: str

homeassistant-stubs/components/reolink/host.pyi

@@ -1,7 +1,7 @@
 import asyncio
-from .const import CONF_SUPPORTS_PRIVACY_MODE as CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS as CONF_USE_HTTPS, DOMAIN as DOMAIN
+from .const import CONF_BC_PORT as CONF_BC_PORT, CONF_SUPPORTS_PRIVACY_MODE as CONF_SUPPORTS_PRIVACY_MODE, CONF_USE_HTTPS as CONF_USE_HTTPS, DOMAIN as DOMAIN
 from .exceptions import PasswordIncompatible as PasswordIncompatible, ReolinkSetupException as ReolinkSetupException, ReolinkWebhookException as ReolinkWebhookException, UserNotAdmin as UserNotAdmin
-from .util import get_store as get_store
+from .util import ReolinkConfigEntry as ReolinkConfigEntry, get_store as get_store
 from _typeshed import Incomplete
 from aiohttp.web import Request as Request
 from collections import defaultdict
@@ -33,7 +33,7 @@ _LOGGER: Incomplete
 
 class ReolinkHost:
     _hass: HomeAssistant
-    _config_entry_id: Incomplete
+    _config_entry: Incomplete
     _config: Incomplete
     _unique_id: str
     _api: Incomplete
@@ -61,7 +61,7 @@ class ReolinkHost:
     _lost_subscription_start: bool
     _lost_subscription: bool
     cancel_refresh_privacy_mode: CALLBACK_TYPE | None
-    def __init__(self, hass: HomeAssistant, config: Mapping[str, Any], options: Mapping[str, Any], config_entry_id: str | None = None) -> None: ...
+    def __init__(self, hass: HomeAssistant, config: Mapping[str, Any], options: Mapping[str, Any], config_entry: ReolinkConfigEntry | None = None) -> None: ...
     @callback
     def async_register_update_cmd(self, cmd: str, channel: int | None = None) -> None: ...
     @callback

homeassistant-stubs/components/reolink/number.pyi

@@ -3,7 +3,7 @@ from .util import ReolinkConfigEntry as ReolinkConfigEntry, ReolinkData as Reoli
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
-from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription, NumberMode as NumberMode
+from homeassistant.components.number import NumberDeviceClass as NumberDeviceClass, NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription, NumberMode as NumberMode
 from homeassistant.const import EntityCategory as EntityCategory, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -20,6 +20,13 @@ class ReolinkNumberEntityDescription(NumberEntityDescription, ReolinkChannelEnti
     mode: NumberMode = ...
     value: Callable[[Host, int], float | None]
 
+@dataclass(frozen=True, kw_only=True)
+class ReolinkSmartAINumberEntityDescription(NumberEntityDescription, ReolinkChannelEntityDescription):
+    smart_type: str
+    method: Callable[[Host, int, int, float], Any]
+    mode: NumberMode = ...
+    value: Callable[[Host, int, int], float | None]
+
 @dataclass(frozen=True, kw_only=True)
 class ReolinkHostNumberEntityDescription(NumberEntityDescription, ReolinkHostEntityDescription):
     method: Callable[[Host, float], Any]
@@ -33,6 +40,7 @@ class ReolinkChimeNumberEntityDescription(NumberEntityDescription, ReolinkChimeE
     value: Callable[[Chime], float | None]
 
 NUMBER_ENTITIES: Incomplete
+SMART_AI_NUMBER_ENTITIES: Incomplete
 HOST_NUMBER_ENTITIES: Incomplete
 CHIME_NUMBER_ENTITIES: Incomplete
 
@@ -49,6 +57,18 @@ class ReolinkNumberEntity(ReolinkChannelCoordinatorEntity, NumberEntity):
     @raise_translated_error
     async def async_set_native_value(self, value: float) -> None: ...
 
+class ReolinkSmartAINumberEntity(ReolinkChannelCoordinatorEntity, NumberEntity):
+    entity_description: ReolinkSmartAINumberEntityDescription
+    _attr_unique_id: Incomplete
+    _location: Incomplete
+    _attr_mode: Incomplete
+    _attr_translation_placeholders: Incomplete
+    def __init__(self, reolink_data: ReolinkData, channel: int, location: int, entity_description: ReolinkSmartAINumberEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> float | None: ...
+    @raise_translated_error
+    async def async_set_native_value(self, value: float) -> None: ...
+
 class ReolinkHostNumberEntity(ReolinkHostCoordinatorEntity, NumberEntity):
     entity_description: ReolinkHostNumberEntityDescription
     _attr_mode: Incomplete

homeassistant-stubs/components/reolink/select.pyi

@@ -1,4 +1,4 @@
-from .entity import ReolinkChannelCoordinatorEntity as ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription as ReolinkChannelEntityDescription, ReolinkChimeCoordinatorEntity as ReolinkChimeCoordinatorEntity, ReolinkChimeEntityDescription as ReolinkChimeEntityDescription
+from .entity import ReolinkChannelCoordinatorEntity as ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription as ReolinkChannelEntityDescription, ReolinkChimeCoordinatorEntity as ReolinkChimeCoordinatorEntity, ReolinkChimeEntityDescription as ReolinkChimeEntityDescription, ReolinkHostCoordinatorEntity as ReolinkHostCoordinatorEntity, ReolinkHostEntityDescription as ReolinkHostEntityDescription
 from .util import ReolinkConfigEntry as ReolinkConfigEntry, ReolinkData as ReolinkData, raise_translated_error as raise_translated_error
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
@@ -19,6 +19,12 @@ class ReolinkSelectEntityDescription(SelectEntityDescription, ReolinkChannelEnti
     method: Callable[[Host, int, str], Any]
     value: Callable[[Host, int], str] | None = ...
 
+@dataclass(frozen=True, kw_only=True)
+class ReolinkHostSelectEntityDescription(SelectEntityDescription, ReolinkHostEntityDescription):
+    get_options: Callable[[Host], list[str]]
+    method: Callable[[Host, str], Any]
+    value: Callable[[Host], str]
+
 @dataclass(frozen=True, kw_only=True)
 class ReolinkChimeSelectEntityDescription(SelectEntityDescription, ReolinkChimeEntityDescription):
     get_options: list[str]
@@ -28,6 +34,7 @@ class ReolinkChimeSelectEntityDescription(SelectEntityDescription, ReolinkChimeE
 def _get_quick_reply_id(api: Host, ch: int, mess: str) -> int: ...
 
 SELECT_ENTITIES: Incomplete
+HOST_SELECT_ENTITIES: Incomplete
 CHIME_SELECT_ENTITIES: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ReolinkConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
@@ -42,6 +49,15 @@ class ReolinkSelectEntity(ReolinkChannelCoordinatorEntity, SelectEntity):
     @raise_translated_error
     async def async_select_option(self, option: str) -> None: ...
 
+class ReolinkHostSelectEntity(ReolinkHostCoordinatorEntity, SelectEntity):
+    entity_description: ReolinkHostSelectEntityDescription
+    _attr_options: Incomplete
+    def __init__(self, reolink_data: ReolinkData, entity_description: ReolinkHostSelectEntityDescription) -> None: ...
+    @property
+    def current_option(self) -> str | None: ...
+    @raise_translated_error
+    async def async_select_option(self, option: str) -> None: ...
+
 class ReolinkChimeSelectEntity(ReolinkChimeCoordinatorEntity, SelectEntity):
     entity_description: ReolinkChimeSelectEntityDescription
     _log_error: bool

homeassistant-stubs/components/reolink/update.pyi

@@ -1,7 +1,7 @@
 from . import DEVICE_UPDATE_INTERVAL as DEVICE_UPDATE_INTERVAL
 from .const import DOMAIN as DOMAIN
 from .entity import ReolinkChannelCoordinatorEntity as ReolinkChannelCoordinatorEntity, ReolinkChannelEntityDescription as ReolinkChannelEntityDescription, ReolinkHostCoordinatorEntity as ReolinkHostCoordinatorEntity, ReolinkHostEntityDescription as ReolinkHostEntityDescription
-from .util import ReolinkConfigEntry as ReolinkConfigEntry, ReolinkData as ReolinkData
+from .util import ReolinkConfigEntry as ReolinkConfigEntry, ReolinkData as ReolinkData, raise_translated_error as raise_translated_error
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.update import UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityDescription as UpdateEntityDescription, UpdateEntityFeature as UpdateEntityFeature
@@ -51,6 +51,7 @@ class ReolinkUpdateBaseEntity(CoordinatorEntity[DataUpdateCoordinator[None]], Up
     def available(self) -> bool: ...
     def version_is_newer(self, latest_version: str, installed_version: str) -> bool: ...
     async def async_release_notes(self) -> str | None: ...
+    @raise_translated_error
     async def async_install(self, version: str | None, backup: bool, **kwargs: Any) -> None: ...
     async def _pause_update_coordinator(self) -> None: ...
     async def _resume_update_coordinator(self, *args: Any) -> None: ...

homeassistant-stubs/components/reolink/util.pyi

@@ -1,5 +1,6 @@
 from .const import DOMAIN as DOMAIN
 from .host import ReolinkHost as ReolinkHost
+from _typeshed import Incomplete
 from collections.abc import Awaitable, Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant import config_entries as config_entries
@@ -8,7 +9,9 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
 from homeassistant.helpers import device_registry as dr
 from homeassistant.helpers.storage import Store as Store
+from homeassistant.helpers.translation import async_get_exception_message as async_get_exception_message
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+from reolink_aio.exceptions import ReolinkError
 from typing import Any
 
 STORAGE_VERSION: int
@@ -24,4 +27,8 @@ def is_connected(hass: HomeAssistant, config_entry: config_entries.ConfigEntry)
 def get_host(hass: HomeAssistant, config_entry_id: str) -> ReolinkHost: ...
 def get_store(hass: HomeAssistant, config_entry_id: str) -> Store[str]: ...
 def get_device_uid_and_ch(device: dr.DeviceEntry, host: ReolinkHost) -> tuple[list[str], int | None, bool]: ...
+def check_translation_key(err: ReolinkError) -> str | None: ...
+
+_EXCEPTION_TO_TRANSLATION_KEY: Incomplete
+
 def raise_translated_error[**P, R](func: Callable[P, Awaitable[R]]) -> Callable[P, Coroutine[Any, Any, R]]: ...

homeassistant-stubs/components/roborock/__init__.pyi

@@ -6,17 +6,19 @@ from collections.abc import Coroutine
 from homeassistant.const import CONF_USERNAME as CONF_USERNAME, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from roborock import HomeDataRoom as HomeDataRoom
 from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, UserData
+from roborock.web_api import RoborockApiClient
 from typing import Any
 
 SCAN_INTERVAL: Incomplete
 _LOGGER: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> bool: ...
-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: ...
+def build_setup_functions(hass: HomeAssistant, entry: RoborockConfigEntry, device_map: dict[str, HomeDataDevice], user_data: UserData, product_info: dict[str, HomeDataProduct], home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient) -> list[Coroutine[Any, Any, RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None]]: ...
+async def setup_device(hass: HomeAssistant, entry: RoborockConfigEntry, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct, home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient) -> RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01 | None: ...
+async def setup_device_v1(hass: HomeAssistant, entry: RoborockConfigEntry, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct, home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient) -> RoborockDataUpdateCoordinator | None: ...
 async def setup_device_a01(hass: HomeAssistant, entry: RoborockConfigEntry, user_data: UserData, device: HomeDataDevice, product_info: HomeDataProduct) -> RoborockDataUpdateCoordinatorA01 | None: ...
 async def async_unload_entry(hass: HomeAssistant, entry: RoborockConfigEntry) -> bool: ...
 async def update_listener(hass: HomeAssistant, entry: RoborockConfigEntry) -> None: ...

homeassistant-stubs/components/roborock/binary_sensor.pyi

@@ -4,14 +4,17 @@ 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.const import EntityCategory as EntityCategory
+from homeassistant.const import ATTR_BATTERY_CHARGING as ATTR_BATTERY_CHARGING, EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from roborock.roborock_typing import DeviceProp as DeviceProp
 
+PARALLEL_UPDATES: int
+
 @dataclass(frozen=True, kw_only=True)
 class RoborockBinarySensorDescription(BinarySensorEntityDescription):
     value_fn: Callable[[DeviceProp], bool | int | None]
+    is_dock_entity: bool = ...
 
 BINARY_SENSOR_DESCRIPTIONS: Incomplete
 

homeassistant-stubs/components/roborock/button.pyi

@@ -1,5 +1,5 @@
 from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
-from .entity import RoborockEntityV1 as RoborockEntityV1
+from .entity import RoborockEntity as RoborockEntity, RoborockEntityV1 as RoborockEntityV1
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
@@ -7,6 +7,9 @@ from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from roborock.roborock_typing import RoborockCommand
+from typing import Any
+
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockButtonDescription(ButtonEntityDescription):
@@ -21,3 +24,10 @@ class RoborockButtonEntity(RoborockEntityV1, ButtonEntity):
     entity_description: RoborockButtonDescription
     def __init__(self, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockButtonDescription) -> None: ...
     async def async_press(self) -> None: ...
+
+class RoborockRoutineButtonEntity(RoborockEntity, ButtonEntity):
+    entity_description: ButtonEntityDescription
+    _routine_id: Incomplete
+    _coordinator: Incomplete
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator, entity_description: ButtonEntityDescription) -> None: ...
+    async def async_press(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/roborock/config_flow.pyi

@@ -1,9 +1,12 @@
+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 _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_REAUTH as SOURCE_REAUTH
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_REAUTH as SOURCE_REAUTH
 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.web_api import RoborockApiClient
 from typing import Any
@@ -18,15 +21,16 @@ class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def _request_code(self) -> dict: ...
     async def async_step_code(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    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: ...
     @staticmethod
     @callback
-    def async_get_options_flow(config_entry: ConfigEntry) -> RoborockOptionsFlowHandler: ...
+    def async_get_options_flow(config_entry: RoborockConfigEntry) -> RoborockOptionsFlowHandler: ...
 
 class RoborockOptionsFlowHandler(OptionsFlow):
     options: Incomplete
-    def __init__(self, config_entry: ConfigEntry) -> None: ...
+    def __init__(self, config_entry: RoborockConfigEntry) -> None: ...
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_drawables(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/roborock/const.pyi

@@ -8,10 +8,16 @@ CONF_USER_DATA: str
 DRAWABLES: str
 DEFAULT_DRAWABLES: Incomplete
 PLATFORMS: Incomplete
-IMAGE_CACHE_INTERVAL: int
+IMAGE_CACHE_INTERVAL: Incomplete
 MAP_SLEEP: int
 GET_MAPS_SERVICE_NAME: str
+MAP_SCALE: int
 MAP_FILE_FORMAT: str
 MAP_FILENAME_SUFFIX: str
 SET_VACUUM_GOTO_POSITION_SERVICE_NAME: str
 GET_VACUUM_CURRENT_POSITION_SERVICE_NAME: str
+A01_UPDATE_INTERVAL: Incomplete
+V1_CLOUD_IN_CLEANING_INTERVAL: Incomplete
+V1_CLOUD_NOT_CLEANING_INTERVAL: Incomplete
+V1_LOCAL_IN_CLEANING_INTERVAL: Incomplete
+V1_LOCAL_NOT_CLEANING_INTERVAL: Incomplete

homeassistant-stubs/components/roborock/coordinator.pyi

@@ -1,4 +1,4 @@
-from .const import DOMAIN as DOMAIN
+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 _typeshed import Incomplete
@@ -6,18 +6,21 @@ from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_CONNECTIONS as ATTR_CONNECTIONS
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.typing import StateType as StateType
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 from homeassistant.util import slugify as slugify
 from propcache.api import cached_property
 from roborock import HomeDataRoom as HomeDataRoom
-from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, NetworkInfo as NetworkInfo
+from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, HomeDataScene as HomeDataScene, NetworkInfo as NetworkInfo, UserData as UserData
 from roborock.roborock_message import RoborockDyadDataProtocol, RoborockZeoProtocol
 from roborock.roborock_typing import DeviceProp
 from roborock.version_1_apis.roborock_local_client_v1 import RoborockLocalClientV1
 from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1 as RoborockMqttClientV1
 from roborock.version_a01_apis import RoborockClientA01 as RoborockClientA01
+from roborock.web_api import RoborockApiClient as RoborockApiClient
+from vacuum_map_parser_base.map_data import MapData as MapData
 
 SCAN_INTERVAL: Incomplete
 _LOGGER: Incomplete
@@ -39,18 +42,30 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
     maps: dict[int, RoborockMapInfo]
     _home_data_rooms: Incomplete
     map_storage: Incomplete
-    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: HomeDataDevice, device_networking: NetworkInfo, product_info: HomeDataProduct, cloud_api: RoborockMqttClientV1, home_data_rooms: list[HomeDataRoom]) -> None: ...
+    _user_data: Incomplete
+    _api_client: Incomplete
+    _is_cloud_api: bool
+    map_parser: Incomplete
+    def __init__(self, hass: HomeAssistant, config_entry: RoborockConfigEntry, device: HomeDataDevice, device_networking: NetworkInfo, product_info: HomeDataProduct, cloud_api: RoborockMqttClientV1, home_data_rooms: list[HomeDataRoom], api_client: RoborockApiClient, user_data: UserData) -> None: ...
+    @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 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: ...
 
 class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]]):
     config_entry: RoborockConfigEntry

homeassistant-stubs/components/roborock/entity.pyi

@@ -26,6 +26,8 @@ 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: ...
     async def send(self, command: RoborockCommand | str, params: dict[str, Any] | list[Any] | int | None = None) -> dict: ...
     @property
     def api(self) -> RoborockClientV1: ...
@@ -38,7 +40,7 @@ class RoborockCoordinatedEntityV1(RoborockEntityV1, CoordinatorEntity[RoborockDa
     _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) -> None: ...
+    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: ...
     @property

homeassistant-stubs/components/roborock/image.pyi

@@ -1,17 +1,15 @@
-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 RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator
 from .entity import RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable
 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
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 
 _LOGGER: Incomplete
+PARALLEL_UPDATES: int
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
@@ -20,16 +18,13 @@ class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
     image_last_updated: datetime
     _attr_name: str
     config_entry: Incomplete
-    parser: Incomplete
     map_flag: Incomplete
     cached_map: bytes
     _attr_entity_category: Incomplete
-    def __init__(self, config_entry: ConfigEntry, unique_id: str, coordinator: RoborockDataUpdateCoordinator, map_flag: int, map_name: str, parser: Callable[[bytes], bytes | None]) -> None: ...
+    def __init__(self, config_entry: ConfigEntry, unique_id: str, coordinator: RoborockDataUpdateCoordinator, map_flag: int, map_name: str) -> None: ...
     @property
     def is_selected(self) -> bool: ...
     _attr_image_last_updated: Incomplete
     async def async_added_to_hass(self) -> None: ...
     def _handle_coordinator_update(self) -> None: ...
     async def async_image(self) -> bytes | None: ...
-
-async def refresh_coordinators(hass: HomeAssistant, coord: RoborockDataUpdateCoordinator) -> None: ...

homeassistant-stubs/components/roborock/models.pyi

@@ -1,7 +1,9 @@
 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 typing import Any
+from vacuum_map_parser_base.map_data import MapData as MapData
 
 @dataclass
 class RoborockHassDeviceInfo:
@@ -22,3 +24,8 @@ 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

@@ -14,6 +14,7 @@ from roborock.version_1_apis.roborock_client_v1 import AttributeCache as Attribu
 from typing import Any
 
 _LOGGER: Incomplete
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockNumberDescription(NumberEntityDescription):

homeassistant-stubs/components/roborock/select.pyi

@@ -8,17 +8,19 @@ from homeassistant.components.select import SelectEntity as SelectEntity, Select
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 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
+from roborock.roborock_typing import DeviceProp as DeviceProp, RoborockCommand
+
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSelectDescription(SelectEntityDescription):
     api_command: RoborockCommand
-    value_fn: Callable[[Status], str | None]
-    options_lambda: Callable[[Status], list[str] | None]
-    parameter_lambda: Callable[[str, Status], list[int]]
+    value_fn: Callable[[DeviceProp], str | None]
+    options_lambda: Callable[[DeviceProp], list[str] | None]
+    parameter_lambda: Callable[[str, DeviceProp], list[int]]
     protocol_listener: RoborockDataProtocol | None = ...
+    is_dock_entity: bool = ...
 
 SELECT_DESCRIPTIONS: list[RoborockSelectDescription]
 

homeassistant-stubs/components/roborock/sensor.pyi

@@ -1,6 +1,6 @@
 import datetime
 from .coordinator import RoborockConfigEntry as RoborockConfigEntry, RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
-from .entity import RoborockCoordinatedEntityA01 as RoborockCoordinatedEntityA01, RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1
+from .entity import RoborockCoordinatedEntityA01 as RoborockCoordinatedEntityA01, RoborockCoordinatedEntityV1 as RoborockCoordinatedEntityV1, RoborockEntity as RoborockEntity
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
@@ -12,10 +12,13 @@ from homeassistant.helpers.typing import StateType as StateType
 from roborock.roborock_message import RoborockDataProtocol, RoborockDyadDataProtocol, RoborockZeoProtocol
 from roborock.roborock_typing import DeviceProp as DeviceProp
 
+PARALLEL_UPDATES: int
+
 @dataclass(frozen=True, kw_only=True)
 class RoborockSensorDescription(SensorEntityDescription):
     value_fn: Callable[[DeviceProp], StateType | datetime.datetime]
     protocol_listener: RoborockDataProtocol | None = ...
+    is_dock_entity: bool = ...
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockSensorDescriptionA01(SensorEntityDescription):
@@ -34,6 +37,16 @@ class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
     @property
     def native_value(self) -> StateType | datetime.datetime: ...
 
+class RoborockCurrentRoom(RoborockCoordinatedEntityV1, SensorEntity):
+    _attr_device_class: Incomplete
+    _attr_translation_key: str
+    _attr_entity_category: Incomplete
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator) -> None: ...
+    @property
+    def options(self) -> list[str]: ...
+    @property
+    def native_value(self) -> str | None: ...
+
 class RoborockSensorEntityA01(RoborockCoordinatedEntityA01, SensorEntity):
     entity_description: RoborockSensorDescriptionA01
     def __init__(self, coordinator: RoborockDataUpdateCoordinatorA01, description: RoborockSensorDescriptionA01) -> None: ...

homeassistant-stubs/components/roborock/switch.pyi

@@ -14,12 +14,14 @@ from roborock.version_1_apis.roborock_client_v1 import AttributeCache as Attribu
 from typing import Any
 
 _LOGGER: Incomplete
+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
+    is_dock_entity: bool = ...
 
 SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription]
 

homeassistant-stubs/components/roborock/time.pyi

@@ -16,6 +16,7 @@ from roborock.version_1_apis.roborock_client_v1 import AttributeCache as Attribu
 from typing import Any
 
 _LOGGER: Incomplete
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class RoborockTimeDescription(TimeEntityDescription):

homeassistant-stubs/components/roborock/vacuum.pyi

@@ -1,7 +1,6 @@
 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 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
 from homeassistant.components.vacuum import StateVacuumEntity as StateVacuumEntity, VacuumActivity as VacuumActivity, VacuumEntityFeature as VacuumEntityFeature
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse
@@ -11,6 +10,7 @@ from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 from typing import Any
 
 STATE_CODE_TO_STATE: Incomplete
+PARALLEL_UPDATES: int
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: RoborockConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/sensor/const.pyi

@@ -74,6 +74,7 @@ DEVICE_CLASSES: Final[list[str]]
 
 class SensorStateClass(StrEnum):
     MEASUREMENT = 'measurement'
+    MEASUREMENT_ANGLE = 'measurement_angle'
     TOTAL = 'total'
     TOTAL_INCREASING = 'total_increasing'
 
@@ -82,3 +83,4 @@ STATE_CLASSES: Final[list[str]]
 UNIT_CONVERTERS: dict[SensorDeviceClass | str | None, type[BaseUnitConverter]]
 DEVICE_CLASS_UNITS: dict[SensorDeviceClass, set[type[StrEnum] | str | None]]
 DEVICE_CLASS_STATE_CLASSES: dict[SensorDeviceClass, set[SensorStateClass]]
+STATE_CLASS_UNITS: dict[SensorStateClass | str, set[type[StrEnum] | str | None]]