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

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: ...
 

homeassistant-stubs/components/schedule/__init__.pyi

@@ -9,7 +9,7 @@ from homeassistant.helpers.entity_component import EntityComponent as EntityComp
 from homeassistant.helpers.event import async_track_point_in_utc_time as async_track_point_in_utc_time
 from homeassistant.helpers.service import async_register_admin_service as async_register_admin_service
 from homeassistant.helpers.storage import Store as Store
-from homeassistant.helpers.typing import ConfigType as ConfigType
+from homeassistant.helpers.typing import ConfigType as ConfigType, VolDictType as VolDictType
 from typing import Any, Literal
 
 STORAGE_VERSION: int
@@ -19,11 +19,11 @@ def valid_schedule(schedule: list[dict[str, str]]) -> list[dict[str, str]]: ...
 def deserialize_to_time(value: Any) -> Any: ...
 def serialize_to_time(value: Any) -> Any: ...
 
-BASE_SCHEMA: Incomplete
-TIME_RANGE_SCHEMA: Incomplete
+BASE_SCHEMA: VolDictType
+TIME_RANGE_SCHEMA: VolDictType
 STORAGE_TIME_RANGE_SCHEMA: Incomplete
-SCHEDULE_SCHEMA: Incomplete
-STORAGE_SCHEDULE_SCHEMA: Incomplete
+SCHEDULE_SCHEMA: VolDictType
+STORAGE_SCHEDULE_SCHEMA: VolDictType
 CONFIG_SCHEMA: Incomplete
 STORAGE_SCHEMA: Incomplete
 ENTITY_SCHEMA: Incomplete

homeassistant-stubs/components/select/__init__.pyi

@@ -1,12 +1,14 @@
 from .const import ATTR_CYCLE as ATTR_CYCLE, ATTR_OPTION as ATTR_OPTION, ATTR_OPTIONS as ATTR_OPTIONS, DOMAIN as DOMAIN, SERVICE_SELECT_FIRST as SERVICE_SELECT_FIRST, SERVICE_SELECT_LAST as SERVICE_SELECT_LAST, SERVICE_SELECT_NEXT as SERVICE_SELECT_NEXT, SERVICE_SELECT_OPTION as SERVICE_SELECT_OPTION, SERVICE_SELECT_PREVIOUS as SERVICE_SELECT_PREVIOUS
 from _typeshed import Incomplete
 from functools import cached_property
-from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.entity import Entity, EntityDescription
 from typing import Any
 
 __all__ = ['ATTR_CYCLE', 'ATTR_OPTION', 'ATTR_OPTIONS', 'DOMAIN', 'PLATFORM_SCHEMA_BASE', 'PLATFORM_SCHEMA', 'SelectEntity', 'SelectEntityDescription', 'SERVICE_SELECT_FIRST', 'SERVICE_SELECT_LAST', 'SERVICE_SELECT_NEXT', 'SERVICE_SELECT_OPTION', 'SERVICE_SELECT_PREVIOUS']
 
+PLATFORM_SCHEMA: Incomplete
+PLATFORM_SCHEMA_BASE: Incomplete
+
 class SelectEntityDescription(EntityDescription, frozen_or_thawed=True):
     options: list[str] | None
     def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, options) -> None: ...

homeassistant-stubs/components/sensor/__init__.pyi

@@ -7,7 +7,6 @@ from decimal import Decimal
 from functools import cached_property
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.entity import Entity, EntityDescription
 from homeassistant.helpers.entity_platform import EntityPlatform
 from homeassistant.helpers.restore_state import ExtraStoredData, RestoreEntity
@@ -16,6 +15,9 @@ from typing import Any, Self
 
 __all__ = ['ATTR_LAST_RESET', 'ATTR_OPTIONS', 'ATTR_STATE_CLASS', 'CONF_STATE_CLASS', 'DEVICE_CLASS_STATE_CLASSES', 'DOMAIN', 'PLATFORM_SCHEMA_BASE', 'PLATFORM_SCHEMA', 'RestoreSensor', 'SensorDeviceClass', 'SensorEntity', 'SensorEntityDescription', 'SensorExtraStoredData', 'SensorStateClass']
 
