homeassistant-stubs 2025.7.3__py3-none-any.whl → 2025.8.0__py3-none-any.whl

homeassistant-stubs/components/zwave_js/fan.pyi

@@ -1,11 +1,11 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .discovery_data_template import FanValueMapping as FanValueMapping, FanValueMappingDataTemplate as FanValueMappingDataTemplate
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
 from .helpers import get_value_of_zwave_value as get_value_of_zwave_value
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.fan import FanEntity as FanEntity, FanEntityFeature as FanEntityFeature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -19,13 +19,13 @@ PARALLEL_UPDATES: int
 DEFAULT_SPEED_RANGE: Incomplete
 ATTR_FAN_STATE: str
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZwaveFan(ZWaveBaseEntity, FanEntity):
     _attr_supported_features: Incomplete
     _target_value: Incomplete
     _use_optimistic_state: bool
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     async def async_set_percentage(self, percentage: int) -> None: ...
     async def async_turn_on(self, percentage: int | None = None, preset_mode: str | None = None, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
@@ -38,7 +38,7 @@ class ZwaveFan(ZWaveBaseEntity, FanEntity):
 
 class ValueMappingZwaveFan(ZwaveFan):
     data_template: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     async def async_set_percentage(self, percentage: int) -> None: ...
     async def async_set_preset_mode(self, preset_mode: str) -> None: ...
     @property
@@ -66,7 +66,7 @@ class ZwaveThermostatFan(ZWaveBaseEntity, FanEntity):
     _fan_mode: ZwaveValue
     _fan_off: ZwaveValue | None
     _fan_state: ZwaveValue | None
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     async def async_turn_on(self, percentage: int | None = None, preset_mode: str | None = None, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     @property

homeassistant-stubs/components/zwave_js/helpers.pyi

@@ -1,9 +1,10 @@
 import logging
 import voluptuous as vol
-from .const import ATTR_COMMAND_CLASS as ATTR_COMMAND_CLASS, ATTR_ENDPOINT as ATTR_ENDPOINT, ATTR_PROPERTY as ATTR_PROPERTY, ATTR_PROPERTY_KEY as ATTR_PROPERTY_KEY, DATA_CLIENT as DATA_CLIENT, DATA_OLD_SERVER_LOG_LEVEL as DATA_OLD_SERVER_LOG_LEVEL, DOMAIN as DOMAIN, LIB_LOGGER as LIB_LOGGER, LOGGER as LOGGER
-from collections.abc import Callable as Callable
+from .const import ATTR_COMMAND_CLASS as ATTR_COMMAND_CLASS, ATTR_ENDPOINT as ATTR_ENDPOINT, ATTR_PROPERTY as ATTR_PROPERTY, ATTR_PROPERTY_KEY as ATTR_PROPERTY_KEY, DOMAIN as DOMAIN, LIB_LOGGER as LIB_LOGGER, LOGGER as LOGGER
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
+from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.const import ATTR_AREA_ID as ATTR_AREA_ID, ATTR_DEVICE_ID as ATTR_DEVICE_ID, ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_TYPE as CONF_TYPE
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -19,6 +20,7 @@ from zwave_js_server.model.node import Node as ZwaveNode
 from zwave_js_server.model.value import Value as ZwaveValue, ValueDataType as ValueDataType
 from zwave_js_server.version import VersionInfo as VersionInfo
 
+DRIVER_READY_EVENT_TIMEOUT: int
 SERVER_VERSION_TIMEOUT: int
 
 @dataclass
@@ -41,8 +43,8 @@ def get_value_id_from_unique_id(unique_id: str) -> str | None: ...
 def get_state_key_from_unique_id(unique_id: str) -> int | None: ...
 def get_value_of_zwave_value(value: ZwaveValue | None) -> Any | None: ...
 async def async_enable_statistics(driver: Driver) -> None: ...
-async def async_enable_server_logging_if_needed(hass: HomeAssistant, entry: ConfigEntry, driver: Driver) -> None: ...
-async def async_disable_server_logging_if_needed(hass: HomeAssistant, entry: ConfigEntry, driver: Driver) -> None: ...
+async def async_enable_server_logging_if_needed(hass: HomeAssistant, entry: ZwaveJSConfigEntry, driver: Driver) -> None: ...
+async def async_disable_server_logging_if_needed(hass: HomeAssistant, entry: ZwaveJSConfigEntry, driver: Driver) -> None: ...
 def get_valueless_base_unique_id(driver: Driver, node: ZwaveNode) -> str: ...
 def get_unique_id(driver: Driver, value_id: str) -> str: ...
 def get_device_id(driver: Driver, node: ZwaveNode) -> tuple[str, str]: ...
@@ -66,7 +68,9 @@ def check_type_schema_map(schema_map: dict[str, vol.Schema]) -> Callable[[Config
 def copy_available_params(input_dict: dict[str, Any], output_dict: dict[str, Any], params: list[str]) -> None: ...
 def get_value_state_schema(value: ZwaveValue) -> VolSchemaType | vol.Coerce | vol.In | None: ...
 def get_device_info(driver: Driver, node: ZwaveNode) -> DeviceInfo: ...
-def get_network_identifier_for_notification(hass: HomeAssistant, config_entry: ConfigEntry, controller: Controller) -> str: ...
+def get_network_identifier_for_notification(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, controller: Controller) -> str: ...
 async def async_get_version_info(hass: HomeAssistant, ws_address: str) -> VersionInfo: ...
+@callback
+def async_wait_for_driver_ready_event(config_entry: ZwaveJSConfigEntry, driver: Driver) -> Callable[[], Coroutine[Any, Any, None]]: ...
 
 class CannotConnect(HomeAssistantError): ...

homeassistant-stubs/components/zwave_js/humidifier.pyi

@@ -1,10 +1,10 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.humidifier import DEFAULT_MAX_HUMIDITY as DEFAULT_MAX_HUMIDITY, DEFAULT_MIN_HUMIDITY as DEFAULT_MIN_HUMIDITY, HumidifierDeviceClass as HumidifierDeviceClass, HumidifierEntity as HumidifierEntity, HumidifierEntityDescription as HumidifierEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -24,7 +24,7 @@ class ZwaveHumidifierEntityDescription(HumidifierEntityDescription):
 HUMIDIFIER_ENTITY_DESCRIPTION: Incomplete
 DEHUMIDIFIER_ENTITY_DESCRIPTION: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZWaveHumidifier(ZWaveBaseEntity, HumidifierEntity):
     entity_description: ZwaveHumidifierEntityDescription
@@ -32,7 +32,7 @@ class ZWaveHumidifier(ZWaveBaseEntity, HumidifierEntity):
     _setpoint: ZwaveValue | None
     _attr_name: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, description: ZwaveHumidifierEntityDescription) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, description: ZwaveHumidifierEntityDescription) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
     def _supports_inverse_mode(self) -> bool: ...

homeassistant-stubs/components/zwave_js/light.pyi

@@ -1,9 +1,9 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP_KELVIN as ATTR_COLOR_TEMP_KELVIN, ATTR_HS_COLOR as ATTR_HS_COLOR, ATTR_RGBW_COLOR as ATTR_RGBW_COLOR, ATTR_TRANSITION as ATTR_TRANSITION, ColorMode as ColorMode, LightEntity as LightEntity, LightEntityFeature as LightEntityFeature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -17,7 +17,7 @@ MULTI_COLOR_MAP: Incomplete
 MIN_MIREDS: int
 MAX_MIREDS: int
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 def byte_to_zwave_brightness(value: int) -> int: ...
 
 class ZwaveLight(ZWaveBaseEntity, LightEntity):
@@ -41,7 +41,7 @@ class ZwaveLight(ZWaveBaseEntity, LightEntity):
     supports_brightness_transition: Incomplete
     supports_color_transition: Incomplete
     _set_optimistic_state: bool
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @callback
     def on_value_update(self) -> None: ...
     @property
@@ -73,8 +73,18 @@ class ZwaveLight(ZWaveBaseEntity, LightEntity):
 class ZwaveColorOnOffLight(ZwaveLight):
     _last_on_color: dict[ColorComponent, int] | None
     _last_brightness: int | None
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def brightness(self) -> int | None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
+
+class ZWA2LEDColorLight(ZwaveColorOnOffLight):
+    _attr_has_entity_name: bool
+    _attr_name: str
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+
+class ZWA2LEDOnOffLight(ZwaveLight):
+    _attr_has_entity_name: bool
+    _attr_name: str
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...

homeassistant-stubs/components/zwave_js/lock.pyi

@@ -1,9 +1,9 @@
-from .const import ATTR_AUTO_RELOCK_TIME as ATTR_AUTO_RELOCK_TIME, ATTR_BLOCK_TO_BLOCK as ATTR_BLOCK_TO_BLOCK, ATTR_HOLD_AND_RELEASE_TIME as ATTR_HOLD_AND_RELEASE_TIME, ATTR_LOCK_TIMEOUT as ATTR_LOCK_TIMEOUT, ATTR_OPERATION_TYPE as ATTR_OPERATION_TYPE, ATTR_TWIST_ASSIST as ATTR_TWIST_ASSIST, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN, LOGGER as LOGGER, SERVICE_CLEAR_LOCK_USERCODE as SERVICE_CLEAR_LOCK_USERCODE, SERVICE_SET_LOCK_CONFIGURATION as SERVICE_SET_LOCK_CONFIGURATION, SERVICE_SET_LOCK_USERCODE as SERVICE_SET_LOCK_USERCODE
+from .const import ATTR_AUTO_RELOCK_TIME as ATTR_AUTO_RELOCK_TIME, ATTR_BLOCK_TO_BLOCK as ATTR_BLOCK_TO_BLOCK, ATTR_HOLD_AND_RELEASE_TIME as ATTR_HOLD_AND_RELEASE_TIME, ATTR_LOCK_TIMEOUT as ATTR_LOCK_TIMEOUT, ATTR_OPERATION_TYPE as ATTR_OPERATION_TYPE, ATTR_TWIST_ASSIST as ATTR_TWIST_ASSIST, DOMAIN as DOMAIN, LOGGER as LOGGER, SERVICE_CLEAR_LOCK_USERCODE as SERVICE_CLEAR_LOCK_USERCODE, SERVICE_SET_LOCK_CONFIGURATION as SERVICE_SET_LOCK_CONFIGURATION, SERVICE_SET_LOCK_USERCODE as SERVICE_SET_LOCK_USERCODE
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.lock import LockEntity as LockEntity, LockState as LockState
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import entity_platform as entity_platform
@@ -16,7 +16,7 @@ PARALLEL_UPDATES: int
 STATE_TO_ZWAVE_MAP: dict[int, dict[str, int | bool]]
 UNIT16_SCHEMA: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZWaveLock(ZWaveBaseEntity, LockEntity):
     @property

homeassistant-stubs/components/zwave_js/models.pyi

@@ -0,0 +1,12 @@
+from . import DriverEvents as DriverEvents
+from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from zwave_js_server.client import Client as ZwaveClient
+from zwave_js_server.const import LogLevel as LogLevel
+
+@dataclass
+class ZwaveJSData:
+    client: ZwaveClient
+    driver_events: DriverEvents
+    old_server_log_level: LogLevel | None = ...
+type ZwaveJSConfigEntry = ConfigEntry[ZwaveJSData]

homeassistant-stubs/components/zwave_js/number.pyi

@@ -1,10 +1,10 @@
-from .const import ATTR_RESERVED_VALUES as ATTR_RESERVED_VALUES, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import ATTR_RESERVED_VALUES as ATTR_RESERVED_VALUES, DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.components.number import NumberEntity as NumberEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -16,12 +16,12 @@ from zwave_js_server.model.value import Value as Value
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZwaveNumberEntity(ZWaveBaseEntity, NumberEntity):
     _target_value: Value | None
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def native_min_value(self) -> float: ...
     @property
@@ -35,7 +35,7 @@ class ZwaveNumberEntity(ZWaveBaseEntity, NumberEntity):
 class ZWaveConfigParameterNumberEntity(ZwaveNumberEntity):
     _attr_entity_category: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def extra_state_attributes(self) -> Mapping[str, Any] | None: ...
 
@@ -45,7 +45,7 @@ class ZwaveVolumeNumberEntity(ZWaveBaseEntity, NumberEntity):
     _attr_native_max_value: int
     _attr_native_step: float
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def native_value(self) -> float | None: ...
     async def async_set_native_value(self, value: float) -> None: ...

homeassistant-stubs/components/zwave_js/select.pyi

@@ -1,9 +1,9 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.select import SelectEntity as SelectEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -12,32 +12,32 @@ from zwave_js_server.model.driver import Driver as Driver
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZwaveSelectEntity(ZWaveBaseEntity, SelectEntity):
     _attr_entity_category: Incomplete
     _attr_name: Incomplete
     _attr_options: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def current_option(self) -> str | None: ...
     async def async_select_option(self, option: str) -> None: ...
 
 class ZWaveDoorLockSelectEntity(ZwaveSelectEntity):
     _target_value: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     async def async_select_option(self, option: str) -> None: ...
 
 class ZWaveConfigParameterSelectEntity(ZwaveSelectEntity):
     _attr_entity_category: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
 
 class ZwaveDefaultToneSelectEntity(ZWaveBaseEntity, SelectEntity):
     _attr_entity_category: Incomplete
     _tones_value: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def options(self) -> list[str]: ...
     @property
@@ -48,7 +48,7 @@ class ZwaveMultilevelSwitchSelectEntity(ZWaveBaseEntity, SelectEntity):
     _target_value: Incomplete
     _lookup_map: Incomplete
     _attr_options: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def current_option(self) -> str | None: ...
     async def async_select_option(self, option: str) -> None: ...

homeassistant-stubs/components/zwave_js/sensor.pyi

@@ -1,15 +1,15 @@
 from .binary_sensor import is_valid_notification_binary_sensor as is_valid_notification_binary_sensor
-from .const import ATTR_METER_TYPE as ATTR_METER_TYPE, ATTR_METER_TYPE_NAME as ATTR_METER_TYPE_NAME, ATTR_VALUE as ATTR_VALUE, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN, ENTITY_DESC_KEY_BATTERY_LEVEL as ENTITY_DESC_KEY_BATTERY_LEVEL, ENTITY_DESC_KEY_BATTERY_LIST_STATE as ENTITY_DESC_KEY_BATTERY_LIST_STATE, ENTITY_DESC_KEY_BATTERY_MAXIMUM_CAPACITY as ENTITY_DESC_KEY_BATTERY_MAXIMUM_CAPACITY, ENTITY_DESC_KEY_BATTERY_TEMPERATURE as ENTITY_DESC_KEY_BATTERY_TEMPERATURE, ENTITY_DESC_KEY_CO as ENTITY_DESC_KEY_CO, ENTITY_DESC_KEY_CO2 as ENTITY_DESC_KEY_CO2, ENTITY_DESC_KEY_CURRENT as ENTITY_DESC_KEY_CURRENT, ENTITY_DESC_KEY_ENERGY_MEASUREMENT as ENTITY_DESC_KEY_ENERGY_MEASUREMENT, ENTITY_DESC_KEY_ENERGY_PRODUCTION_POWER as ENTITY_DESC_KEY_ENERGY_PRODUCTION_POWER, ENTITY_DESC_KEY_ENERGY_PRODUCTION_TIME as ENTITY_DESC_KEY_ENERGY_PRODUCTION_TIME, ENTITY_DESC_KEY_ENERGY_PRODUCTION_TODAY as ENTITY_DESC_KEY_ENERGY_PRODUCTION_TODAY, ENTITY_DESC_KEY_ENERGY_PRODUCTION_TOTAL as ENTITY_DESC_KEY_ENERGY_PRODUCTION_TOTAL, ENTITY_DESC_KEY_ENERGY_TOTAL_INCREASING as ENTITY_DESC_KEY_ENERGY_TOTAL_INCREASING, ENTITY_DESC_KEY_HUMIDITY as ENTITY_DESC_KEY_HUMIDITY, ENTITY_DESC_KEY_ILLUMINANCE as ENTITY_DESC_KEY_ILLUMINANCE, ENTITY_DESC_KEY_MEASUREMENT as ENTITY_DESC_KEY_MEASUREMENT, ENTITY_DESC_KEY_POWER as ENTITY_DESC_KEY_POWER, ENTITY_DESC_KEY_POWER_FACTOR as ENTITY_DESC_KEY_POWER_FACTOR, ENTITY_DESC_KEY_PRESSURE as ENTITY_DESC_KEY_PRESSURE, ENTITY_DESC_KEY_SIGNAL_STRENGTH as ENTITY_DESC_KEY_SIGNAL_STRENGTH, ENTITY_DESC_KEY_TARGET_TEMPERATURE as ENTITY_DESC_KEY_TARGET_TEMPERATURE, ENTITY_DESC_KEY_TEMPERATURE as ENTITY_DESC_KEY_TEMPERATURE, ENTITY_DESC_KEY_TOTAL_INCREASING as ENTITY_DESC_KEY_TOTAL_INCREASING, ENTITY_DESC_KEY_UV_INDEX as ENTITY_DESC_KEY_UV_INDEX, ENTITY_DESC_KEY_VOLTAGE as ENTITY_DESC_KEY_VOLTAGE, LOGGER as LOGGER, SERVICE_RESET_METER as SERVICE_RESET_METER
+from .const import ATTR_METER_TYPE as ATTR_METER_TYPE, ATTR_METER_TYPE_NAME as ATTR_METER_TYPE_NAME, ATTR_VALUE as ATTR_VALUE, DOMAIN as DOMAIN, ENTITY_DESC_KEY_BATTERY_LEVEL as ENTITY_DESC_KEY_BATTERY_LEVEL, ENTITY_DESC_KEY_BATTERY_LIST_STATE as ENTITY_DESC_KEY_BATTERY_LIST_STATE, ENTITY_DESC_KEY_BATTERY_MAXIMUM_CAPACITY as ENTITY_DESC_KEY_BATTERY_MAXIMUM_CAPACITY, ENTITY_DESC_KEY_BATTERY_TEMPERATURE as ENTITY_DESC_KEY_BATTERY_TEMPERATURE, ENTITY_DESC_KEY_CO as ENTITY_DESC_KEY_CO, ENTITY_DESC_KEY_CO2 as ENTITY_DESC_KEY_CO2, ENTITY_DESC_KEY_CURRENT as ENTITY_DESC_KEY_CURRENT, ENTITY_DESC_KEY_ENERGY_MEASUREMENT as ENTITY_DESC_KEY_ENERGY_MEASUREMENT, ENTITY_DESC_KEY_ENERGY_PRODUCTION_POWER as ENTITY_DESC_KEY_ENERGY_PRODUCTION_POWER, ENTITY_DESC_KEY_ENERGY_PRODUCTION_TIME as ENTITY_DESC_KEY_ENERGY_PRODUCTION_TIME, ENTITY_DESC_KEY_ENERGY_PRODUCTION_TODAY as ENTITY_DESC_KEY_ENERGY_PRODUCTION_TODAY, ENTITY_DESC_KEY_ENERGY_PRODUCTION_TOTAL as ENTITY_DESC_KEY_ENERGY_PRODUCTION_TOTAL, ENTITY_DESC_KEY_ENERGY_TOTAL_INCREASING as ENTITY_DESC_KEY_ENERGY_TOTAL_INCREASING, ENTITY_DESC_KEY_HUMIDITY as ENTITY_DESC_KEY_HUMIDITY, ENTITY_DESC_KEY_ILLUMINANCE as ENTITY_DESC_KEY_ILLUMINANCE, ENTITY_DESC_KEY_MEASUREMENT as ENTITY_DESC_KEY_MEASUREMENT, ENTITY_DESC_KEY_POWER as ENTITY_DESC_KEY_POWER, ENTITY_DESC_KEY_POWER_FACTOR as ENTITY_DESC_KEY_POWER_FACTOR, ENTITY_DESC_KEY_PRESSURE as ENTITY_DESC_KEY_PRESSURE, ENTITY_DESC_KEY_SIGNAL_STRENGTH as ENTITY_DESC_KEY_SIGNAL_STRENGTH, ENTITY_DESC_KEY_TARGET_TEMPERATURE as ENTITY_DESC_KEY_TARGET_TEMPERATURE, ENTITY_DESC_KEY_TEMPERATURE as ENTITY_DESC_KEY_TEMPERATURE, ENTITY_DESC_KEY_TOTAL_INCREASING as ENTITY_DESC_KEY_TOTAL_INCREASING, ENTITY_DESC_KEY_UV_INDEX as ENTITY_DESC_KEY_UV_INDEX, ENTITY_DESC_KEY_VOLTAGE as ENTITY_DESC_KEY_VOLTAGE, LOGGER as LOGGER, SERVICE_RESET_METER as SERVICE_RESET_METER
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .discovery_data_template import NumericSensorDataTemplate as NumericSensorDataTemplate, NumericSensorDataTemplateData as NumericSensorDataTemplateData
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
 from .helpers import get_device_info as get_device_info, get_valueless_base_unique_id as get_valueless_base_unique_id
 from .migrate import async_migrate_statistics_sensors as async_migrate_statistics_sensors
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Mapping
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, EntityCategory as EntityCategory, LIGHT_LUX as LIGHT_LUX, PERCENTAGE as PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UV_INDEX as UV_INDEX, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfPower as UnitOfPower, UnitOfPressure as UnitOfPressure, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -41,14 +41,14 @@ ENTITY_DESCRIPTION_NODE_STATISTICS_LIST: Incomplete
 NODE_STATISTICS_KEY_MAP: dict[str, str]
 
 def get_entity_description(data: NumericSensorDataTemplateData) -> SensorEntityDescription: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZwaveSensor(ZWaveBaseEntity, SensorEntity):
     entity_description: Incomplete
     _attr_native_unit_of_measurement: Incomplete
     _attr_force_update: bool
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
     @property
     def native_value(self) -> StateType: ...
     @property
@@ -56,7 +56,7 @@ class ZwaveSensor(ZWaveBaseEntity, SensorEntity):
 
 class ZWaveNumericSensor(ZwaveSensor):
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
     entity_description: Incomplete
     _attr_native_unit_of_measurement: Incomplete
     @callback
@@ -73,14 +73,14 @@ class ZWaveListSensor(ZwaveSensor):
     _attr_name: Incomplete
     _attr_device_class: Incomplete
     _attr_options: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
     @property
     def extra_state_attributes(self) -> dict[str, str] | None: ...
 
 class ZWaveConfigParameterSensor(ZWaveListSensor):
     _attr_entity_category: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, entity_description: SensorEntityDescription, unit_of_measurement: str | None = None) -> None: ...
     @property
     def extra_state_attributes(self) -> dict[str, str] | None: ...
 
@@ -94,7 +94,7 @@ class ZWaveNodeStatusSensor(SensorEntity):
     _base_unique_id: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, node: ZwaveNode) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, node: ZwaveNode) -> None: ...
     async def async_poll_value(self, _: bool) -> None: ...
     _attr_native_value: Incomplete
     @callback
