homeassistant-stubs 2024.2.4__py3-none-any.whl → 2024.3.0b0__py3-none-any.whl

homeassistant-stubs/components/unifi/diagnostics.pyi

@@ -1,4 +1,4 @@
-from .controller import UniFiController as UniFiController
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.components.diagnostics import REDACTED as REDACTED, async_redact_data as async_redact_data

homeassistant-stubs/components/unifi/entity.pyi

@@ -1,7 +1,7 @@
 import abc
 import aiounifi
 from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, DOMAIN as DOMAIN
-from .controller import UniFiController as UniFiController
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from abc import abstractmethod
 from aiounifi.interfaces.api_handlers import APIHandler, CallbackType, ItemEvent, UnsubscribeType
@@ -18,25 +18,25 @@ from typing import Generic, TypeVar
 HandlerT = TypeVar('HandlerT', bound=APIHandler)
 SubscriptionT = Callable[[CallbackType, ItemEvent], UnsubscribeType]
 
-def async_device_available_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_wlan_available_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_device_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
-def async_wlan_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
-def async_client_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
+def async_device_available_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_wlan_available_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_device_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+def async_wlan_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+def async_client_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
 
 @dataclass(frozen=True)
 class UnifiDescription(Generic[HandlerT, ApiItemT]):
-    allowed_fn: Callable[[UniFiController, str], bool]
+    allowed_fn: Callable[[UnifiHub, str], bool]
     api_handler_fn: Callable[[aiounifi.Controller], HandlerT]
-    available_fn: Callable[[UniFiController, str], bool]
-    device_info_fn: Callable[[UniFiController, str], DeviceInfo | None]
+    available_fn: Callable[[UnifiHub, str], bool]
+    device_info_fn: Callable[[UnifiHub, str], DeviceInfo | None]
     event_is_on: tuple[EventKey, ...] | None
     event_to_subscribe: tuple[EventKey, ...] | None
     name_fn: Callable[[ApiItemT], str | None]
     object_fn: Callable[[aiounifi.Controller, str], ApiItemT]
     should_poll: bool
-    supported_fn: Callable[[UniFiController, str], bool | None]
-    unique_id_fn: Callable[[UniFiController, str], str]
+    supported_fn: Callable[[UnifiHub, str], bool | None]
+    unique_id_fn: Callable[[UnifiHub, str], str]
     def __init__(self, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn) -> None: ...
 
 @dataclass(frozen=True)
@@ -47,13 +47,13 @@ class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
     entity_description: UnifiEntityDescription[HandlerT, ApiItemT]
     _attr_unique_id: str
     _obj_id: Incomplete
-    controller: Incomplete
+    hub: Incomplete
     _removed: bool
     _attr_available: Incomplete
     _attr_device_info: Incomplete
     _attr_should_poll: Incomplete
     _attr_name: Incomplete