+PLATFORM_SCHEMA: Incomplete
+PLATFORM_SCHEMA_BASE: Incomplete
+
 class SensorEntityDescription(EntityDescription, frozen_or_thawed=True):
     device_class: SensorDeviceClass | None
     last_reset: datetime | None

homeassistant-stubs/components/sensor/const.pyi

@@ -1,8 +1,8 @@
 from _typeshed import Incomplete
 from enum import StrEnum
-from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_BILLION as CONCENTRATION_PARTS_PER_BILLION, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, LIGHT_LUX as LIGHT_LUX, PERCENTAGE as PERCENTAGE, POWER_VOLT_AMPERE_REACTIVE as POWER_VOLT_AMPERE_REACTIVE, SIGNAL_STRENGTH_DECIBELS as SIGNAL_STRENGTH_DECIBELS, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfApparentPower as UnitOfApparentPower, UnitOfDataRate as UnitOfDataRate, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfFrequency as UnitOfFrequency, UnitOfInformation as UnitOfInformation, UnitOfIrradiance as UnitOfIrradiance, UnitOfLength as UnitOfLength, UnitOfMass as UnitOfMass, UnitOfPower as UnitOfPower, UnitOfPrecipitationDepth as UnitOfPrecipitationDepth, UnitOfPressure as UnitOfPressure, UnitOfSoundPressure as UnitOfSoundPressure, UnitOfSpeed as UnitOfSpeed, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime, UnitOfVolume as UnitOfVolume, UnitOfVolumeFlowRate as UnitOfVolumeFlowRate, UnitOfVolumetricFlux as UnitOfVolumetricFlux
+from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_BILLION as CONCENTRATION_PARTS_PER_BILLION, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, LIGHT_LUX as LIGHT_LUX, PERCENTAGE as PERCENTAGE, POWER_VOLT_AMPERE_REACTIVE as POWER_VOLT_AMPERE_REACTIVE, SIGNAL_STRENGTH_DECIBELS as SIGNAL_STRENGTH_DECIBELS, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfApparentPower as UnitOfApparentPower, UnitOfConductivity as UnitOfConductivity, UnitOfDataRate as UnitOfDataRate, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfFrequency as UnitOfFrequency, UnitOfInformation as UnitOfInformation, UnitOfIrradiance as UnitOfIrradiance, UnitOfLength as UnitOfLength, UnitOfMass as UnitOfMass, UnitOfPower as UnitOfPower, UnitOfPrecipitationDepth as UnitOfPrecipitationDepth, UnitOfPressure as UnitOfPressure, UnitOfSoundPressure as UnitOfSoundPressure, UnitOfSpeed as UnitOfSpeed, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime, UnitOfVolume as UnitOfVolume, UnitOfVolumeFlowRate as UnitOfVolumeFlowRate, UnitOfVolumetricFlux as UnitOfVolumetricFlux
 from homeassistant.helpers.deprecation import DeprecatedConstantEnum as DeprecatedConstantEnum, all_with_deprecated_constants as all_with_deprecated_constants, check_if_deprecated_constant as check_if_deprecated_constant, dir_with_deprecated_constants as dir_with_deprecated_constants