@@ -111,7 +111,7 @@ class ZWaveControllerStatusSensor(SensorEntity):
     _base_unique_id: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver) -> None: ...
     async def async_poll_value(self, _: bool) -> None: ...
     _attr_native_value: Incomplete
     @callback
@@ -128,7 +128,7 @@ class ZWaveStatisticsSensor(SensorEntity):
     _base_unique_id: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, statistics_src: ZwaveNode | Controller, description: ZWaveJSStatisticsSensorEntityDescription) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, statistics_src: ZwaveNode | Controller, description: ZWaveJSStatisticsSensorEntityDescription) -> None: ...
     async def async_poll_value(self, _: bool) -> None: ...
     _attr_native_value: Incomplete
     @callback

homeassistant-stubs/components/zwave_js/siren.pyi

@@ -1,9 +1,9 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.siren import ATTR_TONE as ATTR_TONE, ATTR_VOLUME_LEVEL as ATTR_VOLUME_LEVEL, SirenEntity as SirenEntity, SirenEntityFeature as SirenEntityFeature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -12,13 +12,13 @@ from zwave_js_server.model.driver import Driver as Driver
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZwaveSirenEntity(ZWaveBaseEntity, SirenEntity):
     _attr_available_tones: Incomplete
     _attr_supported_features: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/zwave_js/switch.pyi

