homeassistant-stubs 2025.4.0b9__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/components/lamarzocco/binary_sensor.pyi

@@ -1,21 +1,20 @@
 from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry
-from .entity import LaMarzoccScaleEntity as LaMarzoccScaleEntity, LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
+from .entity import LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from pylamarzocco.models import LaMarzoccoMachineConfig as LaMarzoccoMachineConfig
+from pylamarzocco import LaMarzoccoMachine as LaMarzoccoMachine
 
 PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoBinarySensorEntityDescription(LaMarzoccoEntityDescription, BinarySensorEntityDescription):
-    is_on_fn: Callable[[LaMarzoccoMachineConfig], bool | None]
+    is_on_fn: Callable[[LaMarzoccoMachine], bool | None]
 
 ENTITIES: tuple[LaMarzoccoBinarySensorEntityDescription, ...]
-SCALE_ENTITIES: tuple[LaMarzoccoBinarySensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
@@ -23,6 +22,3 @@ class LaMarzoccoBinarySensorEntity(LaMarzoccoEntity, BinarySensorEntity):
     entity_description: LaMarzoccoBinarySensorEntityDescription
     @property
     def is_on(self) -> bool | None: ...
-
-class LaMarzoccoScaleBinarySensorEntity(LaMarzoccoBinarySensorEntity, LaMarzoccScaleEntity):
-    entity_description: LaMarzoccoBinarySensorEntityDescription

homeassistant-stubs/components/lamarzocco/calendar.pyi

@@ -6,19 +6,18 @@ from datetime import datetime
 from homeassistant.components.calendar import CalendarEntity as CalendarEntity, CalendarEvent as CalendarEvent
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from pylamarzocco.models import LaMarzoccoWakeUpSleepEntry as LaMarzoccoWakeUpSleepEntry
 
 PARALLEL_UPDATES: int
 CALENDAR_KEY: str
-DAY_OF_WEEK: Incomplete
+WEEKDAY_TO_ENUM: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class LaMarzoccoCalendarEntity(LaMarzoccoBaseEntity, CalendarEntity):
     _attr_translation_key = CALENDAR_KEY
-    wake_up_sleep_entry: Incomplete
+    _identifier: Incomplete
     _attr_translation_placeholders: Incomplete
-    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, key: str, wake_up_sleep_entry: LaMarzoccoWakeUpSleepEntry) -> None: ...
+    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, key: str, identifier: str) -> None: ...
     @property
     def event(self) -> CalendarEvent | None: ...
     async def async_get_events(self, hass: HomeAssistant, start_date: datetime, end_date: datetime) -> list[CalendarEvent]: ...

homeassistant-stubs/components/lamarzocco/config_flow.pyi

@@ -1,26 +1,29 @@
-from .const import CONF_USE_BLUETOOTH as CONF_USE_BLUETOOTH, DOMAIN as DOMAIN
+from . import create_client_session as create_client_session
+from .const import CONF_INSTALLATION_KEY as CONF_INSTALLATION_KEY, CONF_OFFLINE_MODE as CONF_OFFLINE_MODE, CONF_USE_BLUETOOTH as CONF_USE_BLUETOOTH, DOMAIN as DOMAIN
 from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry
 from _typeshed import Incomplete
 from aiohttp import ClientSession as ClientSession
 from collections.abc import Mapping
 from homeassistant.components.bluetooth import BluetoothServiceInfo as BluetoothServiceInfo, async_discovered_service_info as async_discovered_service_info
-from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_REAUTH as SOURCE_REAUTH, SOURCE_RECONFIGURE as SOURCE_RECONFIGURE
-from homeassistant.const import CONF_ADDRESS as CONF_ADDRESS, CONF_HOST as CONF_HOST, CONF_MAC as CONF_MAC, CONF_MODEL as CONF_MODEL, CONF_NAME as CONF_NAME, CONF_PASSWORD as CONF_PASSWORD, CONF_TOKEN as CONF_TOKEN, CONF_USERNAME as CONF_USERNAME
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlowWithReload as OptionsFlowWithReload, SOURCE_REAUTH as SOURCE_REAUTH, SOURCE_RECONFIGURE as SOURCE_RECONFIGURE
+from homeassistant.const import CONF_ADDRESS as CONF_ADDRESS, CONF_MAC as CONF_MAC, CONF_NAME as CONF_NAME, CONF_PASSWORD as CONF_PASSWORD, CONF_TOKEN as CONF_TOKEN, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import callback as callback
-from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession
 from homeassistant.helpers.selector import SelectOptionDict as SelectOptionDict, SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TextSelector as TextSelector, TextSelectorConfig as TextSelectorConfig, TextSelectorType as TextSelectorType
 from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo as DhcpServiceInfo