-    def __init__(self, obj_id: str, controller: UniFiController, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
+    def __init__(self, obj_id: str, hub: UnifiHub, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     def async_signalling_callback(self, event: ItemEvent, obj_id: str) -> None: ...
     def async_signal_reachable_callback(self) -> None: ...

homeassistant-stubs/components/unifi/hub/__init__.pyi

@@ -0,0 +1,2 @@
+from .api import get_unifi_api as get_unifi_api
+from .hub import UnifiHub as UnifiHub

homeassistant-stubs/components/unifi/hub/api.pyi

@@ -0,0 +1,10 @@
+import aiounifi
+from ..const import CONF_SITE_ID as CONF_SITE_ID, LOGGER as LOGGER
+from ..errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers import aiohttp_client as aiohttp_client
+from types import MappingProxyType
+from typing import Any
+
+async def get_unifi_api(hass: HomeAssistant, config: MappingProxyType[str, Any]) -> aiounifi.Controller: ...

homeassistant-stubs/components/unifi/{controller.pyi → hub/hub.pyi}

@@ -1,32 +1,25 @@
 import aiounifi
-from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_ALLOW_BANDWIDTH_SENSORS as DEFAULT_ALLOW_BANDWIDTH_SENSORS, DEFAULT_ALLOW_UPTIME_SENSORS as DEFAULT_ALLOW_UPTIME_SENSORS, DEFAULT_DETECTION_TIME as DEFAULT_DETECTION_TIME, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DEFAULT_IGNORE_WIRED_BUG as DEFAULT_IGNORE_WIRED_BUG, DEFAULT_TRACK_CLIENTS as DEFAULT_TRACK_CLIENTS, DEFAULT_TRACK_DEVICES as DEFAULT_TRACK_DEVICES, DEFAULT_TRACK_WIRED_CLIENTS as DEFAULT_TRACK_WIRED_CLIENTS, LOGGER as LOGGER, PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
-from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
-from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_ALLOW_BANDWIDTH_SENSORS as DEFAULT_ALLOW_BANDWIDTH_SENSORS, DEFAULT_ALLOW_UPTIME_SENSORS as DEFAULT_ALLOW_UPTIME_SENSORS, DEFAULT_DETECTION_TIME as DEFAULT_DETECTION_TIME, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DEFAULT_IGNORE_WIRED_BUG as DEFAULT_IGNORE_WIRED_BUG, DEFAULT_TRACK_CLIENTS as DEFAULT_TRACK_CLIENTS, DEFAULT_TRACK_DEVICES as DEFAULT_TRACK_DEVICES, DEFAULT_TRACK_WIRED_CLIENTS as DEFAULT_TRACK_WIRED_CLIENTS, PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
+from ..entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
+from .websocket import UnifiWebsocket as UnifiWebsocket
 from _typeshed import Incomplete
 from datetime import datetime
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL, Platform as Platform
+from homeassistant.const import CONF_HOST as CONF_HOST, Platform as Platform
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers import aiohttp_client as aiohttp_client
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry, DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.entity_registry import async_entries_for_config_entry as async_entries_for_config_entry
 from homeassistant.helpers.event import async_call_later as async_call_later, async_track_time_interval as async_track_time_interval
-from types import MappingProxyType
-from typing import Any
 
-RETRY_TIMER: int
 CHECK_HEARTBEAT_INTERVAL: Incomplete
-CHECK_WEBSOCKET_INTERVAL: Incomplete
 
-class UniFiController:
+class UnifiHub:
     hass: Incomplete
     config_entry: Incomplete
     api: Incomplete
-    ws_task: Incomplete
-    _cancel_websocket_check: Incomplete
-    available: bool
+    websocket: Incomplete
     wireless_clients: Incomplete
     site: Incomplete
     is_admin: bool
@@ -51,8 +44,11 @@ class UniFiController:
     def load_config_entry_options(self) -> None: ...
     @property
     def host(self) -> str: ...
+    @property
+    def available(self) -> bool: ...
     @staticmethod
     def register_platform(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, entity_class: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], requires_admin: bool = False) -> None: ...
+    def _async_should_add_entity(self, description: UnifiEntityDescription, obj_id: str) -> bool: ...
     def register_platform_add_entities(self, unifi_platform_entity: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], async_add_entities: AddEntitiesCallback) -> None: ...
     @property
     def signal_reachable(self) -> str: ...
