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

homeassistant-stubs/components/zwave_js/api.pyi

@@ -1,13 +1,14 @@
 from .config_validation import BITMASK_SCHEMA as BITMASK_SCHEMA
-from .const import ATTR_COMMAND_CLASS as ATTR_COMMAND_CLASS, ATTR_ENDPOINT as ATTR_ENDPOINT, ATTR_METHOD_NAME as ATTR_METHOD_NAME, ATTR_PARAMETERS as ATTR_PARAMETERS, ATTR_WAIT_FOR_RESULT as ATTR_WAIT_FOR_RESULT, CONF_DATA_COLLECTION_OPTED_IN as CONF_DATA_COLLECTION_OPTED_IN, CONF_INSTALLER_MODE as CONF_INSTALLER_MODE, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN, DRIVER_READY_TIMEOUT as DRIVER_READY_TIMEOUT, EVENT_DEVICE_ADDED_TO_REGISTRY as EVENT_DEVICE_ADDED_TO_REGISTRY, LOGGER as LOGGER, USER_AGENT as USER_AGENT
+from .const import ATTR_COMMAND_CLASS as ATTR_COMMAND_CLASS, ATTR_ENDPOINT as ATTR_ENDPOINT, ATTR_METHOD_NAME as ATTR_METHOD_NAME, ATTR_PARAMETERS as ATTR_PARAMETERS, ATTR_WAIT_FOR_RESULT as ATTR_WAIT_FOR_RESULT, CONF_DATA_COLLECTION_OPTED_IN as CONF_DATA_COLLECTION_OPTED_IN, CONF_INSTALLER_MODE as CONF_INSTALLER_MODE, DOMAIN as DOMAIN, DRIVER_READY_TIMEOUT as DRIVER_READY_TIMEOUT, EVENT_DEVICE_ADDED_TO_REGISTRY as EVENT_DEVICE_ADDED_TO_REGISTRY, LOGGER as LOGGER, USER_AGENT as USER_AGENT
 from .helpers import CannotConnect as CannotConnect, async_enable_statistics as async_enable_statistics, async_get_node_from_device_id as async_get_node_from_device_id, async_get_provisioning_entry_from_device_id as async_get_provisioning_entry_from_device_id, async_get_version_info as async_get_version_info, get_device_id as get_device_id
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from aiohttp import web as web
 from collections.abc import Callable as Callable, Coroutine
 from homeassistant.components import websocket_api as websocket_api
 from homeassistant.components.http import HomeAssistantView as HomeAssistantView, KEY_HASS as KEY_HASS, require_admin as require_admin
 from homeassistant.components.websocket_api import ActiveConnection as ActiveConnection, ERR_INVALID_FORMAT as ERR_INVALID_FORMAT, ERR_NOT_FOUND as ERR_NOT_FOUND, ERR_NOT_SUPPORTED as ERR_NOT_SUPPORTED, ERR_UNKNOWN_ERROR as ERR_UNKNOWN_ERROR
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.const import CONF_URL as CONF_URL
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers import device_registry as dr
@@ -80,8 +81,8 @@ PLANNED_PROVISIONING_ENTRY_SCHEMA: Incomplete
 QR_PROVISIONING_INFORMATION_SCHEMA: Incomplete
 QR_CODE_STRING_SCHEMA: Incomplete
 
-async def _async_get_entry(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry_id: str) -> tuple[ConfigEntry, Client, Driver] | tuple[None, None, None]: ...
-def async_get_entry(orig_func: Callable[[HomeAssistant, ActiveConnection, dict[str, Any], ConfigEntry, Client, Driver], Coroutine[Any, Any, None]]) -> Callable[[HomeAssistant, ActiveConnection, dict[str, Any]], Coroutine[Any, Any, None]]: ...
+async def _async_get_entry(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry_id: str) -> tuple[ZwaveJSConfigEntry, Client, Driver] | tuple[None, None, None]: ...
+def async_get_entry(orig_func: Callable[[HomeAssistant, ActiveConnection, dict[str, Any], ZwaveJSConfigEntry, Client, Driver], Coroutine[Any, Any, None]]) -> Callable[[HomeAssistant, ActiveConnection, dict[str, Any]], Coroutine[Any, Any, None]]: ...
 async def _async_get_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict, device_id: str) -> Node | None: ...
 def async_get_node(orig_func: Callable[[HomeAssistant, ActiveConnection, dict[str, Any], Node], Coroutine[Any, Any, None]]) -> Callable[[HomeAssistant, ActiveConnection, dict[str, Any]], Coroutine[Any, Any, None]]: ...
 def async_handle_failed_command[**_P](orig_func: Callable[Concatenate[HomeAssistant, ActiveConnection, dict[str, Any], _P], Coroutine[Any, Any, None]]) -> Callable[Concatenate[HomeAssistant, ActiveConnection, dict[str, Any], _P], Coroutine[Any, Any, None]]: ...