-from pylamarzocco.models import LaMarzoccoDeviceInfo as LaMarzoccoDeviceInfo
+from pylamarzocco.models import Thing as Thing
+from pylamarzocco.util import InstallationKey as InstallationKey
 from typing import Any
 
 CONF_MACHINE: str
+BT_MODEL_PREFIXES: Incomplete
 _LOGGER: Incomplete
 
 class LmConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
     _client: ClientSession
+    _installation_key: InstallationKey
     _config: dict[str, Any]
-    _fleet: dict[str, LaMarzoccoDeviceInfo]
+    _things: dict[str, Thing]
     _discovered: dict[str, str]
     def __init__(self) -> None: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
@@ -35,5 +38,5 @@ class LmConfigFlow(ConfigFlow, domain=DOMAIN):
     @callback
     def async_get_options_flow(config_entry: LaMarzoccoConfigEntry) -> LmOptionsFlowHandler: ...
 
-class LmOptionsFlowHandler(OptionsFlow):
+class LmOptionsFlowHandler(OptionsFlowWithReload):
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/lamarzocco/const.pyi

@@ -2,3 +2,5 @@ from typing import Final
 
 DOMAIN: Final[str]
 CONF_USE_BLUETOOTH: Final[str]
+CONF_INSTALLATION_KEY: Final[str]
+CONF_OFFLINE_MODE: Final[str]

homeassistant-stubs/components/lamarzocco/coordinator.pyi

@@ -1,52 +1,70 @@
 import abc
-from .const import DOMAIN as DOMAIN
+from .const import CONF_OFFLINE_MODE as CONF_OFFLINE_MODE, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from abc import abstractmethod
-from collections.abc import Callable as Callable
+from asyncio import Task
+from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
+from datetime import timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from pylamarzocco.clients.local import LaMarzoccoLocalClient as LaMarzoccoLocalClient
-from pylamarzocco.devices.machine import LaMarzoccoMachine as LaMarzoccoMachine
+from pylamarzocco import LaMarzoccoMachine as LaMarzoccoMachine
+from typing import Any
 
 SCAN_INTERVAL: Incomplete
-FIRMWARE_UPDATE_INTERVAL: Incomplete
+SETTINGS_UPDATE_INTERVAL: Incomplete
+SCHEDULE_UPDATE_INTERVAL: Incomplete
 STATISTICS_UPDATE_INTERVAL: Incomplete
 _LOGGER: Incomplete
 
 @dataclass
 class LaMarzoccoRuntimeData:
     config_coordinator: LaMarzoccoConfigUpdateCoordinator
-    firmware_coordinator: LaMarzoccoFirmwareUpdateCoordinator
+    settings_coordinator: LaMarzoccoSettingsUpdateCoordinator
+    schedule_coordinator: LaMarzoccoScheduleUpdateCoordinator
     statistics_coordinator: LaMarzoccoStatisticsUpdateCoordinator
+    bluetooth_coordinator: LaMarzoccoBluetoothUpdateCoordinator | None = ...
 type LaMarzoccoConfigEntry = ConfigEntry[LaMarzoccoRuntimeData]
 
 class LaMarzoccoUpdateCoordinator(DataUpdateCoordinator[None], metaclass=abc.ABCMeta):
-    _default_update_interval = SCAN_INTERVAL
+    _default_update_interval: timedelta | None
+    _ignore_offline_mode: bool
     config_entry: LaMarzoccoConfigEntry
+    update_success: bool
     device: Incomplete
-    local_connection_configured: Incomplete
-    _local_client: Incomplete
-    new_device_callback: list[Callable]
-    def __init__(self, hass: HomeAssistant, entry: LaMarzoccoConfigEntry, device: LaMarzoccoMachine, local_client: LaMarzoccoLocalClient | None = None) -> None: ...
+    _websocket_task: Task | None
+    def __init__(self, hass: HomeAssistant, entry: LaMarzoccoConfigEntry, device: LaMarzoccoMachine) -> None: ...
+    @property
+    def websocket_terminated(self) -> bool: ...
+    async def __handle_internal_update(self, func: Callable[[], Coroutine[Any, Any, None]]) -> None: ...
+    async def _async_setup(self) -> None: ...
     async def _async_update_data(self) -> None: ...