@@ -1,9 +1,9 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.switch import SwitchEntity as SwitchEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -13,11 +13,11 @@ from zwave_js_server.model.driver import Driver as Driver
 
 PARALLEL_UPDATES: int
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ZWaveSwitch(ZWaveBaseEntity, SwitchEntity):
     _target_value: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
@@ -26,12 +26,12 @@ class ZWaveSwitch(ZWaveBaseEntity, SwitchEntity):
 class ZWaveIndicatorSwitch(ZWaveSwitch):
     _target_value: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
 
 class ZWaveBarrierEventSignalingSwitch(ZWaveBaseEntity, SwitchEntity):
     _state: bool | None
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     @callback
     def on_value_update(self) -> None: ...
     @property
@@ -44,6 +44,6 @@ class ZWaveBarrierEventSignalingSwitch(ZWaveBaseEntity, SwitchEntity):
 class ZWaveConfigParameterSwitch(ZWaveSwitch):
     _attr_entity_category: Incomplete
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/zwave_js/triggers/event.pyi

@@ -1,4 +1,4 @@
-from ..const import ATTR_CONFIG_ENTRY_ID as ATTR_CONFIG_ENTRY_ID, ATTR_EVENT as ATTR_EVENT, ATTR_EVENT_DATA as ATTR_EVENT_DATA, ATTR_EVENT_SOURCE as ATTR_EVENT_SOURCE, ATTR_NODE_ID as ATTR_NODE_ID, ATTR_PARTIAL_DICT_MATCH as ATTR_PARTIAL_DICT_MATCH, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from ..const import ATTR_CONFIG_ENTRY_ID as ATTR_CONFIG_ENTRY_ID, ATTR_EVENT as ATTR_EVENT, ATTR_EVENT_DATA as ATTR_EVENT_DATA, ATTR_EVENT_SOURCE as ATTR_EVENT_SOURCE, ATTR_NODE_ID as ATTR_NODE_ID, ATTR_PARTIAL_DICT_MATCH as ATTR_PARTIAL_DICT_MATCH, DOMAIN as DOMAIN
 from ..helpers import async_get_nodes_from_targets as async_get_nodes_from_targets, get_device_id as get_device_id, get_home_and_node_id_from_device_entry as get_home_and_node_id_from_device_entry
 from .trigger_helpers import async_bypass_dynamic_config_validation as async_bypass_dynamic_config_validation
 from _typeshed import Incomplete
