homeassistant-stubs 2024.6.4__py3-none-any.whl → 2024.7.0__py3-none-any.whl

homeassistant-stubs/components/rest/notify.pyi

@@ -1,6 +1,6 @@
 import httpx
 from _typeshed import Incomplete
-from homeassistant.components.notify import ATTR_MESSAGE as ATTR_MESSAGE, ATTR_TARGET as ATTR_TARGET, ATTR_TITLE as ATTR_TITLE, ATTR_TITLE_DEFAULT as ATTR_TITLE_DEFAULT, BaseNotificationService as BaseNotificationService, PLATFORM_SCHEMA as PLATFORM_SCHEMA
+from homeassistant.components.notify import ATTR_MESSAGE as ATTR_MESSAGE, ATTR_TARGET as ATTR_TARGET, ATTR_TITLE as ATTR_TITLE, ATTR_TITLE_DEFAULT as ATTR_TITLE_DEFAULT, BaseNotificationService as BaseNotificationService
 from homeassistant.const import CONF_AUTHENTICATION as CONF_AUTHENTICATION, CONF_HEADERS as CONF_HEADERS, CONF_METHOD as CONF_METHOD, CONF_NAME as CONF_NAME, CONF_PARAMS as CONF_PARAMS, CONF_PASSWORD as CONF_PASSWORD, CONF_RESOURCE as CONF_RESOURCE, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL, HTTP_BASIC_AUTHENTICATION as HTTP_BASIC_AUTHENTICATION, HTTP_DIGEST_AUTHENTICATION as HTTP_DIGEST_AUTHENTICATION
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.httpx_client import get_async_client as get_async_client
@@ -16,6 +16,7 @@ CONF_TITLE_PARAMETER_NAME: str
 DEFAULT_MESSAGE_PARAM_NAME: str
 DEFAULT_METHOD: str
 DEFAULT_VERIFY_SSL: bool
+PLATFORM_SCHEMA: Incomplete
 _LOGGER: Incomplete
 
 async def async_get_service(hass: HomeAssistant, config: ConfigType, discovery_info: DiscoveryInfoType | None = None) -> RestNotificationService: ...

homeassistant-stubs/components/rest/sensor.pyi

@@ -5,7 +5,7 @@ from .entity import RestEntity as RestEntity
 from .schema import RESOURCE_SCHEMA as RESOURCE_SCHEMA, SENSOR_SCHEMA as SENSOR_SCHEMA
 from .util import parse_json_attributes as parse_json_attributes
 from _typeshed import Incomplete
-from homeassistant.components.sensor import CONF_STATE_CLASS as CONF_STATE_CLASS, PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorDeviceClass as SensorDeviceClass
+from homeassistant.components.sensor import CONF_STATE_CLASS as CONF_STATE_CLASS, SensorDeviceClass as SensorDeviceClass
 from homeassistant.components.sensor.helpers import async_parse_date_datetime as async_parse_date_datetime
 from homeassistant.const import CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_FORCE_UPDATE as CONF_FORCE_UPDATE, CONF_ICON as CONF_ICON, CONF_NAME as CONF_NAME, CONF_RESOURCE as CONF_RESOURCE, CONF_RESOURCE_TEMPLATE as CONF_RESOURCE_TEMPLATE, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_UNIT_OF_MEASUREMENT as CONF_UNIT_OF_MEASUREMENT, CONF_VALUE_TEMPLATE as CONF_VALUE_TEMPLATE
 from homeassistant.core import HomeAssistant as HomeAssistant
@@ -18,6 +18,7 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as Da
 from typing import Any
 
 _LOGGER: Incomplete
+PLATFORM_SCHEMA: Incomplete
 TRIGGER_ENTITY_OPTIONS: Incomplete
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...

homeassistant-stubs/components/rest/switch.pyi

@@ -1,6 +1,6 @@
 import httpx
 from _typeshed import Incomplete
-from homeassistant.components.switch import DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, PLATFORM_SCHEMA as PLATFORM_SCHEMA, SwitchEntity as SwitchEntity
+from homeassistant.components.switch import DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, SwitchEntity as SwitchEntity
 from homeassistant.const import CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_HEADERS as CONF_HEADERS, CONF_ICON as CONF_ICON, CONF_METHOD as CONF_METHOD, CONF_NAME as CONF_NAME, CONF_PARAMS as CONF_PARAMS, CONF_PASSWORD as CONF_PASSWORD, CONF_RESOURCE as CONF_RESOURCE, CONF_TIMEOUT as CONF_TIMEOUT, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import PlatformNotReady as PlatformNotReady