@@ -106,27 +107,27 @@ async def websocket_node_alerts(hass: HomeAssistant, connection: ActiveConnectio
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_add_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_add_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_cancel_secure_bootstrap_s2(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_cancel_secure_bootstrap_s2(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_subscribe_s2_inclusion(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_subscribe_s2_inclusion(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_grant_security_classes(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_grant_security_classes(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_validate_dsk_and_enter_pin(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_validate_dsk_and_enter_pin(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 async def websocket_subscribe_new_devices(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]) -> None: ...
@@ -134,52 +135,52 @@ async def websocket_subscribe_new_devices(hass: HomeAssistant, connection: Activ
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_provision_smart_start_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_provision_smart_start_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_unprovision_smart_start_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_unprovision_smart_start_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_get_provisioning_entries(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_get_provisioning_entries(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_parse_qr_code_string(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_parse_qr_code_string(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_try_parse_dsk_from_qr_code_string(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_try_parse_dsk_from_qr_code_string(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_lookup_device(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_lookup_device(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_supports_feature(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_supports_feature(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_stop_inclusion(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_stop_inclusion(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_stop_exclusion(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_stop_exclusion(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_remove_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_remove_node(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
@@ -194,16 +195,16 @@ async def websocket_remove_failed_node(hass: HomeAssistant, connection: ActiveCo
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_begin_rebuilding_routes(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_begin_rebuilding_routes(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_get_entry
-async def websocket_subscribe_rebuild_routes_progress(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_subscribe_rebuild_routes_progress(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_stop_rebuilding_routes(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_stop_rebuilding_routes(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
@@ -248,26 +249,26 @@ def filename_is_present_if_logging_to_file(obj: dict) -> dict: ...
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_subscribe_log_updates(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_subscribe_log_updates(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_update_log_config(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_update_log_config(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_get_entry
-async def websocket_get_log_config(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_get_log_config(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_update_data_collection_preference(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_update_data_collection_preference(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_data_collection_status(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_data_collection_status(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
@@ -293,7 +294,7 @@ async def websocket_get_node_firmware_update_capabilities(hass: HomeAssistant, c
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_is_any_ota_firmware_update_in_progress(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_is_any_ota_firmware_update_in_progress(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 
 class FirmwareUploadView(HomeAssistantView):
     url: str
@@ -307,17 +308,17 @@ class FirmwareUploadView(HomeAssistantView):
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_check_for_config_updates(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_check_for_config_updates(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_install_config_update(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_install_config_update(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 def _get_controller_statistics_dict(statistics: ControllerStatistics) -> dict[str, int]: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_get_entry
-async def websocket_subscribe_controller_statistics(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_subscribe_controller_statistics(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 def _get_node_statistics_dict(hass: HomeAssistant, statistics: NodeStatistics) -> dict[str, Any]: ...
 @websocket_api.require_admin
 @websocket_api.async_response
@@ -327,7 +328,7 @@ async def websocket_subscribe_node_statistics(hass: HomeAssistant, connection: A
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_hard_reset_controller(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_hard_reset_controller(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_node
@@ -344,9 +345,9 @@ def websocket_get_integration_settings(hass: HomeAssistant, connection: ActiveCo
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_backup_nvm(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_backup_nvm(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...
 @websocket_api.require_admin
 @websocket_api.async_response
 @async_handle_failed_command
 @async_get_entry
-async def websocket_restore_nvm(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ConfigEntry, client: Client, driver: Driver) -> None: ...
+async def websocket_restore_nvm(hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any], entry: ZwaveJSConfigEntry, client: Client, driver: Driver) -> None: ...

homeassistant-stubs/components/zwave_js/binary_sensor.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.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -46,12 +46,12 @@ BOOLEAN_SENSOR_MAPPINGS: dict[tuple[int, int | str], BinarySensorEntityDescripti
 
 @callback
 def is_valid_notification_binary_sensor(info: ZwaveDiscoveryInfo) -> bool | NotificationZWaveJSEntityDescription: ...
-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 ZWaveBooleanBinarySensor(ZWaveBaseEntity, BinarySensorEntity):
     _attr_name: Incomplete
     entity_description: 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: ...
 
@@ -60,18 +60,18 @@ class ZWaveNotificationBinarySensor(ZWaveBaseEntity, BinarySensorEntity):
     entity_description: Incomplete
     _attr_name: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, state_key: str, description: NotificationZWaveJSEntityDescription | None = None) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, state_key: str, description: NotificationZWaveJSEntityDescription | None = None) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
 
 class ZWavePropertyBinarySensor(ZWaveBaseEntity, BinarySensorEntity):
     entity_description: PropertyZWaveJSEntityDescription
     _attr_name: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, description: PropertyZWaveJSEntityDescription) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo, description: PropertyZWaveJSEntityDescription) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
 
 class ZWaveConfigParameterBinarySensor(ZWaveBooleanBinarySensor):
     _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: ...

homeassistant-stubs/components/zwave_js/button.pyi

@@ -1,10 +1,10 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN, LOGGER as LOGGER
+from .const import DOMAIN as DOMAIN, LOGGER as LOGGER
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 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 .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.button import ButtonEntity as ButtonEntity
-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
@@ -14,11 +14,11 @@ from zwave_js_server.model.node import Node as ZwaveNode
 
 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 ZwaveBooleanNodeButton(ZWaveBaseEntity, ButtonEntity):
     _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_press(self) -> None: ...
 
 class ZWaveNodePingButton(ButtonEntity):
@@ -39,5 +39,5 @@ class ZWaveNotificationIdleButton(ZWaveBaseEntity, ButtonEntity):
     _attr_entity_category: Incomplete
     _attr_name: Incomplete
     _attr_unique_id: 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_press(self) -> None: ...

homeassistant-stubs/components/zwave_js/climate.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 DynamicCurrentTempClimateDataTemplate as DynamicCurrentTempClimateDataTemplate
 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.climate import ATTR_HVAC_MODE as ATTR_HVAC_MODE, ATTR_TARGET_TEMP_HIGH as ATTR_TARGET_TEMP_HIGH, ATTR_TARGET_TEMP_LOW as ATTR_TARGET_TEMP_LOW, ClimateEntity as ClimateEntity, ClimateEntityFeature as ClimateEntityFeature, HVACAction as HVACAction, HVACMode as HVACMode, PRESET_NONE as PRESET_NONE
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_TEMPERATURE as ATTR_TEMPERATURE, PRECISION_TENTHS as PRECISION_TENTHS, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -22,7 +22,7 @@ ZW_HVAC_MODE_MAP: dict[int, HVACMode]
 HVAC_CURRENT_MAP: dict[int, HVACAction]
 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 ZWaveClimate(ZWaveBaseEntity, ClimateEntity):
     _attr_precision = PRECISION_TENTHS
@@ -38,7 +38,7 @@ class ZWaveClimate(ZWaveBaseEntity, ClimateEntity):
     _current_humidity: Incomplete
     _fan_mode: Incomplete
     _fan_state: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
     def _setpoint_value_or_raise(self, setpoint_type: ThermostatSetpointType) -> ZwaveValue: ...
     def _setpoint_temperature(self, setpoint_type: ThermostatSetpointType) -> float | None: ...
     def _set_modes_and_presets(self) -> None: ...
@@ -85,6 +85,6 @@ class ZWaveClimate(ZWaveBaseEntity, ClimateEntity):
 
 class DynamicCurrentTempClimate(ZWaveClimate):
     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: ...
     @property
     def current_temperature(self) -> float | None: ...

homeassistant-stubs/components/zwave_js/config_flow.pyi

@@ -1,12 +1,13 @@
 import asyncio
 import voluptuous as vol
 from .addon import get_addon_manager as get_addon_manager
-from .const import ADDON_SLUG as ADDON_SLUG, CONF_ADDON_DEVICE as CONF_ADDON_DEVICE, CONF_ADDON_LR_S2_ACCESS_CONTROL_KEY as CONF_ADDON_LR_S2_ACCESS_CONTROL_KEY, CONF_ADDON_LR_S2_AUTHENTICATED_KEY as CONF_ADDON_LR_S2_AUTHENTICATED_KEY, CONF_ADDON_NETWORK_KEY as CONF_ADDON_NETWORK_KEY, CONF_ADDON_S0_LEGACY_KEY as CONF_ADDON_S0_LEGACY_KEY, CONF_ADDON_S2_ACCESS_CONTROL_KEY as CONF_ADDON_S2_ACCESS_CONTROL_KEY, CONF_ADDON_S2_AUTHENTICATED_KEY as CONF_ADDON_S2_AUTHENTICATED_KEY, CONF_ADDON_S2_UNAUTHENTICATED_KEY as CONF_ADDON_S2_UNAUTHENTICATED_KEY, CONF_INTEGRATION_CREATED_ADDON as CONF_INTEGRATION_CREATED_ADDON, CONF_KEEP_OLD_DEVICES as CONF_KEEP_OLD_DEVICES, CONF_LR_S2_ACCESS_CONTROL_KEY as CONF_LR_S2_ACCESS_CONTROL_KEY, CONF_LR_S2_AUTHENTICATED_KEY as CONF_LR_S2_AUTHENTICATED_KEY, CONF_S0_LEGACY_KEY as CONF_S0_LEGACY_KEY, CONF_S2_ACCESS_CONTROL_KEY as CONF_S2_ACCESS_CONTROL_KEY, CONF_S2_AUTHENTICATED_KEY as CONF_S2_AUTHENTICATED_KEY, CONF_S2_UNAUTHENTICATED_KEY as CONF_S2_UNAUTHENTICATED_KEY, CONF_USB_PATH as CONF_USB_PATH, CONF_USE_ADDON as CONF_USE_ADDON, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN, DRIVER_READY_TIMEOUT as DRIVER_READY_TIMEOUT
+from .const import ADDON_SLUG as ADDON_SLUG, CONF_ADDON_DEVICE as CONF_ADDON_DEVICE, CONF_ADDON_LR_S2_ACCESS_CONTROL_KEY as CONF_ADDON_LR_S2_ACCESS_CONTROL_KEY, CONF_ADDON_LR_S2_AUTHENTICATED_KEY as CONF_ADDON_LR_S2_AUTHENTICATED_KEY, CONF_ADDON_NETWORK_KEY as CONF_ADDON_NETWORK_KEY, CONF_ADDON_S0_LEGACY_KEY as CONF_ADDON_S0_LEGACY_KEY, CONF_ADDON_S2_ACCESS_CONTROL_KEY as CONF_ADDON_S2_ACCESS_CONTROL_KEY, CONF_ADDON_S2_AUTHENTICATED_KEY as CONF_ADDON_S2_AUTHENTICATED_KEY, CONF_ADDON_S2_UNAUTHENTICATED_KEY as CONF_ADDON_S2_UNAUTHENTICATED_KEY, CONF_INTEGRATION_CREATED_ADDON as CONF_INTEGRATION_CREATED_ADDON, CONF_KEEP_OLD_DEVICES as CONF_KEEP_OLD_DEVICES, CONF_LR_S2_ACCESS_CONTROL_KEY as CONF_LR_S2_ACCESS_CONTROL_KEY, CONF_LR_S2_AUTHENTICATED_KEY as CONF_LR_S2_AUTHENTICATED_KEY, CONF_S0_LEGACY_KEY as CONF_S0_LEGACY_KEY, CONF_S2_ACCESS_CONTROL_KEY as CONF_S2_ACCESS_CONTROL_KEY, CONF_S2_AUTHENTICATED_KEY as CONF_S2_AUTHENTICATED_KEY, CONF_S2_UNAUTHENTICATED_KEY as CONF_S2_UNAUTHENTICATED_KEY, CONF_USB_PATH as CONF_USB_PATH, CONF_USE_ADDON as CONF_USE_ADDON, DOMAIN as DOMAIN, DRIVER_READY_TIMEOUT as DRIVER_READY_TIMEOUT
 from .helpers import CannotConnect as CannotConnect, async_get_version_info as async_get_version_info
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components import usb as usb
 from homeassistant.components.hassio import AddonError as AddonError, AddonInfo as AddonInfo, AddonManager as AddonManager, AddonState as AddonState
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, SOURCE_USB as SOURCE_USB
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, SOURCE_USB as SOURCE_USB
 from homeassistant.const import CONF_NAME as CONF_NAME, CONF_URL as CONF_URL
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.data_entry_flow import AbortFlow as AbortFlow
@@ -61,7 +62,7 @@ class ZWaveJSConfigFlow(ConfigFlow, domain=DOMAIN):
     backup_filepath: Path | None
     use_addon: bool
     _migrating: bool
-    _reconfigure_config_entry: ConfigEntry | None
+    _reconfigure_config_entry: ZwaveJSConfigEntry | None
     _usb_discovery: bool
     _recommended_install: bool
     def __init__(self) -> None: ...
@@ -82,6 +83,7 @@ class ZWaveJSConfigFlow(ConfigFlow, domain=DOMAIN):
     async def async_step_zeroconf(self, discovery_info: ZeroconfServiceInfo) -> ConfigFlowResult: ...
     async def async_step_zeroconf_confirm(self, user_input: dict | None = None) -> ConfigFlowResult: ...
     async def async_step_usb(self, discovery_info: UsbServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_confirm_usb_migration(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_manual(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_hassio(self, discovery_info: HassioServiceInfo) -> ConfigFlowResult: ...
     async def async_step_hassio_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/zwave_js/const.pyi

@@ -25,8 +25,6 @@ CONF_USB_PATH: str
 CONF_USE_ADDON: str
 CONF_DATA_COLLECTION_OPTED_IN: str
 DOMAIN: str
-DATA_CLIENT: str
-DATA_OLD_SERVER_LOG_LEVEL: str
 EVENT_DEVICE_ADDED_TO_REGISTRY: Incomplete
 EVENT_VALUE_UPDATED: str
 LOGGER: Incomplete

homeassistant-stubs/components/zwave_js/cover.pyi

@@ -1,10 +1,10 @@
-from .const import COVER_POSITION_PROPERTY_KEYS as COVER_POSITION_PROPERTY_KEYS, COVER_TILT_PROPERTY_KEYS as COVER_TILT_PROPERTY_KEYS, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import COVER_POSITION_PROPERTY_KEYS as COVER_POSITION_PROPERTY_KEYS, COVER_TILT_PROPERTY_KEYS as COVER_TILT_PROPERTY_KEYS, DOMAIN as DOMAIN
 from .discovery import ZwaveDiscoveryInfo as ZwaveDiscoveryInfo
 from .discovery_data_template import CoverTiltDataTemplate as CoverTiltDataTemplate
 from .entity import ZWaveBaseEntity as ZWaveBaseEntity
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.cover import ATTR_POSITION as ATTR_POSITION, ATTR_TILT_POSITION as ATTR_TILT_POSITION, CoverDeviceClass as CoverDeviceClass, CoverEntity as CoverEntity, CoverEntityFeature as CoverEntityFeature
-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
@@ -14,7 +14,7 @@ from zwave_js_server.model.value import Value as ZwaveValue
 
 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 CoverPositionMixin(ZWaveBaseEntity, CoverEntity):
     _current_position_value: ZwaveValue | None
@@ -62,17 +62,17 @@ class CoverTiltMixin(ZWaveBaseEntity, CoverEntity):
 
 class ZWaveMultilevelSwitchCover(CoverPositionMixin):
     _attr_device_class: Incomplete
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
 
 class ZWaveTiltCover(ZWaveMultilevelSwitchCover, CoverTiltMixin):
-    def __init__(self, config_entry: ConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
+    def __init__(self, config_entry: ZwaveJSConfigEntry, driver: Driver, info: ZwaveDiscoveryInfo) -> None: ...
 
 class ZWaveWindowCovering(CoverPositionMixin, CoverTiltMixin):
     _up_value: Incomplete
     _down_value: Incomplete
     _attr_name: Incomplete
     _attr_device_class: 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 _fully_open_tilt(self) -> int: ...
     @property
@@ -87,7 +87,7 @@ class ZwaveMotorizedBarrier(ZWaveBaseEntity, CoverEntity):
     _attr_supported_features: Incomplete
     _attr_device_class: Incomplete
     _target_state: 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_opening(self) -> bool | None: ...
     @property

homeassistant-stubs/components/zwave_js/device_automation_helpers.pyi

@@ -1,4 +1,4 @@
-from .const import DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback

homeassistant-stubs/components/zwave_js/diagnostics.pyi

@@ -1,8 +1,8 @@
-from .const import DATA_CLIENT as DATA_CLIENT, USER_AGENT as USER_AGENT
+from .const import USER_AGENT as USER_AGENT
 from .helpers import ZwaveValueMatcher as ZwaveValueMatcher, get_home_and_node_id_from_device_entry as get_home_and_node_id_from_device_entry, get_state_key_from_unique_id as get_state_key_from_unique_id, get_value_id_from_unique_id as get_value_id_from_unique_id, value_matches_matcher as value_matches_matcher
+from .models import ZwaveJSConfigEntry as ZwaveJSConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import REDACTED as REDACTED, async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_URL as CONF_URL
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers import device_registry as dr
@@ -18,6 +18,6 @@ VALUES_TO_REDACT: Incomplete
 def _redacted_value(zwave_value: ValueDataType) -> ValueDataType: ...
 def optionally_redact_value_of_zwave_value(zwave_value: ValueDataType) -> ValueDataType: ...
 def redact_node_state(node_state: dict) -> dict: ...
-def get_device_entities(hass: HomeAssistant, node: Node, config_entry: ConfigEntry, device: dr.DeviceEntry) -> list[dict[str, Any]]: ...
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry) -> dict[str, Any]: ...
-async def async_get_device_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry, device: dr.DeviceEntry) -> dict[str, Any]: ...
+def get_device_entities(hass: HomeAssistant, node: Node, config_entry: ZwaveJSConfigEntry, device: dr.DeviceEntry) -> list[dict[str, Any]]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry) -> dict[str, Any]: ...
+async def async_get_device_diagnostics(hass: HomeAssistant, config_entry: ZwaveJSConfigEntry, device: dr.DeviceEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/zwave_js/entity.pyi

@@ -15,8 +15,6 @@ from zwave_js_server.model.driver import Driver as Driver
 from zwave_js_server.model.value import SetValueResult as SetValueResult, Value as ZwaveValue
 
 EVENT_VALUE_REMOVED: str
-EVENT_DEAD: str
-EVENT_ALIVE: str
 
 class ZWaveBaseEntity(Entity):
     _attr_should_poll: bool
@@ -41,8 +39,6 @@ class ZWaveBaseEntity(Entity):
     @property
     def available(self) -> bool: ...
     @callback
-    def _node_status_alive_or_dead(self, event_data: dict) -> None: ...
-    @callback
     def _value_changed(self, event_data: dict) -> None: ...
     @callback
     def _value_removed(self, event_data: dict) -> None: ...

homeassistant-stubs/components/zwave_js/event.pyi

@@ -1,9 +1,9 @@
-from .const import ATTR_VALUE as ATTR_VALUE, DATA_CLIENT as DATA_CLIENT, DOMAIN as DOMAIN
+from .const import ATTR_VALUE as ATTR_VALUE, 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.event import EventEntity as EventEntity
-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,14 +12,14 @@ from zwave_js_server.model.value import Value as Value, ValueNotification as Val
 
 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: ...
 def _cc_and_label(value: Value) -> str: ...
 
 class ZwaveEventEntity(ZWaveBaseEntity, EventEntity):
     states: dict[int, str]
     _attr_event_types: 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: ...
     @callback
     def _async_handle_event(self, value_notification: ValueNotification) -> None: ...
     async def async_added_to_hass(self) -> None: ...

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 .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
 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
@@ -41,8 +42,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 +67,7 @@ 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: ...
 
 class CannotConnect(HomeAssistantError): ...