@@ -8,7 +8,6 @@ from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HassJob as HassJo
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.trigger import Trigger as Trigger, TriggerActionType as TriggerActionType, TriggerInfo as TriggerInfo
 from homeassistant.helpers.typing import ConfigType as ConfigType
-from zwave_js_server.client import Client as Client
 from zwave_js_server.model.driver import Driver as Driver
 
 PLATFORM_TYPE: Incomplete

homeassistant-stubs/components/zwave_js/triggers/trigger_helpers.pyi

@@ -1,4 +1,4 @@
-from ..const import ATTR_CONFIG_ENTRY_ID as ATTR_CONFIG_ENTRY_ID, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from ..const import ATTR_CONFIG_ENTRY_ID as ATTR_CONFIG_ENTRY_ID, DOMAIN as DOMAIN
 from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, ATTR_ENTITY_ID as ATTR_ENTITY_ID
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback

homeassistant-stubs/components/zwave_js/update.pyi

@@ -1,11 +1,11 @@
-from .const import API_KEY_FIRMWARE_UPDATE_SERVICE as API_KEY_FIRMWARE_UPDATE_SERVICE, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN, LOGGER as LOGGER
+from .const import API_KEY_FIRMWARE_UPDATE_SERVICE as API_KEY_FIRMWARE_UPDATE_SERVICE, DOMAIN as DOMAIN, LOGGER as LOGGER
 from .helpers import get_device_info as get_device_info, get_valueless_base_unique_id as get_valueless_base_unique_id
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
-from collections.abc import Callable as Callable
+from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-from homeassistant.components.update import ATTR_LATEST_VERSION as ATTR_LATEST_VERSION, UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityFeature as UpdateEntityFeature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.components.update import ATTR_LATEST_VERSION as ATTR_LATEST_VERSION, UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityDescription as UpdateEntityDescription, UpdateEntityFeature as UpdateEntityFeature
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import CoreState as CoreState, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
@@ -15,55 +15,65 @@ from homeassistant.helpers.event import async_call_later as async_call_later
 from homeassistant.helpers.restore_state import ExtraStoredData as ExtraStoredData
 from typing import Any, Final
 from zwave_js_server.model.driver import Driver as Driver
