homeassistant-stubs 2025.3.4__py3-none-any.whl → 2025.4.0b1__py3-none-any.whl

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]]

homeassistant-stubs/components/sensor/recorder.pyi

@@ -2,8 +2,9 @@ import datetime
 from .const import ATTR_LAST_RESET as ATTR_LAST_RESET, ATTR_STATE_CLASS as ATTR_STATE_CLASS, DOMAIN as DOMAIN, SensorStateClass as SensorStateClass, UnitOfVolumeFlowRate as UnitOfVolumeFlowRate
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable
+from dataclasses import dataclass
 from homeassistant.components.recorder import get_instance as get_instance, history as history, statistics as statistics
-from homeassistant.components.recorder.models import StatisticData as StatisticData, StatisticMetaData as StatisticMetaData, StatisticResult as StatisticResult
+from homeassistant.components.recorder.models import StatisticData as StatisticData, StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData, StatisticResult as StatisticResult
 from homeassistant.const import ATTR_UNIT_OF_MEASUREMENT as ATTR_UNIT_OF_MEASUREMENT, REVOLUTIONS_PER_MINUTE as REVOLUTIONS_PER_MINUTE, UnitOfIrradiance as UnitOfIrradiance, UnitOfSoundPressure as UnitOfSoundPressure, UnitOfVolume as UnitOfVolume
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, callback as callback, split_entity_id as split_entity_id
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -17,6 +18,12 @@ from sqlalchemy.orm.session import Session as Session
 from typing import Any
 
 _LOGGER: Incomplete
+
+@dataclass
+class _StatisticsConfig:
+    types: set[str]
+    mean_type: StatisticMeanType = ...
+
 DEFAULT_STATISTICS: Incomplete
 EQUIVALENT_UNITS: Incomplete
 SEEN_DIP: HassKey[set[str]]
@@ -24,10 +31,15 @@ WARN_DIP: HassKey[set[str]]
 WARN_NEGATIVE: HassKey[set[str]]
 WARN_UNSUPPORTED_UNIT: HassKey[set[str]]
 WARN_UNSTABLE_UNIT: HassKey[set[str]]
+WARN_STATISTICS_MEAN_CHANGED: HassKey[set[str]]
 LINK_DEV_STATISTICS: str
+STATE_CLASS_REMOVED_ISSUE: str
+UNITS_CHANGED_ISSUE: str
+MEAN_TYPE_CHANGED_ISSUE: str
 
 def _get_sensor_states(hass: HomeAssistant) -> list[State]: ...
-def _time_weighted_average(fstates: list[tuple[float, State]], start: datetime.datetime, end: datetime.datetime) -> float: ...
+def _time_weighted_arithmetic_mean(fstates: list[tuple[float, State]], start: datetime.datetime, end: datetime.datetime) -> float: ...
+def _time_weighted_circular_mean(fstates: list[tuple[float, State]], start: datetime.datetime, end: datetime.datetime) -> float: ...
 def _get_units(fstates: list[tuple[float, State]]) -> set[str | None]: ...
 def _equivalent_units(units: set[str | None]) -> bool: ...
 def _entity_history_to_float_and_state(entity_history: Iterable[State]) -> list[tuple[float, State]]: ...
@@ -37,7 +49,7 @@ def _suggest_report_issue(hass: HomeAssistant, entity_id: str) -> str: ...
 def warn_dip(hass: HomeAssistant, entity_id: str, state: State, previous_fstate: float) -> None: ...
 def warn_negative(hass: HomeAssistant, entity_id: str, state: State) -> None: ...
 def reset_detected(hass: HomeAssistant, entity_id: str, fstate: float, previous_fstate: float | None, state: State) -> bool: ...
-def _wanted_statistics(sensor_states: list[State]) -> dict[str, set[str]]: ...
+def _wanted_statistics(sensor_states: list[State]) -> dict[str, _StatisticsConfig]: ...
 def _last_reset_as_utc_isoformat(last_reset_s: Any, entity_id: str) -> str | None: ...
 def _timestamp_to_isoformat_or_none(timestamp: float | None) -> str | None: ...
 def compile_statistics(hass: HomeAssistant, session: Session, start: datetime.datetime, end: datetime.datetime) -> statistics.PlatformCompiledStatistics: ...

homeassistant-stubs/components/shelly/__init__.pyi