+    async def _internal_async_setup(self) -> None: ...
     @abstractmethod
     async def _internal_async_update_data(self) -> None: ...
 
 class LaMarzoccoConfigUpdateCoordinator(LaMarzoccoUpdateCoordinator):
-    _scale_address: str | None
-    async def _async_connect_websocket(self) -> None: ...
+    async def _internal_async_setup(self) -> None: ...
+    _websocket_task: Incomplete
     async def _internal_async_update_data(self) -> None: ...
-    @callback
-    def _async_add_remove_scale(self) -> None: ...
+    async def connect_websocket(self) -> None: ...
 
-class LaMarzoccoFirmwareUpdateCoordinator(LaMarzoccoUpdateCoordinator):
-    _default_update_interval = FIRMWARE_UPDATE_INTERVAL
+class LaMarzoccoSettingsUpdateCoordinator(LaMarzoccoUpdateCoordinator):
+    _default_update_interval = SETTINGS_UPDATE_INTERVAL
+    async def _internal_async_update_data(self) -> None: ...
+
+class LaMarzoccoScheduleUpdateCoordinator(LaMarzoccoUpdateCoordinator):
+    _default_update_interval = SCHEDULE_UPDATE_INTERVAL
     async def _internal_async_update_data(self) -> None: ...
 
 class LaMarzoccoStatisticsUpdateCoordinator(LaMarzoccoUpdateCoordinator):
     _default_update_interval = STATISTICS_UPDATE_INTERVAL
     async def _internal_async_update_data(self) -> None: ...
+
+class LaMarzoccoBluetoothUpdateCoordinator(LaMarzoccoUpdateCoordinator):
+    _ignore_offline_mode: bool
+    async def _internal_async_setup(self) -> None: ...
+    async def _internal_async_update_data(self) -> None: ...

homeassistant-stubs/components/lamarzocco/diagnostics.pyi

@@ -1,16 +1,11 @@
+from .const import CONF_USE_BLUETOOTH as CONF_USE_BLUETOOTH
 from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
+from homeassistant.const import CONF_MAC as CONF_MAC, CONF_TOKEN as CONF_TOKEN
 from homeassistant.core import HomeAssistant as HomeAssistant
-from pylamarzocco.const import FirmwareType as FirmwareType
-from typing import Any, TypedDict
+from typing import Any
 
 TO_REDACT: Incomplete
 
-class DiagnosticsData(TypedDict):
-    model: str
-    config: dict[str, Any]
-    firmware: list[dict[FirmwareType, dict[str, Any]]]
-    statistics: dict[str, Any]
-
 async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: LaMarzoccoConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/lamarzocco/entity.pyi

@@ -1,5 +1,5 @@
 from .const import DOMAIN as DOMAIN
-from .coordinator import LaMarzoccoUpdateCoordinator as LaMarzoccoUpdateCoordinator
+from .coordinator import LaMarzoccoBluetoothUpdateCoordinator as LaMarzoccoBluetoothUpdateCoordinator, LaMarzoccoUpdateCoordinator as LaMarzoccoUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
@@ -7,25 +7,26 @@ from homeassistant.const import CONF_ADDRESS as CONF_ADDRESS, CONF_MAC as CONF_M
 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 import EntityDescription as EntityDescription
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from pylamarzocco.devices.machine import LaMarzoccoMachine as LaMarzoccoMachine
 
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoEntityDescription(EntityDescription):
-    available_fn: Callable[[LaMarzoccoMachine], bool] = ...
+    available_fn: Callable[[LaMarzoccoUpdateCoordinator], bool] = ...
     supported_fn: Callable[[LaMarzoccoUpdateCoordinator], bool] = ...
+    bt_offline_mode: bool = ...
 
 class LaMarzoccoBaseEntity(CoordinatorEntity[LaMarzoccoUpdateCoordinator]):
     _attr_has_entity_name: bool
+    _unavailable_when_machine_off: bool
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
     def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, key: str) -> None: ...
+    @property
+    def available(self) -> bool: ...
 
 class LaMarzoccoEntity(LaMarzoccoBaseEntity):
     entity_description: LaMarzoccoEntityDescription
     @property
     def available(self) -> bool: ...