+from zwave_js_server.model.firmware import FirmwareUpdateInfo, FirmwareUpdateProgress as FirmwareUpdateProgress, FirmwareUpdateResult as FirmwareUpdateResult
 from zwave_js_server.model.node import Node as ZwaveNode
-from zwave_js_server.model.node.firmware import NodeFirmwareUpdateInfo, NodeFirmwareUpdateProgress as NodeFirmwareUpdateProgress, NodeFirmwareUpdateResult as NodeFirmwareUpdateResult
 
 PARALLEL_UPDATES: int
 UPDATE_DELAY_STRING: str
 UPDATE_DELAY_INTERVAL: int
 ATTR_LATEST_VERSION_FIRMWARE: str
 
+@dataclass(frozen=True, kw_only=True)
+class ZWaveUpdateEntityDescription(UpdateEntityDescription):
+    install_method: Callable[[ZWaveFirmwareUpdateEntity, FirmwareUpdateInfo], Awaitable[FirmwareUpdateResult]]
+    progress_method: Callable[[ZWaveFirmwareUpdateEntity], Callable[[], None]]
+    finished_method: Callable[[ZWaveFirmwareUpdateEntity], Callable[[], None]]
+
+CONTROLLER_UPDATE_ENTITY_DESCRIPTION: Incomplete
+NODE_UPDATE_ENTITY_DESCRIPTION: Incomplete
+
 @dataclass