@@ -1,9 +1,9 @@
 from .const import BLOCK_EXPECTED_SLEEP_PERIOD as BLOCK_EXPECTED_SLEEP_PERIOD, BLOCK_WRONG_SLEEP_PERIOD as BLOCK_WRONG_SLEEP_PERIOD, CONF_COAP_PORT as CONF_COAP_PORT, CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DOMAIN as DOMAIN, FIRMWARE_UNSUPPORTED_ISSUE_ID as FIRMWARE_UNSUPPORTED_ISSUE_ID, LOGGER as LOGGER, MODELS_WITH_WRONG_SLEEP_PERIOD as MODELS_WITH_WRONG_SLEEP_PERIOD, PUSH_UPDATE_ISSUE_ID as PUSH_UPDATE_ISSUE_ID
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyEntryData as ShellyEntryData, ShellyRestCoordinator as ShellyRestCoordinator, ShellyRpcCoordinator as ShellyRpcCoordinator, ShellyRpcPollingCoordinator as ShellyRpcPollingCoordinator
-from .utils import async_create_issue_unsupported_firmware as async_create_issue_unsupported_firmware, get_coap_context as get_coap_context, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_ws_context as get_ws_context
+from .utils import async_create_issue_unsupported_firmware as async_create_issue_unsupported_firmware, get_coap_context as get_coap_context, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_rpc_scripts_event_types as get_rpc_scripts_event_types, get_ws_context as get_ws_context
 from _typeshed import Incomplete
 from homeassistant.components.bluetooth import async_remove_scanner as async_remove_scanner
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME, Platform as Platform
+from homeassistant.const import CONF_HOST as CONF_HOST, CONF_MODEL as CONF_MODEL, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME, Platform as Platform
 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

homeassistant-stubs/components/shelly/bluetooth/__init__.pyi

@@ -1,7 +1,9 @@
 from ..const import BLEScannerMode as BLEScannerMode
 from ..coordinator import ShellyRpcCoordinator as ShellyRpcCoordinator
-from homeassistant.components.bluetooth import async_register_scanner as async_register_scanner
+from _typeshed import Incomplete
+from homeassistant.components.bluetooth import BluetoothScanningMode as BluetoothScanningMode, async_register_scanner as async_register_scanner
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant
-from homeassistant.helpers.device_registry import format_mac as format_mac
+
+BLE_SCANNER_MODE_TO_BLUETOOTH_SCANNING_MODE: Incomplete
 
 async def async_connect_scanner(hass: HomeAssistant, coordinator: ShellyRpcCoordinator, scanner_mode: BLEScannerMode, device_id: str) -> CALLBACK_TYPE: ...

homeassistant-stubs/components/shelly/button.pyi

@@ -1,14 +1,15 @@
-from .const import LOGGER as LOGGER, SHELLY_GAS_MODELS as SHELLY_GAS_MODELS
+from .const import DOMAIN as DOMAIN, LOGGER as LOGGER, SHELLY_GAS_MODELS as SHELLY_GAS_MODELS
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
-from .utils import get_device_entry_gen as get_device_entry_gen
+from .utils import get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable, Coroutine
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import entity_registry as er
-from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceInfo as DeviceInfo
+from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH as CONNECTION_BLUETOOTH, CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from homeassistant.util import slugify as slugify
@@ -16,19 +17,33 @@ from typing import Any, Final
 
 @dataclass(frozen=True, kw_only=True)
 class ShellyButtonDescription[_ShellyCoordinatorT: ShellyBlockCoordinator | ShellyRpcCoordinator](ButtonEntityDescription):
-    press_action: Callable[[_ShellyCoordinatorT], Coroutine[Any, Any, None]]
+    press_action: str
     supported: Callable[[_ShellyCoordinatorT], bool] = ...
 
 BUTTONS: Final[list[ShellyButtonDescription[Any]]]