@@ -69,11 +65,5 @@ class UniFiController:
     def async_update_device_registry(self) -> DeviceEntry: ...
     @staticmethod
     async def async_config_entry_updated(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
-    def start_websocket(self) -> None: ...
-    def reconnect(self, log: bool = False) -> None: ...
-    async def async_reconnect(self) -> None: ...
-    def _async_watch_websocket(self, now: datetime) -> None: ...
     def shutdown(self, event: Event) -> None: ...
     async def async_reset(self) -> bool: ...
-
-async def get_unifi_controller(hass: HomeAssistant, config: MappingProxyType[str, Any]) -> aiounifi.Controller: ...

homeassistant-stubs/components/unifi/hub/websocket.pyi

@@ -0,0 +1,25 @@
+import aiounifi
+from ..const import LOGGER as LOGGER
+from _typeshed import Incomplete
+from datetime import datetime
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
+from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
+
+RETRY_TIMER: int
+CHECK_WEBSOCKET_INTERVAL: Incomplete
+
+class UnifiWebsocket:
+    hass: Incomplete
+    api: Incomplete
+    signal: Incomplete
+    ws_task: Incomplete
+    _cancel_websocket_check: Incomplete
+    available: bool
+    def __init__(self, hass: HomeAssistant, api: aiounifi.Controller, signal: str) -> None: ...
+    def start(self) -> None: ...
+    def stop(self) -> None: ...
+    async def stop_and_wait(self) -> None: ...
+    def start_websocket(self) -> None: ...
+    def reconnect(self, log: bool = False) -> None: ...
+    def _async_watch_websocket(self, now: datetime) -> None: ...

homeassistant-stubs/components/unifi/image.pyi

@@ -1,5 +1,5 @@
-from .controller import UniFiController as UniFiController
 from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
@@ -13,11 +13,11 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Generic
 
-def async_wlan_qr_code_image_fn(controller: UniFiController, wlan: Wlan) -> bytes: ...
+def async_wlan_qr_code_image_fn(hub: UnifiHub, wlan: Wlan) -> bytes: ...
 
 @dataclass(frozen=True)
 class UnifiImageEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
-    image_fn: Callable[[UniFiController, ApiItemT], bytes]
+    image_fn: Callable[[UnifiHub, ApiItemT], bytes]
     value_fn: Callable[[ApiItemT], str | None]
     def __init__(self, image_fn, value_fn) -> None: ...
 
@@ -34,7 +34,7 @@ class UnifiImageEntity(UnifiEntity[HandlerT, ApiItemT], ImageEntity):
     _attr_content_type: str
     current_image: bytes | None
     previous_value: str | None
-    def __init__(self, obj_id: str, controller: UniFiController, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
+    def __init__(self, obj_id: str, hub: UnifiHub, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
     def image(self) -> bytes | None: ...
     _attr_image_last_updated: Incomplete
     def async_update_state(self, event: ItemEvent, obj_id: str) -> None: ...

homeassistant-stubs/components/unifi/sensor.pyi

@@ -1,6 +1,6 @@
 from .const import DEVICE_STATES as DEVICE_STATES
-from .controller import UniFiController as UniFiController
 from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_client_device_info_fn as async_client_device_info_fn, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_available_fn as async_wlan_available_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
@@ -20,28 +20,28 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 from homeassistant.helpers.typing import StateType as StateType
 from typing import Generic
 
-def async_bandwidth_sensor_allowed_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_uptime_sensor_allowed_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_client_rx_value_fn(controller: UniFiController, client: Client) -> float: ...
-def async_client_tx_value_fn(controller: UniFiController, client: Client) -> float: ...
-def async_client_uptime_value_fn(controller: UniFiController, client: Client) -> datetime: ...
-def async_wlan_client_value_fn(controller: UniFiController, wlan: Wlan) -> int: ...
-def async_device_uptime_value_fn(controller: UniFiController, device: Device) -> datetime | None: ...
+def async_bandwidth_sensor_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_uptime_sensor_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_client_rx_value_fn(hub: UnifiHub, client: Client) -> float: ...
+def async_client_tx_value_fn(hub: UnifiHub, client: Client) -> float: ...
+def async_client_uptime_value_fn(hub: UnifiHub, client: Client) -> datetime: ...
+def async_wlan_client_value_fn(hub: UnifiHub, wlan: Wlan) -> int: ...
+def async_device_uptime_value_fn(hub: UnifiHub, device: Device) -> datetime | None: ...
 def async_device_uptime_value_changed_fn(old: StateType | date | datetime | Decimal, new: datetime | float | str | None) -> bool: ...
-def async_device_outlet_power_supported_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_device_outlet_supported_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_client_is_connected_fn(controller: UniFiController, obj_id: str) -> bool: ...
+def async_device_outlet_power_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_device_outlet_supported_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_client_is_connected_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 
 @dataclass(frozen=True)
 class UnifiSensorEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
-    value_fn: Callable[[UniFiController, ApiItemT], datetime | float | str | None]
+    value_fn: Callable[[UnifiHub, ApiItemT], datetime | float | str | None]
     def __init__(self, value_fn) -> None: ...
 
-def async_device_state_value_fn(controller: UniFiController, device: Device) -> str: ...
+def async_device_state_value_fn(hub: UnifiHub, device: Device) -> str: ...
 
 @dataclass(frozen=True)
 class UnifiSensorEntityDescription(SensorEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT], UnifiSensorEntityDescriptionMixin[HandlerT, ApiItemT]):
-    is_connected_fn: Callable[[UniFiController, str], bool] | None = ...
+    is_connected_fn: Callable[[UnifiHub, str], bool] | None = ...
     value_changed_fn: Callable[[StateType | date | datetime | Decimal, datetime | float | str | None], bool] = ...
     def __init__(self, value_fn, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, is_connected_fn, value_changed_fn) -> None: ...
 

homeassistant-stubs/components/unifi/switch.pyi

@@ -1,7 +1,7 @@
 import aiounifi
 from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER
-from .controller import UNIFI_DOMAIN as UNIFI_DOMAIN, UniFiController as UniFiController
 from .entity import HandlerT as HandlerT, SubscriptionT as SubscriptionT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_client_device_info_fn as async_client_device_info_fn, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn, async_wlan_device_info_fn as async_wlan_device_info_fn
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent
 from aiounifi.models.api import ApiItemT
@@ -20,22 +20,22 @@ from typing import Any, Generic
 CLIENT_BLOCKED: Incomplete
 CLIENT_UNBLOCKED: Incomplete
 
-def async_block_client_allowed_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_dpi_group_is_on_fn(controller: UniFiController, dpi_group: DPIRestrictionGroup) -> bool: ...
-def async_dpi_group_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
-def async_port_forward_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
-async def async_block_client_control_fn(controller: UniFiController, obj_id: str, target: bool) -> None: ...
-async def async_dpi_group_control_fn(controller: UniFiController, obj_id: str, target: bool) -> None: ...
-def async_outlet_supports_switching_fn(controller: UniFiController, obj_id: str) -> bool: ...
-async def async_outlet_control_fn(controller: UniFiController, obj_id: str, target: bool) -> None: ...
-async def async_poe_port_control_fn(controller: UniFiController, obj_id: str, target: bool) -> None: ...
-async def async_port_forward_control_fn(controller: UniFiController, obj_id: str, target: bool) -> None: ...
-async def async_wlan_control_fn(controller: UniFiController, obj_id: str, target: bool) -> None: ...
+def async_block_client_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_dpi_group_is_on_fn(hub: UnifiHub, dpi_group: DPIRestrictionGroup) -> bool: ...
+def async_dpi_group_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+def async_port_forward_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+async def async_block_client_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_dpi_group_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+def async_outlet_supports_switching_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+async def async_outlet_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_poe_port_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_port_forward_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_wlan_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 
 @dataclass(frozen=True)
 class UnifiSwitchEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
-    control_fn: Callable[[UniFiController, str, bool], Coroutine[Any, Any, None]]
-    is_on_fn: Callable[[UniFiController, ApiItemT], bool]
+    control_fn: Callable[[UnifiHub, str, bool], Coroutine[Any, Any, None]]
+    is_on_fn: Callable[[UnifiHub, ApiItemT], bool]
     def __init__(self, control_fn, is_on_fn) -> None: ...
 
 @dataclass(frozen=True)

homeassistant-stubs/components/unifi/update.pyi

@@ -1,6 +1,6 @@
 import aiounifi
-from .controller import UniFiController as UniFiController
 from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent
 from aiounifi.interfaces.devices import Devices

homeassistant-stubs/components/unifiprotect/__init__.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_ALLOW_EA as CONF_ALLOW_EA, DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN, MIN_REQUIRED_PROTECT_V as MIN_REQUIRED_PROTECT_V, OUTDATED_LOG_MESSAGE as OUTDATED_LOG_MESSAGE, PLATFORMS as PLATFORMS
+from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_ALLOW_EA as CONF_ALLOW_EA, DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DOMAIN as DOMAIN, MIN_REQUIRED_PROTECT_V as MIN_REQUIRED_PROTECT_V, OUTDATED_LOG_MESSAGE as OUTDATED_LOG_MESSAGE, PLATFORMS as PLATFORMS
 from .data import ProtectData as ProtectData, async_ufp_instance_for_config_entry_ids as async_ufp_instance_for_config_entry_ids
 from .discovery import async_start_discovery as async_start_discovery
 from .migrate import async_migrate_data as async_migrate_data

homeassistant-stubs/components/unifiprotect/const.pyi

@@ -2,6 +2,7 @@ from _typeshed import Incomplete
 from homeassistant.const import Platform as Platform
 
 DOMAIN: str
+AUTH_RETRIES: int
 ATTR_EVENT_SCORE: str
 ATTR_EVENT_ID: str
 ATTR_WIDTH: str

homeassistant-stubs/components/unifiprotect/data.pyi

@@ -1,8 +1,8 @@
-from .const import CONF_DISABLE_RTSP as CONF_DISABLE_RTSP, CONF_MAX_MEDIA as CONF_MAX_MEDIA, DEFAULT_MAX_MEDIA as DEFAULT_MAX_MEDIA, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DISPATCH_ADD as DISPATCH_ADD, DISPATCH_ADOPT as DISPATCH_ADOPT, DISPATCH_CHANNELS as DISPATCH_CHANNELS, DOMAIN as DOMAIN
+from .const import AUTH_RETRIES as AUTH_RETRIES, CONF_DISABLE_RTSP as CONF_DISABLE_RTSP, CONF_MAX_MEDIA as CONF_MAX_MEDIA, DEFAULT_MAX_MEDIA as DEFAULT_MAX_MEDIA, DEVICES_THAT_ADOPT as DEVICES_THAT_ADOPT, DISPATCH_ADD as DISPATCH_ADD, DISPATCH_ADOPT as DISPATCH_ADOPT, DISPATCH_CHANNELS as DISPATCH_CHANNELS, DOMAIN as DOMAIN
 from .utils import async_get_devices_by_type as async_get_devices_by_type
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Generator, Iterable
-from datetime import timedelta
+from datetime import datetime, timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
@@ -43,6 +43,7 @@ class ProtectData:
     def _async_update_device(self, device: ProtectAdoptableDeviceModel | NVR, changed_data: dict[str, Any]) -> None: ...
     def _async_process_ws_message(self, message: WSSubscriptionMessage) -> None: ...
     def _async_process_updates(self, updates: Bootstrap | None) -> None: ...
+    def _async_poll(self, now: datetime) -> None: ...
     def async_subscribe_device_id(self, mac: str, update_callback: Callable[[ProtectDeviceType], None]) -> CALLBACK_TYPE: ...
     def async_unsubscribe_device_id(self, mac: str, update_callback: Callable[[ProtectDeviceType], None]) -> None: ...
     def _async_signal_device_update(self, device: ProtectDeviceType) -> None: ...

homeassistant-stubs/components/unifiprotect/media_source.pyi

@@ -22,6 +22,7 @@ class SimpleEventType(str, Enum):
     RING: str
     MOTION: str
     SMART: str
+    AUDIO: str
 
 class IdentifierType(str, Enum):
     EVENT: str
@@ -32,14 +33,16 @@ class IdentifierTimeType(str, Enum):
     RECENT: str
     RANGE: str
 
-EVENT_MAP: Incomplete
+EVENT_MAP: dict[SimpleEventType, set[EventType]]
 EVENT_NAME_MAP: Incomplete
 
-def get_ufp_event(event_type: SimpleEventType) -> EventType | None: ...
+def get_ufp_event(event_type: SimpleEventType) -> set[EventType]: ...
 async def async_get_media_source(hass: HomeAssistant) -> MediaSource: ...
 def _get_month_start_end(start: datetime) -> tuple[datetime, datetime]: ...
 def _bad_identifier(identifier: str, err: Exception | None = None) -> NoReturn: ...
 def _format_duration(duration: timedelta) -> str: ...
+def _get_object_name(event: Event | dict[str, Any]) -> str: ...
+def _get_audio_name(event: Event | dict[str, Any]) -> str: ...
 
 class ProtectMediaSource(MediaSource):
     name: str
@@ -54,7 +57,7 @@ class ProtectMediaSource(MediaSource):
     def async_get_registry(self) -> er.EntityRegistry: ...
     def _breadcrumb(self, data: ProtectData, base_title: str, camera: Camera | None = None, event_type: SimpleEventType | None = None, count: int | None = None) -> str: ...
     async def _build_event(self, data: ProtectData, event: dict[str, Any] | Event, thumbnail_only: bool = False) -> BrowseMediaSource: ...
-    async def _build_events(self, data: ProtectData, start: datetime, end: datetime, camera_id: str | None = None, event_type: EventType | None = None, reserve: bool = False) -> list[BrowseMediaSource]: ...
+    async def _build_events(self, data: ProtectData, start: datetime, end: datetime, camera_id: str | None = None, event_types: set[EventType] | None = None, reserve: bool = False) -> list[BrowseMediaSource]: ...
     async def _build_recent(self, data: ProtectData, camera_id: str, event_type: SimpleEventType, days: int, build_children: bool = False) -> BrowseMediaSource: ...
     async def _build_month(self, data: ProtectData, camera_id: str, event_type: SimpleEventType, start: date, build_children: bool = False) -> BrowseMediaSource: ...
     async def _build_days(self, data: ProtectData, camera_id: str, event_type: SimpleEventType, start: date, is_all: bool = True, build_children: bool = False) -> BrowseMediaSource: ...

homeassistant-stubs/components/unifiprotect/repairs.pyi

@@ -9,13 +9,19 @@ from pyunifiprotect import ProtectApiClient as ProtectApiClient
 
 _LOGGER: Incomplete
 
-class EAConfirm(RepairsFlow):
+class ProtectRepair(RepairsFlow):
     _api: ProtectApiClient
     _entry: ConfigEntry
     def __init__(self, api: ProtectApiClient, entry: ConfigEntry) -> None: ...
-    def _async_get_placeholders(self) -> dict[str, str] | None: ...
+    def _async_get_placeholders(self) -> dict[str, str]: ...
+
+class EAConfirm(ProtectRepair):
     async def async_step_init(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
     async def async_step_start(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
     async def async_step_confirm(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
 
+class CloudAccount(ProtectRepair):
+    async def async_step_init(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
+    async def async_step_confirm(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
+
 async def async_create_fix_flow(hass: HomeAssistant, issue_id: str, data: dict[str, str | int | float | None] | None) -> RepairsFlow: ...

homeassistant-stubs/components/usb/__init__.pyi

@@ -36,7 +36,7 @@ class USBDiscovery:
     def __init__(self, hass: HomeAssistant, usb: list[USBMatcher]) -> None: ...
     async def async_setup(self) -> None: ...
     async def async_start(self, event: Event) -> None: ...
-    async def async_stop(self, event: Event) -> None: ...
+    def async_stop(self, event: Event) -> None: ...
     async def _async_start_monitor(self) -> None: ...
     def _device_discovered(self, device: Device) -> None: ...
     def async_register_scan_request_callback(self, _callback: CALLBACK_TYPE) -> CALLBACK_TYPE: ...

homeassistant-stubs/components/vacuum/intent.pyi

@@ -0,0 +1,8 @@
+from . import DOMAIN as DOMAIN, SERVICE_RETURN_TO_BASE as SERVICE_RETURN_TO_BASE, SERVICE_START as SERVICE_START
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers import intent as intent
+
+INTENT_VACUUM_START: str
+INTENT_VACUUM_RETURN_TO_BASE: str
+
+async def async_setup_intents(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/vallox/__init__.pyi

@@ -1,17 +1,13 @@
 import voluptuous as vol
-from .const import DEFAULT_FAN_SPEED_AWAY as DEFAULT_FAN_SPEED_AWAY, DEFAULT_FAN_SPEED_BOOST as DEFAULT_FAN_SPEED_BOOST, DEFAULT_FAN_SPEED_HOME as DEFAULT_FAN_SPEED_HOME, DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN, METRIC_KEY_PROFILE_FAN_SPEED_AWAY as METRIC_KEY_PROFILE_FAN_SPEED_AWAY, METRIC_KEY_PROFILE_FAN_SPEED_BOOST as METRIC_KEY_PROFILE_FAN_SPEED_BOOST, METRIC_KEY_PROFILE_FAN_SPEED_HOME as METRIC_KEY_PROFILE_FAN_SPEED_HOME, STATE_SCAN_INTERVAL as STATE_SCAN_INTERVAL
+from .const import DEFAULT_FAN_SPEED_AWAY as DEFAULT_FAN_SPEED_AWAY, DEFAULT_FAN_SPEED_BOOST as DEFAULT_FAN_SPEED_BOOST, DEFAULT_FAN_SPEED_HOME as DEFAULT_FAN_SPEED_HOME, DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN, STATE_SCAN_INTERVAL as STATE_SCAN_INTERVAL
 from _typeshed import Incomplete
-from dataclasses import dataclass
-from datetime import date
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_NAME as CONF_NAME, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
-from homeassistant.helpers.typing import StateType as StateType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity, DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from typing import Any, NamedTuple
-from uuid import UUID
-from vallox_websocket_api import PROFILE as VALLOX_PROFILE, Vallox
+from typing import NamedTuple
+from vallox_websocket_api import MetricData, Vallox
 
 _LOGGER: Incomplete
 CONFIG_SCHEMA: Incomplete
@@ -28,21 +24,7 @@ SERVICE_SET_PROFILE_FAN_SPEED_AWAY: str
 SERVICE_SET_PROFILE_FAN_SPEED_BOOST: str
 SERVICE_TO_METHOD: Incomplete
 
-@dataclass
-class ValloxState:
-    metric_cache: dict[str, Any] = ...
-    profile: VALLOX_PROFILE = ...
-    def get_metric(self, metric_key: str) -> StateType: ...
-    @property
-    def model(self) -> str | None: ...
-    @property
-    def sw_version(self) -> str: ...
-    @property
-    def uuid(self) -> UUID | None: ...
-    def get_next_filter_change_date(self) -> date | None: ...
-    def __init__(self, metric_cache, profile) -> None: ...
-
-class ValloxDataUpdateCoordinator(DataUpdateCoordinator[ValloxState]): ...
+class ValloxDataUpdateCoordinator(DataUpdateCoordinator[MetricData]): ...
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
@@ -50,7 +32,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: .
 class ValloxServiceHandler:
     _client: Incomplete
     _coordinator: Incomplete
-    def __init__(self, client: Vallox, coordinator: DataUpdateCoordinator[ValloxState]) -> None: ...
+    def __init__(self, client: Vallox, coordinator: DataUpdateCoordinator[MetricData]) -> None: ...
     async def async_set_profile_fan_speed_home(self, fan_speed: int = ...) -> bool: ...
     async def async_set_profile_fan_speed_away(self, fan_speed: int = ...) -> bool: ...
     async def async_set_profile_fan_speed_boost(self, fan_speed: int = ...) -> bool: ...

homeassistant-stubs/components/vallox/fan.pyi

@@ -41,4 +41,4 @@ class ValloxFanEntity(ValloxEntity, FanEntity):
     async def async_set_percentage(self, percentage: int) -> None: ...
     async def _async_set_power(self, mode: bool) -> bool: ...
     async def _async_set_preset_mode_internal(self, preset_mode: str) -> bool: ...
-    async def _async_set_percentage_internal(self, percentage: int) -> bool: ...
+    async def _async_set_percentage_internal(self, percentage: int, preset_mode: str | None = None) -> bool: ...

homeassistant-stubs/components/valve/intent.pyi

@@ -0,0 +1,6 @@
+from . import ATTR_POSITION as ATTR_POSITION, DOMAIN as DOMAIN
+from homeassistant.const import SERVICE_SET_VALVE_POSITION as SERVICE_SET_VALVE_POSITION
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers import intent as intent
+
+async def async_setup_intents(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/wake_word/models.pyi

@@ -4,11 +4,13 @@ from dataclasses import dataclass
 class WakeWord:
     id: str
     name: str
-    def __init__(self, id, name) -> None: ...
+    phrase: str | None = ...
+    def __init__(self, id, name, phrase) -> None: ...
 
 @dataclass
 class DetectionResult:
     wake_word_id: str
+    wake_word_phrase: str
     timestamp: int | None
     queued_audio: list[tuple[bytes, int]] | None = ...
-    def __init__(self, wake_word_id, timestamp, queued_audio) -> None: ...
+    def __init__(self, wake_word_id, wake_word_phrase, timestamp, queued_audio) -> None: ...

homeassistant-stubs/components/water_heater/__init__.pyi

@@ -5,6 +5,7 @@ from functools import cached_property as cached_property
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_TEMPERATURE as ATTR_TEMPERATURE, PRECISION_TENTHS as PRECISION_TENTHS, PRECISION_WHOLE as PRECISION_WHOLE, SERVICE_TURN_OFF as SERVICE_TURN_OFF, SERVICE_TURN_ON as SERVICE_TURN_ON, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
 from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.deprecation import DeprecatedConstantEnum as DeprecatedConstantEnum, all_with_deprecated_constants as all_with_deprecated_constants, check_if_deprecated_constant as check_if_deprecated_constant, dir_with_deprecated_constants as dir_with_deprecated_constants
 from homeassistant.helpers.entity import Entity as Entity, EntityDescription as EntityDescription
@@ -109,6 +110,7 @@ class WaterHeaterEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     def set_operation_mode(self, operation_mode: str) -> None: ...
     async def async_set_operation_mode(self, operation_mode: str) -> None: ...
+    async def async_handle_set_operation_mode(self, operation_mode: str) -> None: ...
     def turn_away_mode_on(self) -> None: ...
     async def async_turn_away_mode_on(self) -> None: ...
     def turn_away_mode_off(self) -> None: ...

homeassistant-stubs/components/websocket_api/http.pyi

@@ -14,6 +14,7 @@ from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.event import async_call_later as async_call_later
+from homeassistant.util.async_ import create_eager_task as create_eager_task
 from homeassistant.util.json import json_loads as json_loads
 from typing import Any, Final
 

homeassistant-stubs/components/wiz/light.pyi

@@ -2,7 +2,7 @@ from .const import DOMAIN as DOMAIN
 from .entity import WizToggleEntity as WizToggleEntity
 from .models import WizData as WizData
 from _typeshed import Incomplete
-from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP as ATTR_COLOR_TEMP, ATTR_EFFECT as ATTR_EFFECT, ATTR_RGBWW_COLOR as ATTR_RGBWW_COLOR, ATTR_RGBW_COLOR as ATTR_RGBW_COLOR, ColorMode as ColorMode, LightEntity as LightEntity, LightEntityFeature as LightEntityFeature
+from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP as ATTR_COLOR_TEMP, ATTR_EFFECT as ATTR_EFFECT, ATTR_RGBWW_COLOR as ATTR_RGBWW_COLOR, ATTR_RGBW_COLOR as ATTR_RGBW_COLOR, ColorMode as ColorMode, LightEntity as LightEntity, LightEntityFeature as LightEntityFeature, filter_supported_color_modes as filter_supported_color_modes
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -18,14 +18,15 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 
 class WizBulbEntity(WizToggleEntity, LightEntity):
     _attr_name: Incomplete
+    _fixed_color_mode: ColorMode | None
     _attr_supported_color_modes: Incomplete
+    _attr_color_mode: Incomplete
     _attr_effect_list: Incomplete
     _attr_min_mireds: Incomplete
     _attr_max_mireds: Incomplete
     _attr_supported_features: Incomplete
     def __init__(self, wiz_data: WizData, name: str) -> None: ...
     _attr_brightness: Incomplete
-    _attr_color_mode: Incomplete
     _attr_color_temp: Incomplete
     _attr_rgbww_color: Incomplete
     _attr_rgbw_color: Incomplete

homeassistant-stubs/components/xiaomi_ble/const.pyi

@@ -12,12 +12,21 @@ EVENT_SUBTYPE: Final[str]
 EVENT_PROPERTIES: Final[str]
 XIAOMI_BLE_EVENT: Final[str]
 EVENT_CLASS_BUTTON: Final[str]
+EVENT_CLASS_DIMMER: Final[str]
 EVENT_CLASS_MOTION: Final[str]
+EVENT_CLASS_CUBE: Final[str]
 BUTTON: Final[str]
+CUBE: Final[str]
+DIMMER: Final[str]
 DOUBLE_BUTTON: Final[str]
 TRIPPLE_BUTTON: Final[str]
+REMOTE: Final[str]
+REMOTE_FAN: Final[str]
+REMOTE_VENFAN: Final[str]
+REMOTE_BATHROOM: Final[str]
 MOTION: Final[str]
 BUTTON_PRESS: Final[str]
+BUTTON_PRESS_LONG: Final[str]
 BUTTON_PRESS_DOUBLE_LONG: Final[str]
 DOUBLE_BUTTON_PRESS_DOUBLE_LONG: Final[str]
 TRIPPLE_BUTTON_PRESS_DOUBLE_LONG: Final[str]

homeassistant-stubs/components/xiaomi_ble/device_trigger.pyi

@@ -1,4 +1,4 @@
-from .const import BUTTON as BUTTON, BUTTON_PRESS as BUTTON_PRESS, BUTTON_PRESS_DOUBLE_LONG as BUTTON_PRESS_DOUBLE_LONG, CONF_SUBTYPE as CONF_SUBTYPE, DOMAIN as DOMAIN, DOUBLE_BUTTON as DOUBLE_BUTTON, DOUBLE_BUTTON_PRESS_DOUBLE_LONG as DOUBLE_BUTTON_PRESS_DOUBLE_LONG, EVENT_CLASS as EVENT_CLASS, EVENT_CLASS_BUTTON as EVENT_CLASS_BUTTON, EVENT_CLASS_MOTION as EVENT_CLASS_MOTION, EVENT_TYPE as EVENT_TYPE, MOTION as MOTION, MOTION_DEVICE as MOTION_DEVICE, TRIPPLE_BUTTON as TRIPPLE_BUTTON, TRIPPLE_BUTTON_PRESS_DOUBLE_LONG as TRIPPLE_BUTTON_PRESS_DOUBLE_LONG, XIAOMI_BLE_EVENT as XIAOMI_BLE_EVENT
+from .const import BUTTON as BUTTON, BUTTON_PRESS as BUTTON_PRESS, BUTTON_PRESS_DOUBLE_LONG as BUTTON_PRESS_DOUBLE_LONG, BUTTON_PRESS_LONG as BUTTON_PRESS_LONG, CONF_SUBTYPE as CONF_SUBTYPE, CUBE as CUBE, DIMMER as DIMMER, DOMAIN as DOMAIN, DOUBLE_BUTTON as DOUBLE_BUTTON, DOUBLE_BUTTON_PRESS_DOUBLE_LONG as DOUBLE_BUTTON_PRESS_DOUBLE_LONG, EVENT_CLASS as EVENT_CLASS, EVENT_CLASS_BUTTON as EVENT_CLASS_BUTTON, EVENT_CLASS_CUBE as EVENT_CLASS_CUBE, EVENT_CLASS_DIMMER as EVENT_CLASS_DIMMER, EVENT_CLASS_MOTION as EVENT_CLASS_MOTION, EVENT_TYPE as EVENT_TYPE, MOTION as MOTION, MOTION_DEVICE as MOTION_DEVICE, REMOTE as REMOTE, REMOTE_BATHROOM as REMOTE_BATHROOM, REMOTE_FAN as REMOTE_FAN, REMOTE_VENFAN as REMOTE_VENFAN, TRIPPLE_BUTTON as TRIPPLE_BUTTON, TRIPPLE_BUTTON_PRESS_DOUBLE_LONG as TRIPPLE_BUTTON_PRESS_DOUBLE_LONG, XIAOMI_BLE_EVENT as XIAOMI_BLE_EVENT
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA as DEVICE_TRIGGER_BASE_SCHEMA

homeassistant-stubs/components/xiaomi_ble/event.pyi

@@ -1,5 +1,5 @@
 from . import format_discovered_event_class as format_discovered_event_class, format_event_dispatcher_name as format_event_dispatcher_name
-from .const import DOMAIN as DOMAIN, EVENT_CLASS_BUTTON as EVENT_CLASS_BUTTON, EVENT_CLASS_MOTION as EVENT_CLASS_MOTION, EVENT_PROPERTIES as EVENT_PROPERTIES, EVENT_TYPE as EVENT_TYPE, XiaomiBleEvent as XiaomiBleEvent
+from .const import DOMAIN as DOMAIN, EVENT_CLASS_BUTTON as EVENT_CLASS_BUTTON, EVENT_CLASS_CUBE as EVENT_CLASS_CUBE, EVENT_CLASS_DIMMER as EVENT_CLASS_DIMMER, EVENT_CLASS_MOTION as EVENT_CLASS_MOTION, EVENT_PROPERTIES as EVENT_PROPERTIES, EVENT_TYPE as EVENT_TYPE, XiaomiBleEvent as XiaomiBleEvent
 from .coordinator import XiaomiActiveBluetoothProcessorCoordinator as XiaomiActiveBluetoothProcessorCoordinator
 from _typeshed import Incomplete
 from homeassistant.components.event import EventDeviceClass as EventDeviceClass, EventEntity as EventEntity, EventEntityDescription as EventEntityDescription

homeassistant-stubs/components/yalexs_ble/__init__.pyi

@@ -4,7 +4,7 @@ from .util import async_find_existing_service_info as async_find_existing_servic
 from homeassistant.components import bluetooth as bluetooth
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ADDRESS as CONF_ADDRESS, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
-from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, CoreState as CoreState, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from yalexs_ble import ConnectionInfo as ConnectionInfo, LockInfo as LockInfo, LockState as LockState