-    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, entity_description: LaMarzoccoEntityDescription) -> None: ...
-
-class LaMarzoccScaleEntity(LaMarzoccoEntity):
-    _attr_device_info: Incomplete
-    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, entity_description: LaMarzoccoEntityDescription) -> None: ...
+    bluetooth_coordinator: Incomplete
+    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, entity_description: LaMarzoccoEntityDescription, bluetooth_coordinator: LaMarzoccoBluetoothUpdateCoordinator | None = None) -> None: ...
+    async def async_added_to_hass(self) -> None: ...

homeassistant-stubs/components/lamarzocco/number.pyi

@@ -1,54 +1,29 @@
 from .const import DOMAIN as DOMAIN
-from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator as LaMarzoccoUpdateCoordinator
-from .entity import LaMarzoccScaleEntity as LaMarzoccScaleEntity, LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
-from _typeshed import Incomplete
+from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry
+from .entity import LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.number import NumberDeviceClass as NumberDeviceClass, NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription
-from homeassistant.const import EntityCategory as EntityCategory, PRECISION_TENTHS as PRECISION_TENTHS, PRECISION_WHOLE as PRECISION_WHOLE, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
+from homeassistant.const import EntityCategory as EntityCategory, PRECISION_TENTHS as PRECISION_TENTHS, PRECISION_WHOLE as PRECISION_WHOLE, UnitOfMass as UnitOfMass, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from pylamarzocco.const import PhysicalKey
-from pylamarzocco.devices.machine import LaMarzoccoMachine as LaMarzoccoMachine
-from pylamarzocco.models import LaMarzoccoMachineConfig as LaMarzoccoMachineConfig
+from pylamarzocco import LaMarzoccoMachine as LaMarzoccoMachine
 from typing import Any
 
 PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoNumberEntityDescription(LaMarzoccoEntityDescription, NumberEntityDescription):
-    native_value_fn: Callable[[LaMarzoccoMachineConfig], float | int]
+    native_value_fn: Callable[[LaMarzoccoMachine], float | int]
     set_value_fn: Callable[[LaMarzoccoMachine, float | int], Coroutine[Any, Any, bool]]
 
-@dataclass(frozen=True, kw_only=True)
-class LaMarzoccoKeyNumberEntityDescription(LaMarzoccoEntityDescription, NumberEntityDescription):
-    native_value_fn: Callable[[LaMarzoccoMachineConfig, PhysicalKey], float | int | None]
-    set_value_fn: Callable[[LaMarzoccoMachine, float | int, PhysicalKey], Coroutine[Any, Any, bool]]
-
 ENTITIES: tuple[LaMarzoccoNumberEntityDescription, ...]
-KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...]
-SCALE_KEY_ENTITIES: tuple[LaMarzoccoKeyNumberEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class LaMarzoccoNumberEntity(LaMarzoccoEntity, NumberEntity):
     entity_description: LaMarzoccoNumberEntityDescription
     @property
-    def native_value(self) -> float: ...
-    async def async_set_native_value(self, value: float) -> None: ...
-
-class LaMarzoccoKeyNumberEntity(LaMarzoccoEntity, NumberEntity):
-    entity_description: LaMarzoccoKeyNumberEntityDescription
-    _attr_translation_key: Incomplete
-    _attr_translation_placeholders: Incomplete
-    _attr_unique_id: Incomplete
-    _attr_entity_registry_enabled_default: bool
-    pyhsical_key: Incomplete
-    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, description: LaMarzoccoKeyNumberEntityDescription, pyhsical_key: int) -> None: ...
-    @property
-    def native_value(self) -> float | None: ...
+    def native_value(self) -> float | int: ...
     async def async_set_native_value(self, value: float) -> None: ...
-
-class LaMarzoccoScaleTargetNumberEntity(LaMarzoccoKeyNumberEntity, LaMarzoccScaleEntity):
-    entity_description: LaMarzoccoKeyNumberEntityDescription

homeassistant-stubs/components/lamarzocco/select.pyi

@@ -1,6 +1,6 @@
 from .const import DOMAIN as DOMAIN
 from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry
-from .entity import LaMarzoccScaleEntity as LaMarzoccScaleEntity, LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
+from .entity import LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
@@ -9,8 +9,7 @@ from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from pylamarzocco.devices.machine import LaMarzoccoMachine as LaMarzoccoMachine
-from pylamarzocco.models import LaMarzoccoMachineConfig as LaMarzoccoMachineConfig
+from pylamarzocco.devices import LaMarzoccoMachine as LaMarzoccoMachine
 from typing import Any
 
 PARALLEL_UPDATES: int
@@ -20,14 +19,15 @@ PREBREW_MODE_HA_TO_LM: Incomplete
 PREBREW_MODE_LM_TO_HA: Incomplete
 STANDBY_MODE_HA_TO_LM: Incomplete
 STANDBY_MODE_LM_TO_HA: Incomplete
+DOSE_MODE_HA_TO_LM: Incomplete
+DOSE_MODE_LM_TO_HA: Incomplete
 
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoSelectEntityDescription(LaMarzoccoEntityDescription, SelectEntityDescription):
-    current_option_fn: Callable[[LaMarzoccoMachineConfig], str | None]
+    current_option_fn: Callable[[LaMarzoccoMachine], str | None]
     select_option_fn: Callable[[LaMarzoccoMachine, str], Coroutine[Any, Any, bool]]
 
 ENTITIES: tuple[LaMarzoccoSelectEntityDescription, ...]
-SCALE_ENTITIES: tuple[LaMarzoccoSelectEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
@@ -36,6 +36,3 @@ class LaMarzoccoSelectEntity(LaMarzoccoEntity, SelectEntity):
     @property
     def current_option(self) -> str | None: ...
     async def async_select_option(self, option: str) -> None: ...
-
-class LaMarzoccoScaleSelectEntity(LaMarzoccoSelectEntity, LaMarzoccScaleEntity):
-    entity_description: LaMarzoccoSelectEntityDescription

homeassistant-stubs/components/lamarzocco/sensor.pyi

@@ -1,45 +1,33 @@
-from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry, LaMarzoccoUpdateCoordinator as LaMarzoccoUpdateCoordinator
-from .entity import LaMarzoccScaleEntity as LaMarzoccScaleEntity, LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
-from _typeshed import Incomplete
+from .coordinator import LaMarzoccoConfigEntry as LaMarzoccoConfigEntry
+from .entity import LaMarzoccoEntity as LaMarzoccoEntity, LaMarzoccoEntityDescription as LaMarzoccoEntityDescription
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
+from datetime import datetime
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
+from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from pylamarzocco.const import PhysicalKey
-from pylamarzocco.devices.machine import LaMarzoccoMachine as LaMarzoccoMachine
+from homeassistant.helpers.typing import StateType as StateType
+from pylamarzocco.const import WidgetType
+from pylamarzocco.models import BaseWidgetOutput as BaseWidgetOutput
 
 PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoSensorEntityDescription(LaMarzoccoEntityDescription, SensorEntityDescription):
-    value_fn: Callable[[LaMarzoccoMachine], float | int]
-
-@dataclass(frozen=True, kw_only=True)
-class LaMarzoccoKeySensorEntityDescription(LaMarzoccoEntityDescription, SensorEntityDescription):
-    value_fn: Callable[[LaMarzoccoMachine, PhysicalKey], int | None]
+    value_fn: Callable[[dict[WidgetType, BaseWidgetOutput]], StateType | datetime | None]
 
 ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...]
 STATISTIC_ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...]
-KEY_STATISTIC_ENTITIES: tuple[LaMarzoccoKeySensorEntityDescription, ...]
-SCALE_ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class LaMarzoccoSensorEntity(LaMarzoccoEntity, SensorEntity):
     entity_description: LaMarzoccoSensorEntityDescription
     @property
-    def native_value(self) -> int | float | None: ...
+    def native_value(self) -> StateType | datetime | None: ...
 
-class LaMarzoccoKeySensorEntity(LaMarzoccoEntity, SensorEntity):
-    entity_description: LaMarzoccoKeySensorEntityDescription
-    key: Incomplete
-    _attr_translation_placeholders: Incomplete
-    _attr_unique_id: Incomplete
-    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, description: LaMarzoccoKeySensorEntityDescription, key: int) -> None: ...
+class LaMarzoccoStatisticSensorEntity(LaMarzoccoSensorEntity):
+    _unavailable_when_machine_off: bool
     @property