+BLU_TRV_BUTTONS: Final[list[ShellyButtonDescription]]
 
 @callback
 def async_migrate_unique_ids(coordinator: ShellyRpcCoordinator | ShellyBlockCoordinator, entity_entry: er.RegistryEntry) -> dict[str, Any] | None: ...
 async def async_setup_entry(hass: HomeAssistant, config_entry: ShellyConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class ShellyButton(CoordinatorEntity[ShellyRpcCoordinator | ShellyBlockCoordinator], ButtonEntity):
+class ShellyBaseButton(CoordinatorEntity[ShellyRpcCoordinator | ShellyBlockCoordinator], ButtonEntity):
     entity_description: ShellyButtonDescription[ShellyRpcCoordinator | ShellyBlockCoordinator]
+    def __init__(self, coordinator: ShellyRpcCoordinator | ShellyBlockCoordinator, description: ShellyButtonDescription[ShellyRpcCoordinator | ShellyBlockCoordinator]) -> None: ...
+    async def async_press(self) -> None: ...
+    async def _press_method(self) -> None: ...
+
+class ShellyButton(ShellyBaseButton):
     _attr_name: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
     def __init__(self, coordinator: ShellyRpcCoordinator | ShellyBlockCoordinator, description: ShellyButtonDescription[ShellyRpcCoordinator | ShellyBlockCoordinator]) -> None: ...
-    async def async_press(self) -> None: ...
+    async def _press_method(self) -> None: ...
+
+class ShellyBluTrvButton(ShellyBaseButton):
+    _attr_name: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
+    _id: Incomplete
+    def __init__(self, coordinator: ShellyRpcCoordinator, description: ShellyButtonDescription, id_: int) -> None: ...
+    async def _press_method(self) -> None: ...

homeassistant-stubs/components/shelly/climate.pyi

@@ -1,4 +1,4 @@
-from .const import BLU_TRV_TEMPERATURE_SETTINGS as BLU_TRV_TEMPERATURE_SETTINGS, BLU_TRV_TIMEOUT as BLU_TRV_TIMEOUT, DOMAIN as DOMAIN, LOGGER as LOGGER, NOT_CALIBRATED_ISSUE_ID as NOT_CALIBRATED_ISSUE_ID, RPC_THERMOSTAT_SETTINGS as RPC_THERMOSTAT_SETTINGS, SHTRV_01_TEMPERATURE_SETTINGS as SHTRV_01_TEMPERATURE_SETTINGS
+from .const import BLU_TRV_TEMPERATURE_SETTINGS as BLU_TRV_TEMPERATURE_SETTINGS, DOMAIN as DOMAIN, LOGGER as LOGGER, NOT_CALIBRATED_ISSUE_ID as NOT_CALIBRATED_ISSUE_ID, RPC_THERMOSTAT_SETTINGS as RPC_THERMOSTAT_SETTINGS, SHTRV_01_TEMPERATURE_SETTINGS as SHTRV_01_TEMPERATURE_SETTINGS
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .entity import ShellyRpcEntity as ShellyRpcEntity
 from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, is_rpc_thermostat_internal_actuator as is_rpc_thermostat_internal_actuator

homeassistant-stubs/components/shelly/config_flow.pyi

@@ -1,10 +1,10 @@
 from .const import BLEScannerMode as BLEScannerMode, CONF_BLE_SCANNER_MODE as CONF_BLE_SCANNER_MODE, CONF_GEN as CONF_GEN, CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DOMAIN as DOMAIN, LOGGER as LOGGER
-from .coordinator import async_reconnect_soon as async_reconnect_soon
+from .coordinator import ShellyConfigEntry as ShellyConfigEntry, async_reconnect_soon as async_reconnect_soon
 from .utils import get_block_device_sleep_period as get_block_device_sleep_period, get_coap_context as get_coap_context, get_device_entry_gen as get_device_entry_gen, get_http_port as get_http_port, get_info_auth as get_info_auth, get_info_gen as get_info_gen, get_model_name as get_model_name, get_rpc_device_wakeup_period as get_rpc_device_wakeup_period, get_ws_context as get_ws_context, mac_address_from_name as mac_address_from_name
 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
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_MAC as CONF_MAC, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
+from homeassistant.const import CONF_HOST as CONF_HOST, CONF_MAC as CONF_MAC, CONF_MODEL as CONF_MODEL, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.selector import SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig
@@ -35,10 +35,10 @@ class ShellyConfigFlow(ConfigFlow, domain=DOMAIN):
     async def _async_get_info(self, host: str, port: int) -> dict[str, Any]: ...
     @staticmethod
     @callback
-    def async_get_options_flow(config_entry: ConfigEntry) -> OptionsFlowHandler: ...
+    def async_get_options_flow(config_entry: ShellyConfigEntry) -> OptionsFlowHandler: ...
     @classmethod
     @callback
-    def async_supports_options_flow(cls, config_entry: ConfigEntry) -> bool: ...
+    def async_supports_options_flow(cls, config_entry: ShellyConfigEntry) -> bool: ...
 
 class OptionsFlowHandler(OptionsFlow):
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/shelly/const.pyi

@@ -79,5 +79,4 @@ VIRTUAL_COMPONENTS_MAP: Incomplete
 VIRTUAL_NUMBER_MODE_MAP: Incomplete
 API_WS_URL: str
 COMPONENT_ID_PATTERN: Incomplete
-BLU_TRV_TIMEOUT: int
 ROLE_TO_DEVICE_CLASS_MAP: Incomplete

homeassistant-stubs/components/shelly/coordinator.pyi

@@ -9,7 +9,7 @@ from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.bluetooth import async_remove_scanner as async_remove_scanner
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
-from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_HOST as CONF_HOST, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
+from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_HOST as CONF_HOST, CONF_MODEL as CONF_MODEL, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.debounce import Debouncer as Debouncer
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, format_mac as format_mac
@@ -24,6 +24,8 @@ class ShellyEntryData:
     rest: ShellyRestCoordinator | None = ...
     rpc: ShellyRpcCoordinator | None = ...
     rpc_poll: ShellyRpcPollingCoordinator | None = ...
+    rpc_script_events: dict[int, list[str]] | None = ...
+    rpc_supports_scripts: bool | None = ...
 type ShellyConfigEntry = ConfigEntry[ShellyEntryData]
 
 class ShellyCoordinatorBase[_DeviceT: BlockDevice | RpcDevice](DataUpdateCoordinator[None]):
@@ -82,6 +84,8 @@ class ShellyRpcCoordinator(ShellyCoordinatorBase[RpcDevice]):
     _input_event_listeners: list[Callable[[dict[str, Any]], None]]
     _connect_task: asyncio.Task | None
     def __init__(self, hass: HomeAssistant, entry: ShellyConfigEntry, device: RpcDevice) -> None: ...
+    @cached_property
+    def bluetooth_source(self) -> str: ...
     async def async_device_online(self, source: str) -> None: ...
     update_interval: Incomplete
     def update_sleep_period(self) -> bool: ...

homeassistant-stubs/components/shelly/diagnostics.pyi

@@ -3,9 +3,8 @@ from .utils import get_rpc_ws_url as get_rpc_ws_url
 from _typeshed import Incomplete
 from homeassistant.components.bluetooth import async_scanner_by_source as async_scanner_by_source
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
+from homeassistant.const import ATTR_MODEL as ATTR_MODEL, ATTR_NAME as ATTR_NAME, ATTR_SW_VERSION as ATTR_SW_VERSION, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.device_registry import format_mac as format_mac
 from typing import Any
 
 TO_REDACT: Incomplete

homeassistant-stubs/components/shelly/entity.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, LOGGER as LOGGER
+from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DOMAIN as DOMAIN, LOGGER as LOGGER
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_block_entity_name as get_block_entity_name, get_rpc_entity_name as get_rpc_entity_name, get_rpc_key_instances as get_rpc_key_instances
 from _typeshed import Incomplete
@@ -46,7 +46,6 @@ class RpcEntityDescription(EntityDescription):
     value: Callable[[Any, Any], Any] | None = ...
     available: Callable[[dict], bool] | None = ...
     removal_condition: Callable[[dict, dict, str], bool] | None = ...
-    extra_state_attributes: Callable[[dict, dict], dict | None] | None = ...
     use_polling_coordinator: bool = ...
     supported: Callable = ...
     unit: Callable[[dict], str | None] | None = ...
@@ -57,7 +56,6 @@ class RpcEntityDescription(EntityDescription):
 class RestEntityDescription(EntityDescription):
     name: str = ...
     value: Callable[[dict, Any], Any] | None = ...
-    extra_state_attributes: Callable[[dict], dict | None] | None = ...
 
 class ShellyBlockEntity(CoordinatorEntity[ShellyBlockCoordinator]):
     block: Incomplete

homeassistant-stubs/components/shelly/event.pyi

@@ -1,7 +1,7 @@
 from .const import BASIC_INPUTS_EVENTS_TYPES as BASIC_INPUTS_EVENTS_TYPES, RPC_INPUTS_EVENTS_TYPES as RPC_INPUTS_EVENTS_TYPES, SHIX3_1_INPUTS_EVENTS_TYPES as SHIX3_1_INPUTS_EVENTS_TYPES
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .entity import ShellyBlockEntity as ShellyBlockEntity
-from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_entity_name as get_rpc_entity_name, get_rpc_key_instances as get_rpc_key_instances, get_rpc_script_event_types as get_rpc_script_event_types, is_block_momentary_input as is_block_momentary_input, is_rpc_momentary_input as is_rpc_momentary_input
+from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_entity_name as get_rpc_entity_name, get_rpc_key_instances as get_rpc_key_instances, is_block_momentary_input as is_block_momentary_input, is_rpc_momentary_input as is_rpc_momentary_input
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from collections.abc import Callable as Callable

homeassistant-stubs/components/shelly/number.pyi

@@ -1,4 +1,4 @@
-from .const import BLU_TRV_TIMEOUT as BLU_TRV_TIMEOUT, CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, LOGGER as LOGGER, VIRTUAL_NUMBER_MODE_MAP as VIRTUAL_NUMBER_MODE_MAP
+from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DOMAIN as DOMAIN, LOGGER as LOGGER, VIRTUAL_NUMBER_MODE_MAP as VIRTUAL_NUMBER_MODE_MAP
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .entity import BlockEntityDescription as BlockEntityDescription, RpcEntityDescription as RpcEntityDescription, ShellyRpcAttributeEntity as ShellyRpcAttributeEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities, async_setup_entry_rpc as async_setup_entry_rpc
 from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, get_device_entry_gen as get_device_entry_gen, get_virtual_component_ids as get_virtual_component_ids

homeassistant-stubs/components/shelly/sensor.pyi

@@ -1,7 +1,7 @@
-from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, ROLE_TO_DEVICE_CLASS_MAP as ROLE_TO_DEVICE_CLASS_MAP, SHAIR_MAX_WORK_HOURS as SHAIR_MAX_WORK_HOURS
+from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, ROLE_TO_DEVICE_CLASS_MAP as ROLE_TO_DEVICE_CLASS_MAP
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .entity import BlockEntityDescription as BlockEntityDescription, RestEntityDescription as RestEntityDescription, RpcEntityDescription as RpcEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, ShellyRestAttributeEntity as ShellyRestAttributeEntity, ShellyRpcAttributeEntity as ShellyRpcAttributeEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, ShellySleepingRpcAttributeEntity as ShellySleepingRpcAttributeEntity, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities, async_setup_entry_rest as async_setup_entry_rest, async_setup_entry_rpc as async_setup_entry_rpc
-from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, get_device_entry_gen as get_device_entry_gen, get_device_uptime as get_device_uptime, get_virtual_component_ids as get_virtual_component_ids, is_rpc_wifi_stations_disabled as is_rpc_wifi_stations_disabled
+from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, get_device_entry_gen as get_device_entry_gen, get_device_uptime as get_device_uptime, get_shelly_air_lamp_life as get_shelly_air_lamp_life, get_virtual_component_ids as get_virtual_component_ids, is_rpc_wifi_stations_disabled as is_rpc_wifi_stations_disabled
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from collections.abc import Callable as Callable

homeassistant-stubs/components/shelly/switch.pyi

@@ -1,7 +1,6 @@
-from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, MOTION_MODELS as MOTION_MODELS
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
-from .entity import BlockEntityDescription as BlockEntityDescription, RpcEntityDescription as RpcEntityDescription, ShellyBlockEntity as ShellyBlockEntity, ShellyRpcAttributeEntity as ShellyRpcAttributeEntity, ShellyRpcEntity as ShellyRpcEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities, async_setup_entry_rpc as async_setup_entry_rpc
-from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, get_virtual_component_ids as get_virtual_component_ids, is_block_channel_type_light as is_block_channel_type_light, is_rpc_channel_type_light as is_rpc_channel_type_light, is_rpc_thermostat_internal_actuator as is_rpc_thermostat_internal_actuator, is_rpc_thermostat_mode as is_rpc_thermostat_mode
+from .entity import BlockEntityDescription as BlockEntityDescription, RpcEntityDescription as RpcEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, ShellyRpcAttributeEntity as ShellyRpcAttributeEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities, async_setup_entry_rpc as async_setup_entry_rpc
+from .utils import async_remove_orphaned_entities as async_remove_orphaned_entities, get_device_entry_gen as get_device_entry_gen, get_virtual_component_ids as get_virtual_component_ids, is_block_exclude_from_relay as is_block_exclude_from_relay, is_rpc_exclude_from_relay as is_rpc_exclude_from_relay
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from collections.abc import Callable as Callable
@@ -17,7 +16,8 @@ from typing import Any
 @dataclass(frozen=True, kw_only=True)
 class BlockSwitchDescription(BlockEntityDescription, SwitchEntityDescription): ...
 
-MOTION_SWITCH: Incomplete
+BLOCK_RELAY_SWITCHES: Incomplete
+BLOCK_SLEEPING_MOTION_SWITCH: Incomplete
 
 @dataclass(frozen=True, kw_only=True)
 class RpcSwitchDescription(RpcEntityDescription, SwitchEntityDescription):
@@ -26,6 +26,7 @@ class RpcSwitchDescription(RpcEntityDescription, SwitchEntityDescription):
     method_off: str
     method_params_fn: Callable[[int | None, bool], dict]
 
+RPC_RELAY_SWITCHES: Incomplete
 RPC_SWITCHES: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ShellyConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
@@ -45,9 +46,11 @@ class BlockSleepingMotionSwitch(ShellySleepingBlockAttributeEntity, RestoreEntit
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     async def async_added_to_hass(self) -> None: ...
 
-class BlockRelaySwitch(ShellyBlockEntity, SwitchEntity):
+class BlockRelaySwitch(ShellyBlockAttributeEntity, SwitchEntity):
+    entity_description: BlockSwitchDescription
     control_result: dict[str, Any] | None
-    def __init__(self, coordinator: ShellyBlockCoordinator, block: Block) -> None: ...
+    _attr_unique_id: str
+    def __init__(self, coordinator: ShellyBlockCoordinator, block: Block, attribute: str, description: BlockSwitchDescription) -> None: ...
     @property
     def is_on(self) -> bool: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
@@ -55,14 +58,6 @@ class BlockRelaySwitch(ShellyBlockEntity, SwitchEntity):
     @callback
     def _update_callback(self) -> None: ...
 
-class RpcRelaySwitch(ShellyRpcEntity, SwitchEntity):
-    _id: Incomplete
-    def __init__(self, coordinator: ShellyRpcCoordinator, id_: int) -> None: ...
-    @property
-    def is_on(self) -> bool: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-
 class RpcSwitch(ShellyRpcAttributeEntity, SwitchEntity):
     entity_description: RpcSwitchDescription
     _attr_has_entity_name: bool
@@ -70,3 +65,8 @@ class RpcSwitch(ShellyRpcAttributeEntity, SwitchEntity):
     def is_on(self) -> bool: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
+
+class RpcRelaySwitch(RpcSwitch):
+    _attr_has_entity_name: bool
+    _attr_unique_id: str
+    def __init__(self, coordinator: ShellyRpcCoordinator, key: str, attribute: str, description: RpcEntityDescription) -> None: ...

homeassistant-stubs/components/shelly/update.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, OTA_BEGIN as OTA_BEGIN, OTA_ERROR as OTA_ERROR, OTA_PROGRESS as OTA_PROGRESS, OTA_SUCCESS as OTA_SUCCESS
+from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, DOMAIN as DOMAIN, OTA_BEGIN as OTA_BEGIN, OTA_ERROR as OTA_ERROR, OTA_PROGRESS as OTA_PROGRESS, OTA_SUCCESS as OTA_SUCCESS
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyConfigEntry as ShellyConfigEntry, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .entity import RestEntityDescription as RestEntityDescription, RpcEntityDescription as RpcEntityDescription, ShellyRestAttributeEntity as ShellyRestAttributeEntity, ShellyRpcAttributeEntity as ShellyRpcAttributeEntity, ShellySleepingRpcAttributeEntity as ShellySleepingRpcAttributeEntity, async_setup_entry_rest as async_setup_entry_rest, async_setup_entry_rpc as async_setup_entry_rpc
 from .utils import get_device_entry_gen as get_device_entry_gen, get_release_url as get_release_url