-class ZWaveNodeFirmwareUpdateExtraStoredData(ExtraStoredData):
-    latest_version_firmware: NodeFirmwareUpdateInfo | None
+class ZWaveFirmwareUpdateExtraStoredData(ExtraStoredData):
+    latest_version_firmware: FirmwareUpdateInfo | None
     def as_dict(self) -> dict[str, Any]: ...
     @classmethod
-    def from_dict(cls, data: dict[str, Any]) -> ZWaveNodeFirmwareUpdateExtraStoredData: ...
+    def from_dict(cls, data: dict[str, Any]) -> ZWaveFirmwareUpdateExtraStoredData: ...
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class ZWaveNodeFirmwareUpdate(UpdateEntity):
+class ZWaveFirmwareUpdateEntity(UpdateEntity):
+    driver: Driver
+    entity_description: ZWaveUpdateEntityDescription
+    node: ZwaveNode
     _attr_entity_category: Incomplete
     _attr_device_class: Incomplete
     _attr_supported_features: Incomplete
     _attr_has_entity_name: bool
     _attr_should_poll: bool
-    driver: Incomplete
-    node: Incomplete
-    _latest_version_firmware: NodeFirmwareUpdateInfo | None
+    _latest_version_firmware: FirmwareUpdateInfo | None
     _status_unsub: Callable[[], None] | None
     _poll_unsub: Callable[[], None] | None
     _progress_unsub: Callable[[], None] | None
     _finished_unsub: Callable[[], None] | None
     _finished_event: Incomplete