-    def native_value(self) -> int | None: ...
-
-class LaMarzoccoScaleSensorEntity(LaMarzoccoSensorEntity, LaMarzoccScaleEntity):
-    entity_description: LaMarzoccoSensorEntityDescription
+    def native_value(self) -> StateType | datetime | None: ...

homeassistant-stubs/components/lamarzocco/switch.pyi

@@ -9,8 +9,8 @@ from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from pylamarzocco.devices.machine import LaMarzoccoMachine as LaMarzoccoMachine
-from pylamarzocco.models import LaMarzoccoMachineConfig as LaMarzoccoMachineConfig
+from pylamarzocco import LaMarzoccoMachine as LaMarzoccoMachine
+from pylamarzocco.models import WakeUpScheduleSettings as WakeUpScheduleSettings
 from typing import Any
 
 PARALLEL_UPDATES: int
@@ -18,9 +18,10 @@ PARALLEL_UPDATES: int
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoSwitchEntityDescription(LaMarzoccoEntityDescription, SwitchEntityDescription):
     control_fn: Callable[[LaMarzoccoMachine, bool], Coroutine[Any, Any, bool]]
-    is_on_fn: Callable[[LaMarzoccoMachineConfig], bool]
+    is_on_fn: Callable[[LaMarzoccoMachine], bool]
 
 ENTITIES: tuple[LaMarzoccoSwitchEntityDescription, ...]
+MAIN_SWITCH_ENTITY: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: LaMarzoccoConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
@@ -31,13 +32,18 @@ class LaMarzoccoSwitchEntity(LaMarzoccoEntity, SwitchEntity):
     @property
     def is_on(self) -> bool: ...
 
+class LaMarzoccoMainSwitchEntity(LaMarzoccoSwitchEntity):
+    @property
+    def entity_picture(self) -> str | None: ...
+
 class LaMarzoccoAutoOnOffSwitchEntity(LaMarzoccoBaseEntity, SwitchEntity):
     coordinator: LaMarzoccoUpdateCoordinator
     _attr_translation_key: str
+    _schedule_entry: Incomplete
     _identifier: Incomplete
     _attr_translation_placeholders: Incomplete
-    entity_category: Incomplete
-    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, identifier: str) -> None: ...
+    _attr_entity_category: Incomplete
+    def __init__(self, coordinator: LaMarzoccoUpdateCoordinator, schedule_entry: WakeUpScheduleSettings) -> None: ...
     async def _async_enable(self, state: bool) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/lamarzocco/update.pyi

@@ -12,6 +12,7 @@ from pylamarzocco.const import FirmwareType
 from typing import Any
 
 PARALLEL_UPDATES: int
+MAX_UPDATE_WAIT: int
 
 @dataclass(frozen=True, kw_only=True)
 class LaMarzoccoUpdateEntityDescription(LaMarzoccoEntityDescription, UpdateEntityDescription):
@@ -25,10 +26,12 @@ class LaMarzoccoUpdateEntity(LaMarzoccoEntity, UpdateEntity):
     entity_description: LaMarzoccoUpdateEntityDescription
     _attr_supported_features: Incomplete
     @property
-    def installed_version(self) -> str | None: ...
+    def installed_version(self) -> str: ...
     @property
     def latest_version(self) -> str: ...
     @property
     def release_url(self) -> str | None: ...
+    def release_notes(self) -> str | None: ...
     _attr_in_progress: bool
+    _attr_update_percentage: Incomplete
     async def async_install(self, version: str | None, backup: bool, **kwargs: Any) -> None: ...

homeassistant-stubs/components/lametric/__init__.pyi

@@ -1,8 +1,7 @@
 from .const import DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
-from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
+from .coordinator import LaMetricConfigEntry as LaMetricConfigEntry, LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
 from .services import async_setup_services as async_setup_services
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_NAME as CONF_NAME, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers import discovery as discovery
@@ -11,5 +10,5 @@ from homeassistant.helpers.typing import ConfigType as ConfigType
 CONFIG_SCHEMA: Incomplete
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: LaMetricConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: LaMetricConfigEntry) -> bool: ...

homeassistant-stubs/components/lametric/button.pyi

@@ -1,5 +1,4 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
+from .coordinator import LaMetricConfigEntry as LaMetricConfigEntry, LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
 from .entity import LaMetricEntity as LaMetricEntity
 from .helpers import lametric_exception_handler as lametric_exception_handler
 from _typeshed import Incomplete