-from homeassistant.util.unit_conversion import BaseUnitConverter as BaseUnitConverter, DataRateConverter as DataRateConverter, DistanceConverter as DistanceConverter, DurationConverter as DurationConverter, ElectricCurrentConverter as ElectricCurrentConverter, ElectricPotentialConverter as ElectricPotentialConverter, EnergyConverter as EnergyConverter, InformationConverter as InformationConverter, MassConverter as MassConverter, PowerConverter as PowerConverter, PressureConverter as PressureConverter, SpeedConverter as SpeedConverter, TemperatureConverter as TemperatureConverter, UnitlessRatioConverter as UnitlessRatioConverter, VolumeConverter as VolumeConverter, VolumeFlowRateConverter as VolumeFlowRateConverter
+from homeassistant.util.unit_conversion import BaseUnitConverter as BaseUnitConverter, ConductivityConverter as ConductivityConverter, DataRateConverter as DataRateConverter, DistanceConverter as DistanceConverter, DurationConverter as DurationConverter, ElectricCurrentConverter as ElectricCurrentConverter, ElectricPotentialConverter as ElectricPotentialConverter, EnergyConverter as EnergyConverter, InformationConverter as InformationConverter, MassConverter as MassConverter, PowerConverter as PowerConverter, PressureConverter as PressureConverter, SpeedConverter as SpeedConverter, TemperatureConverter as TemperatureConverter, UnitlessRatioConverter as UnitlessRatioConverter, VolumeConverter as VolumeConverter, VolumeFlowRateConverter as VolumeFlowRateConverter
 from typing import Final
 
 DOMAIN: Final[str]
@@ -21,6 +21,7 @@ class SensorDeviceClass(StrEnum):
     BATTERY: str
     CO: str
     CO2: str
+    CONDUCTIVITY: str
     CURRENT: str
     DATA_RATE: str
     DATA_SIZE: str

homeassistant-stubs/components/sensor/device_condition.pyi

@@ -17,6 +17,7 @@ CONF_IS_ATMOSPHERIC_PRESSURE: str
 CONF_IS_BATTERY_LEVEL: str
 CONF_IS_CO: str
 CONF_IS_CO2: str
+CONF_IS_CONDUCTIVITY: str
 CONF_IS_CURRENT: str
 CONF_IS_DATA_RATE: str
 CONF_IS_DATA_SIZE: str

homeassistant-stubs/components/sensor/device_trigger.pyi

@@ -17,6 +17,7 @@ CONF_ATMOSPHERIC_PRESSURE: str
 CONF_BATTERY_LEVEL: str
 CONF_CO: str
 CONF_CO2: str
+CONF_CONDUCTIVITY: str
 CONF_CURRENT: str
 CONF_DATA_RATE: str
 CONF_DATA_SIZE: str

homeassistant-stubs/components/shelly/__init__.pyi

@@ -10,9 +10,8 @@ from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONN
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from typing import Final
 
-BLOCK_PLATFORMS: Final[Incomplete]
+PLATFORMS: Final[Incomplete]
 BLOCK_SLEEPING_PLATFORMS: Final[Incomplete]
-RPC_PLATFORMS: Final[Incomplete]
 RPC_SLEEPING_PLATFORMS: Final[Incomplete]
 COAP_SCHEMA: Final[Incomplete]
 CONFIG_SCHEMA: Final[Incomplete]

homeassistant-stubs/components/shelly/climate.pyi

@@ -85,12 +85,15 @@ class RpcClimate(ShellyRpcEntity, ClimateEntity):
     _id: Incomplete
     _thermostat_type: Incomplete
     _attr_hvac_modes: Incomplete
+    _humidity_key: Incomplete
     def __init__(self, coordinator: ShellyRpcCoordinator, id_: int) -> None: ...
     @property
     def target_temperature(self) -> float | None: ...
     @property
     def current_temperature(self) -> float | None: ...
     @property
+    def current_humidity(self) -> float | None: ...
+    @property
     def hvac_mode(self) -> HVACMode: ...
     @property
     def hvac_action(self) -> HVACAction: ...

homeassistant-stubs/components/shelly/coordinator.pyi

@@ -16,11 +16,12 @@ from typing import Any
 
 @dataclass
 class ShellyEntryData:
+    platforms: list[Platform]
     block: ShellyBlockCoordinator | None = ...
     rest: ShellyRestCoordinator | None = ...
     rpc: ShellyRpcCoordinator | None = ...
     rpc_poll: ShellyRpcPollingCoordinator | None = ...
-    def __init__(self, block, rest, rpc, rpc_poll) -> None: ...
+    def __init__(self, platforms, block, rest, rpc, rpc_poll) -> None: ...
 ShellyConfigEntry = ConfigEntry[ShellyEntryData]
 
 class ShellyCoordinatorBase(DataUpdateCoordinator[None]):