@@ -24,6 +24,7 @@ DEFAULT_NAME: str
 DEFAULT_TIMEOUT: int
 DEFAULT_VERIFY_SSL: bool
 SUPPORT_REST_METHODS: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 

homeassistant-stubs/components/rfxtrx/config_flow.pyi

@@ -8,6 +8,7 @@ from homeassistant.core import Event as Event, EventStateChangedData as EventSta
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import device_registry as dr
 from homeassistant.helpers.event import async_track_state_change_event as async_track_state_change_event
+from homeassistant.helpers.typing import VolDictType as VolDictType
 from typing import Any, TypedDict
 
 CONF_EVENT_CODE: str

homeassistant-stubs/components/roborock/__init__.pyi

@@ -0,0 +1,30 @@
+from .const import CONF_BASE_URL as CONF_BASE_URL, CONF_USER_DATA as CONF_USER_DATA, DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
+from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
+from _typeshed import Incomplete
+from collections.abc import Coroutine
+from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_USERNAME as CONF_USERNAME
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from roborock import HomeDataRoom as HomeDataRoom
+from roborock.containers import HomeDataDevice as HomeDataDevice, HomeDataProduct as HomeDataProduct, UserData
+from typing import Any
+
+SCAN_INTERVAL: Incomplete
+_LOGGER: Incomplete
+
+@dataclass
+class RoborockCoordinators:
+    v1: list[RoborockDataUpdateCoordinator]
+    a01: list[RoborockDataUpdateCoordinatorA01]
+    def values(self) -> list[RoborockDataUpdateCoordinator | RoborockDataUpdateCoordinatorA01]: ...
+    def __init__(self, v1, a01) -> None: ...
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> 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: ...

homeassistant-stubs/components/roborock/binary_sensor.pyi

@@ -0,0 +1,29 @@
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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: ...
+
+BINARY_SENSOR_DESCRIPTIONS: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RoborockBinarySensorEntity(RoborockCoordinatedEntityV1, BinarySensorEntity):
+    entity_description: RoborockBinarySensorDescription
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator, description: RoborockBinarySensorDescription) -> None: ...
+    @property
+    def is_on(self) -> bool: ...

homeassistant-stubs/components/roborock/button.pyi

@@ -0,0 +1,28 @@
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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: ...
+
+CONSUMABLE_BUTTON_DESCRIPTIONS: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RoborockButtonEntity(RoborockEntityV1, ButtonEntity):
+    entity_description: RoborockButtonDescription
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockButtonDescription) -> None: ...
+    async def async_press(self) -> None: ...

homeassistant-stubs/components/roborock/config_flow.pyi

@@ -0,0 +1,30 @@
+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, OptionsFlowWithConfigEntry as OptionsFlowWithConfigEntry
+from homeassistant.const import CONF_USERNAME as CONF_USERNAME
+from homeassistant.core import callback as callback
+from roborock.containers import UserData as UserData
+from roborock.web_api import RoborockApiClient
+from typing import Any
+
+_LOGGER: Incomplete
+
+class RoborockFlowHandler(ConfigFlow, domain=DOMAIN):
+    VERSION: int
+    reauth_entry: ConfigEntry | None
+    _username: Incomplete
+    _client: Incomplete
+    def __init__(self) -> None: ...
+    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_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
+    def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlow: ...
+
+class RoborockOptionsFlowHandler(OptionsFlowWithConfigEntry):
+    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

@@ -0,0 +1,13 @@
+from _typeshed import Incomplete
+from homeassistant.const import Platform as Platform
+
+DOMAIN: str
+CONF_ENTRY_CODE: str
+CONF_BASE_URL: str
+CONF_USER_DATA: str
+DRAWABLES: str
+DEFAULT_DRAWABLES: Incomplete
+PLATFORMS: Incomplete
+IMAGE_CACHE_INTERVAL: int
+MAP_SLEEP: int
+GET_MAPS_SERVICE_NAME: str

homeassistant-stubs/components/roborock/coordinator.pyi