@@ -7,7 +6,6 @@ from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from demetriek import LaMetricDevice as LaMetricDevice
 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 AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -19,7 +17,7 @@ class LaMetricButtonEntityDescription(ButtonEntityDescription):
 
 BUTTONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: LaMetricConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class LaMetricButtonEntity(LaMetricEntity, ButtonEntity):
     entity_description: LaMetricButtonEntityDescription

homeassistant-stubs/components/lametric/config_flow.pyi

@@ -14,6 +14,8 @@ from homeassistant.helpers.service_info.ssdp import ATTR_UPNP_FRIENDLY_NAME as A
 from homeassistant.util.network import is_link_local as is_link_local
 from typing import Any
 
+DEVICES_URL: str
+
 class LaMetricFlowHandler(AbstractOAuth2FlowHandler, domain=DOMAIN):
     DOMAIN = DOMAIN
     VERSION: int

homeassistant-stubs/components/lametric/coordinator.pyi

@@ -8,8 +8,9 @@ from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFai
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 
+type LaMetricConfigEntry = ConfigEntry[LaMetricDataUpdateCoordinator]
 class LaMetricDataUpdateCoordinator(DataUpdateCoordinator[Device]):
-    config_entry: ConfigEntry
+    config_entry: LaMetricConfigEntry
     lametric: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+    def __init__(self, hass: HomeAssistant, entry: LaMetricConfigEntry) -> None: ...
     async def _async_update_data(self) -> Device: ...

homeassistant-stubs/components/lametric/diagnostics.pyi

@@ -1,11 +1,9 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
+from .coordinator import LaMetricConfigEntry as LaMetricConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
 
 TO_REDACT: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: LaMetricConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/lametric/entity.pyi

@@ -1,7 +1,7 @@
 from .const import DOMAIN as DOMAIN
 from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
 from _typeshed import Incomplete
-from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceInfo as DeviceInfo, format_mac as format_mac
+from homeassistant.helpers.device_registry import CONNECTION_BLUETOOTH as CONNECTION_BLUETOOTH, CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, DeviceInfo as DeviceInfo, format_mac as format_mac
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 
 class LaMetricEntity(CoordinatorEntity[LaMetricDataUpdateCoordinator]):

homeassistant-stubs/components/lametric/helpers.pyi

@@ -1,5 +1,5 @@
 from .const import DOMAIN as DOMAIN
-from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
+from .coordinator import LaMetricConfigEntry as LaMetricConfigEntry, LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
 from .entity import LaMetricEntity as LaMetricEntity
 from collections.abc import Callable as Callable, Coroutine
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback

homeassistant-stubs/components/lametric/notify.pyi

@@ -1,5 +1,5 @@
-from .const import CONF_CYCLES as CONF_CYCLES, CONF_ICON_TYPE as CONF_ICON_TYPE, CONF_PRIORITY as CONF_PRIORITY, CONF_SOUND as CONF_SOUND, DOMAIN as DOMAIN
-from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
+from .const import CONF_CYCLES as CONF_CYCLES, CONF_ICON_TYPE as CONF_ICON_TYPE, CONF_PRIORITY as CONF_PRIORITY, CONF_SOUND as CONF_SOUND
+from .coordinator import LaMetricConfigEntry as LaMetricConfigEntry
 from _typeshed import Incomplete
 from demetriek import LaMetricDevice as LaMetricDevice
 from homeassistant.components.notify import ATTR_DATA as ATTR_DATA, BaseNotificationService as BaseNotificationService

homeassistant-stubs/components/lametric/number.pyi

@@ -1,5 +1,4 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
+from .coordinator import LaMetricConfigEntry as LaMetricConfigEntry, LaMetricDataUpdateCoordinator as LaMetricDataUpdateCoordinator
 from .entity import LaMetricEntity as LaMetricEntity
 from .helpers import lametric_exception_handler as lametric_exception_handler
 from _typeshed import Incomplete
@@ -7,7 +6,6 @@ from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from demetriek import Device as Device, LaMetricDevice as LaMetricDevice, Range as Range
 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 AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -22,7 +20,7 @@ class LaMetricNumberEntityDescription(NumberEntityDescription):
 
 NUMBERS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: LaMetricConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class LaMetricNumberEntity(LaMetricEntity, NumberEntity):
     entity_description: LaMetricNumberEntityDescription