-    _result: NodeFirmwareUpdateResult | None
+    _result: FirmwareUpdateResult | None
     _delay: Final[timedelta]
     _attr_name: str
     _base_unique_id: Incomplete
     _attr_unique_id: Incomplete
     _attr_installed_version: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, driver: Driver, node: ZwaveNode, delay: timedelta) -> None: ...
+    def __init__(self, driver: Driver, node: ZwaveNode, delay: timedelta, entity_description: ZWaveUpdateEntityDescription) -> None: ...
     @property
-    def extra_restore_state_data(self) -> ZWaveNodeFirmwareUpdateExtraStoredData: ...
+    def extra_restore_state_data(self) -> ZWaveFirmwareUpdateExtraStoredData: ...
     @callback
     def _update_on_status_change(self, _: dict[str, Any]) -> None: ...
     _attr_in_progress: bool
     _attr_update_percentage: Incomplete
     @callback
-    def _update_progress(self, event: dict[str, Any]) -> None: ...
+    def update_progress(self, event: dict[str, Any]) -> None: ...
     @callback
-    def _update_finished(self, event: dict[str, Any]) -> None: ...
+    def update_finished(self, event: dict[str, Any]) -> None: ...
     @callback
     def _unsub_firmware_events_and_reset_progress(self, write_state: bool = True) -> None: ...
     _attr_latest_version: Incomplete

homeassistant-stubs/config.pyi

@@ -18,8 +18,6 @@ from enum import StrEnum
 from typing import Any
 
 _LOGGER: Incomplete
-RE_YAML_ERROR: Incomplete
-RE_ASCII: Incomplete
 YAML_CONFIG_FILE: str
 VERSION_FILE: str
 CONFIG_DIR_NAME: str

homeassistant-stubs/config_entries.pyi

@@ -481,6 +481,9 @@ class OptionsFlowWithConfigEntry(OptionsFlow):
     @property
     def options(self) -> dict[str, Any]: ...
 
+class OptionsFlowWithReload(OptionsFlow):
+    automatic_reload: bool
+
 class EntityRegistryDisabledHandler:
     hass: Incomplete
     registry: er.EntityRegistry | None