@@ -0,0 +1,47 @@
+from .const import DOMAIN as DOMAIN
+from .models import RoborockA01HassDeviceInfo as RoborockA01HassDeviceInfo, RoborockHassDeviceInfo as RoborockHassDeviceInfo, RoborockMapInfo as RoborockMapInfo
+from _typeshed import Incomplete
+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 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
+from roborock.roborock_typing import DeviceProp
+from roborock.version_1_apis.roborock_mqtt_client_v1 import RoborockMqttClientV1 as RoborockMqttClientV1
+from roborock.version_a01_apis import RoborockClientA01 as RoborockClientA01
+
+SCAN_INTERVAL: Incomplete
+_LOGGER: Incomplete
+
+class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]):
+    roborock_device_info: Incomplete
+    api: Incomplete
+    cloud_api: Incomplete
+    device_info: Incomplete
+    current_map: Incomplete
+    maps: Incomplete
+    _home_data_rooms: Incomplete
+    def __init__(self, hass: HomeAssistant, device: HomeDataDevice, device_networking: NetworkInfo, product_info: HomeDataProduct, cloud_api: RoborockMqttClientV1, home_data_rooms: list[HomeDataRoom]) -> None: ...
+    async def verify_api(self) -> None: ...
+    async def release(self) -> None: ...
+    async def _update_device_prop(self) -> None: ...
+    async def _async_update_data(self) -> DeviceProp: ...
+    def _set_current_map(self) -> None: ...
+    async def get_maps(self) -> None: ...
+    async def get_rooms(self) -> None: ...
+    @property
+    def duid(self) -> str: ...
+
+class RoborockDataUpdateCoordinatorA01(DataUpdateCoordinator[dict[RoborockDyadDataProtocol | RoborockZeoProtocol, StateType]]):
+    api: Incomplete
+    device_info: Incomplete
+    request_protocols: Incomplete
+    roborock_device_info: Incomplete
+    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
+    def duid(self) -> str: ...

homeassistant-stubs/components/roborock/device.pyi

@@ -0,0 +1,53 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import RoborockDataUpdateCoordinator as RoborockDataUpdateCoordinator, RoborockDataUpdateCoordinatorA01 as RoborockDataUpdateCoordinatorA01
+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.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: ...
+
+class RoborockEntityV1(RoborockEntity):
+    _api: RoborockClientV1
+    def __init__(self, unique_id: str, device_info: DeviceInfo, api: RoborockClientV1) -> None: ...
+    def get_cache(self, attribute: CacheableAttribute) -> AttributeCache: ...
+    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) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    async def async_will_remove_from_hass(self) -> 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]):
+    _attr_unique_id: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinatorA01) -> None: ...

homeassistant-stubs/components/roborock/diagnostics.pyi

@@ -0,0 +1,13 @@
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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
+
+TO_REDACT_CONFIG: Incomplete
+TO_REDACT_COORD: Incomplete
+
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/roborock/image.pyi

@@ -0,0 +1,37 @@
+from . import RoborockCoordinators as RoborockCoordinators
+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
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+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: ...
+
+class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity):
+    _attr_has_entity_name: bool
+    image_last_updated: datetime
+    _attr_name: Incomplete
+    parser: Incomplete
+    _attr_image_last_updated: Incomplete
+    map_flag: Incomplete
+    cached_map: Incomplete
+    _attr_entity_category: Incomplete
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, map_flag: int, starting_map: bytes, map_name: str, drawables: list[Drawable]) -> None: ...
+    @property
+    def available(self) -> bool: ...
+    @property
+    def is_selected(self) -> bool: ...
+    def is_map_valid(self) -> bool: ...
+    def _handle_coordinator_update(self) -> None: ...
+    async def async_image(self) -> bytes | None: ...
+    def _create_image(self, map_bytes: bytes) -> bytes: ...
+
+async def create_coordinator_maps(coord: RoborockDataUpdateCoordinator, drawables: list[Drawable]) -> list[RoborockMap]: ...

homeassistant-stubs/components/roborock/models.pyi

@@ -0,0 +1,27 @@
+from dataclasses import dataclass
+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
+
+@dataclass
+class RoborockHassDeviceInfo:
+    device: HomeDataDevice
+    network_info: NetworkInfo
+    product: HomeDataProduct
+    props: DeviceProp
+    def as_dict(self) -> dict[str, dict[str, Any]]: ...
+    def __init__(self, device, network_info, product, props) -> None: ...
+
+@dataclass
+class RoborockA01HassDeviceInfo:
+    device: HomeDataDevice
+    product: HomeDataProduct
+    def as_dict(self) -> dict[str, dict[str, Any]]: ...
+    def __init__(self, device, product) -> None: ...
+
+@dataclass
+class RoborockMapInfo:
+    flag: int
+    name: str
+    rooms: dict[int, str]
+    def __init__(self, flag, name, rooms) -> None: ...

homeassistant-stubs/components/roborock/number.pyi

@@ -0,0 +1,35 @@
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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
+
+_LOGGER: Incomplete
+
+@dataclass(frozen=True, kw_only=True)
+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: ...
+
+NUMBER_DESCRIPTIONS: list[RoborockNumberDescription]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RoborockNumberEntity(RoborockEntityV1, NumberEntity):
+    entity_description: RoborockNumberDescription
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockNumberDescription) -> None: ...
+    @property
+    def native_value(self) -> float | None: ...
+    async def async_set_native_value(self, value: float) -> None: ...

homeassistant-stubs/components/roborock/select.pyi

@@ -0,0 +1,37 @@
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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
+
+@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]]
+    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: ...
+
+SELECT_DESCRIPTIONS: list[RoborockSelectDescription]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RoborockSelectEntity(RoborockCoordinatedEntityV1, SelectEntity):
+    entity_description: RoborockSelectDescription
+    _attr_options: Incomplete
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockSelectDescription, options: list[str]) -> None: ...
+    async def async_select_option(self, option: str) -> None: ...
+    @property
+    def current_option(self) -> str | None: ...

homeassistant-stubs/components/roborock/sensor.pyi

@@ -0,0 +1,47 @@
+import datetime
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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_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: ...
+
+@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: ...
+
+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: ...
+
+class RoborockSensorEntity(RoborockCoordinatedEntityV1, SensorEntity):
+    entity_description: RoborockSensorDescription
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator, description: RoborockSensorDescription) -> None: ...
+    @property
+    def native_value(self) -> StateType | datetime.datetime: ...
+
+class RoborockSensorEntityA01(RoborockCoordinatedEntityA01, SensorEntity):
+    entity_description: RoborockSensorDescriptionA01
+    def __init__(self, coordinator: RoborockDataUpdateCoordinatorA01, description: RoborockSensorDescriptionA01) -> None: ...
+    @property
+    def native_value(self) -> StateType: ...

homeassistant-stubs/components/roborock/switch.pyi

@@ -0,0 +1,37 @@
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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
+
+_LOGGER: Incomplete
+
+@dataclass(frozen=True, kw_only=True)
+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: ...
+
+SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class RoborockSwitch(RoborockEntityV1, SwitchEntity):
+    entity_description: RoborockSwitchDescription
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockSwitchDescription) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+    @property
+    def is_on(self) -> bool | None: ...

homeassistant-stubs/components/roborock/time.pyi

@@ -0,0 +1,38 @@
+import datetime
+from . import RoborockCoordinators as RoborockCoordinators
+from .const import DOMAIN as DOMAIN
+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 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
+
+_LOGGER: Incomplete
+
+@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]
+    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: ...
+
+class RoborockTimeEntity(RoborockEntityV1, TimeEntity):
+    entity_description: RoborockTimeDescription
+    def __init__(self, unique_id: str, coordinator: RoborockDataUpdateCoordinator, entity_description: RoborockTimeDescription) -> None: ...
+    @property
+    def native_value(self) -> time | None: ...
+    async def async_set_value(self, value: time) -> None: ...

homeassistant-stubs/components/roborock/vacuum.pyi

@@ -0,0 +1,39 @@
+from . import RoborockCoordinators as RoborockCoordinators
+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: ...
+
+class RoborockVacuum(RoborockCoordinatedEntityV1, StateVacuumEntity):
+    _attr_icon: str
+    _attr_supported_features: Incomplete
+    _attr_translation_key = DOMAIN
+    _attr_name: Incomplete
+    _attr_fan_speed_list: Incomplete
+    def __init__(self, coordinator: RoborockDataUpdateCoordinator) -> None: ...
+    @property
+    def state(self) -> str | None: ...
+    @property
+    def battery_level(self) -> int | None: ...
+    @property
+    def fan_speed(self) -> str | None: ...
+    async def async_start(self) -> None: ...
+    async def async_pause(self) -> None: ...
+    async def async_stop(self, **kwargs: Any) -> None: ...
+    async def async_return_to_base(self, **kwargs: Any) -> None: ...
+    async def async_clean_spot(self, **kwargs: Any) -> None: ...
+    async def async_locate(self, **kwargs: Any) -> None: ...
+    async def async_set_fan_speed(self, fan_speed: str, **kwargs: Any) -> None: ...
+    async def async_send_command(self, command: str, params: dict[str, Any] | list[Any] | None = None, **kwargs: Any) -> None: ...
+    async def get_maps(self) -> ServiceResponse: ...

homeassistant-stubs/components/roku/media_player.pyi

@@ -13,6 +13,7 @@ from homeassistant.const import ATTR_NAME as ATTR_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import VolDictType as VolDictType
 from typing import Any
 
 _LOGGER: Incomplete
@@ -20,7 +21,7 @@ STREAM_FORMAT_TO_MEDIA_TYPE: Incomplete
 ATTRS_TO_LAUNCH_PARAMS: Incomplete
 ATTRS_TO_PLAY_ON_ROKU_PARAMS: Incomplete
 ATTRS_TO_PLAY_ON_ROKU_AUDIO_PARAMS: Incomplete
-SEARCH_SCHEMA: Incomplete
+SEARCH_SCHEMA: VolDictType
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...