homeassistant-stubs 2024.5.5__py3-none-any.whl → 2024.6.0b0__py3-none-any.whl

homeassistant-stubs/components/trafikverket_ferry/coordinator.pyi

@@ -1,7 +1,7 @@
+from . import TVFerryConfigEntry as TVFerryConfigEntry
 from .const import CONF_FROM as CONF_FROM, CONF_TIME as CONF_TIME, CONF_TO as CONF_TO, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from datetime import date
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_WEEKDAY as CONF_WEEKDAY, WEEKDAYS as WEEKDAYS
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
@@ -17,10 +17,11 @@ def next_weekday(fromdate: date, weekday: int) -> date: ...
 def next_departuredate(departure: list[str]) -> date: ...
 
 class TVDataUpdateCoordinator(DataUpdateCoordinator):
+    config_entry: TVFerryConfigEntry
     _ferry_api: Incomplete
     _from: Incomplete
     _to: Incomplete
     _time: Incomplete
     _weekdays: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+    def __init__(self, hass: HomeAssistant) -> None: ...
     async def _async_update_data(self) -> dict[str, Any]: ...

homeassistant-stubs/components/trafikverket_ferry/sensor.pyi

@@ -1,3 +1,4 @@
+from . import TVFerryConfigEntry as TVFerryConfigEntry
 from .const import ATTRIBUTION as ATTRIBUTION, DOMAIN as DOMAIN
 from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from _typeshed import Incomplete
@@ -5,7 +6,6 @@ from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from datetime import datetime
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_NAME as CONF_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -29,7 +29,7 @@ class TrafikverketSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TVFerryConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class FerrySensor(CoordinatorEntity[TVDataUpdateCoordinator], SensorEntity):
     entity_description: TrafikverketSensorEntityDescription

homeassistant-stubs/components/trafikverket_train/__init__.pyi

@@ -1,4 +1,4 @@
-from .const import CONF_FILTER_PRODUCT as CONF_FILTER_PRODUCT, CONF_FROM as CONF_FROM, CONF_TO as CONF_TO, DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
+from .const import CONF_FROM as CONF_FROM, CONF_TO as CONF_TO, PLATFORMS as PLATFORMS
 from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY
@@ -6,6 +6,8 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+TVTrainConfigEntry = ConfigEntry[TVDataUpdateCoordinator]
+
+async def async_setup_entry(hass: HomeAssistant, entry: TVTrainConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/trafikverket_train/coordinator.pyi

@@ -1,9 +1,9 @@
-from .const import CONF_TIME as CONF_TIME, DOMAIN as DOMAIN
+from . import TVTrainConfigEntry as TVTrainConfigEntry
+from .const import CONF_FILTER_PRODUCT as CONF_FILTER_PRODUCT, CONF_TIME as CONF_TIME, DOMAIN as DOMAIN
 from .util import next_departuredate as next_departuredate
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from datetime import datetime
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_WEEKDAY as CONF_WEEKDAY
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
@@ -34,11 +34,12 @@ def _get_as_utc(date_value: datetime | None) -> datetime | None: ...
 def _get_as_joined(information: list[str] | None) -> str | None: ...
 
 class TVDataUpdateCoordinator(DataUpdateCoordinator[TrainData]):
+    config_entry: TVTrainConfigEntry
     _train_api: Incomplete
     from_station: Incomplete
     to_station: Incomplete
     _time: Incomplete
     _weekdays: Incomplete
     _filter_product: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry, to_station: StationInfo, from_station: StationInfo, filter_product: str | None) -> None: ...
+    def __init__(self, hass: HomeAssistant, to_station: StationInfo, from_station: StationInfo) -> None: ...
     async def _async_update_data(self) -> TrainData: ...

homeassistant-stubs/components/trafikverket_train/sensor.pyi

@@ -1,3 +1,4 @@
+from . import TVTrainConfigEntry as TVTrainConfigEntry
 from .const import ATTRIBUTION as ATTRIBUTION, DOMAIN as DOMAIN
 from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator, TrainData as TrainData
 from _typeshed import Incomplete
@@ -5,7 +6,6 @@ from collections.abc import Callable as Callable, Mapping
 from dataclasses import dataclass
 from datetime import datetime
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_NAME as CONF_NAME, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -23,7 +23,7 @@ class TrafikverketSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TVTrainConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TrainSensor(CoordinatorEntity[TVDataUpdateCoordinator], SensorEntity):
     entity_description: TrafikverketSensorEntityDescription

homeassistant-stubs/components/trafikverket_weatherstation/__init__.pyi

@@ -1,7 +1,9 @@
-from .const import DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
+from .const import PLATFORMS as PLATFORMS
 from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+TVWeatherConfigEntry = ConfigEntry[TVDataUpdateCoordinator]
+
+async def async_setup_entry(hass: HomeAssistant, entry: TVWeatherConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/trafikverket_weatherstation/coordinator.pyi

@@ -1,6 +1,6 @@
+from . import TVWeatherConfigEntry as TVWeatherConfigEntry
 from .const import CONF_STATION as CONF_STATION, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
@@ -12,7 +12,8 @@ _LOGGER: Incomplete
 TIME_BETWEEN_UPDATES: Incomplete
 
 class TVDataUpdateCoordinator(DataUpdateCoordinator[WeatherStationInfo]):
+    config_entry: TVWeatherConfigEntry
     _weather_api: Incomplete
     _station: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+    def __init__(self, hass: HomeAssistant) -> None: ...
     async def _async_update_data(self) -> WeatherStationInfo: ...

homeassistant-stubs/components/trafikverket_weatherstation/sensor.pyi

@@ -1,3 +1,4 @@
+from . import TVWeatherConfigEntry as TVWeatherConfigEntry
 from .const import ATTRIBUTION as ATTRIBUTION, CONF_STATION as CONF_STATION, DOMAIN as DOMAIN
 from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from _typeshed import Incomplete
@@ -5,7 +6,6 @@ from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from datetime import datetime
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import DEGREE as DEGREE, PERCENTAGE as PERCENTAGE, UnitOfLength as UnitOfLength, UnitOfSpeed as UnitOfSpeed, UnitOfTemperature as UnitOfTemperature, UnitOfVolumetricFlux as UnitOfVolumetricFlux
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -25,7 +25,7 @@ def add_utc_timezone(date_time: datetime | None) -> datetime | None: ...
 
 SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TVWeatherConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TrafikverketWeatherStation(CoordinatorEntity[TVDataUpdateCoordinator], SensorEntity):
     entity_description: TrafikverketSensorEntityDescription

homeassistant-stubs/components/twentemilieu/__init__.pyi

@@ -1,16 +1,19 @@
 from .const import CONF_HOUSE_LETTER as CONF_HOUSE_LETTER, CONF_HOUSE_NUMBER as CONF_HOUSE_NUMBER, CONF_POST_CODE as CONF_POST_CODE, DOMAIN as DOMAIN, LOGGER as LOGGER
 from _typeshed import Incomplete
+from datetime import date
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ID as CONF_ID, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from twentemilieu import WasteType as WasteType
+from twentemilieu import WasteType
 
 SCAN_INTERVAL: Incomplete
 SERVICE_UPDATE: str
 SERVICE_SCHEMA: Incomplete
 PLATFORMS: Incomplete
+TwenteMilieuDataUpdateCoordinator = DataUpdateCoordinator[dict[WasteType, list[date]]]
+TwenteMilieuConfigEntry = ConfigEntry[TwenteMilieuDataUpdateCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/twentemilieu/calendar.pyi

@@ -1,16 +1,14 @@
-from .const import DOMAIN as DOMAIN, WASTE_TYPE_TO_DESCRIPTION as WASTE_TYPE_TO_DESCRIPTION
+from . import TwenteMilieuConfigEntry as TwenteMilieuConfigEntry
+from .const import WASTE_TYPE_TO_DESCRIPTION as WASTE_TYPE_TO_DESCRIPTION
 from .entity import TwenteMilieuEntity as TwenteMilieuEntity
 from _typeshed import Incomplete
-from datetime import date, datetime
+from datetime import datetime
 from homeassistant.components.calendar import CalendarEntity as CalendarEntity, CalendarEvent as CalendarEvent
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ID as CONF_ID
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from twentemilieu import WasteType as WasteType
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TwenteMilieuConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TwenteMilieuCalendar(TwenteMilieuEntity, CalendarEntity):
     _attr_has_entity_name: bool
@@ -18,7 +16,7 @@ class TwenteMilieuCalendar(TwenteMilieuEntity, CalendarEntity):
     _attr_translation_key: str
     _attr_unique_id: Incomplete
     _event: Incomplete
-    def __init__(self, coordinator: DataUpdateCoordinator[dict[WasteType, list[date]]], entry: ConfigEntry) -> None: ...
+    def __init__(self, entry: TwenteMilieuConfigEntry) -> None: ...
     @property
     def event(self) -> CalendarEvent | None: ...
     async def async_get_events(self, hass: HomeAssistant, start_date: datetime, end_date: datetime) -> list[CalendarEvent]: ...

homeassistant-stubs/components/twentemilieu/diagnostics.pyi

@@ -1,9 +1,5 @@
-from .const import DOMAIN as DOMAIN
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_ID as CONF_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from twentemilieu import WasteType as WasteType
 from typing import Any
 
 async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/twentemilieu/entity.pyi

@@ -1,14 +1,13 @@
+from . import TwenteMilieuDataUpdateCoordinator as TwenteMilieuDataUpdateCoordinator
 from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from datetime import date
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ID as CONF_ID
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity import Entity as Entity
-from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity, DataUpdateCoordinator as DataUpdateCoordinator
-from twentemilieu import WasteType
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 
-class TwenteMilieuEntity(CoordinatorEntity[DataUpdateCoordinator[dict[WasteType, list[date]]]], Entity):
+class TwenteMilieuEntity(CoordinatorEntity[TwenteMilieuDataUpdateCoordinator], Entity):
     _attr_has_entity_name: bool
     _attr_device_info: Incomplete
-    def __init__(self, coordinator: DataUpdateCoordinator[dict[WasteType, list[date]]], entry: ConfigEntry) -> None: ...
+    def __init__(self, entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/twentemilieu/sensor.pyi

@@ -8,7 +8,6 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ID as CONF_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
 from twentemilieu import WasteType
 
 @dataclass(frozen=True, kw_only=True)
@@ -23,6 +22,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 class TwenteMilieuSensor(TwenteMilieuEntity, SensorEntity):
     entity_description: TwenteMilieuSensorDescription
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: DataUpdateCoordinator[dict[WasteType, list[date]]], description: TwenteMilieuSensorDescription, entry: ConfigEntry) -> None: ...
+    def __init__(self, entry: ConfigEntry, description: TwenteMilieuSensorDescription) -> None: ...
     @property
     def native_value(self) -> date | None: ...

homeassistant-stubs/components/unifi/__init__.pyi

@@ -1,7 +1,7 @@
 from .const import PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
 from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
-from .services import async_setup_services as async_setup_services, async_unload_services as async_unload_services
+from .services import async_setup_services as async_setup_services
 from _typeshed import Incomplete
 from aiounifi.models.client import Client as Client
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -12,15 +12,16 @@ from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
 from homeassistant.helpers.storage import Store as Store
 from homeassistant.helpers.typing import ConfigType as ConfigType
 
+UnifiConfigEntry = ConfigEntry[UnifiHub]
 SAVE_DELAY: int
 STORAGE_KEY: str
 STORAGE_VERSION: int
 CONFIG_SCHEMA: Incomplete
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
-async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: ConfigEntry, device_entry: DeviceEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: UnifiConfigEntry, device_entry: DeviceEntry) -> bool: ...
 
 class UnifiWirelessClients:
     hass: Incomplete

homeassistant-stubs/components/unifi/button.pyi

@@ -1,12 +1,11 @@
 import aiounifi
+from . import UnifiConfigEntry as UnifiConfigEntry
 from .entity import HandlerT as HandlerT, 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, 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 aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonDeviceClass as ButtonDeviceClass, ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -23,7 +22,7 @@ class UnifiButtonEntityDescription(ButtonEntityDescription, UnifiEntityDescripti
 
 ENTITY_DESCRIPTIONS: tuple[UnifiButtonEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UnifiButtonEntity(UnifiEntity[HandlerT, ApiItemT], ButtonEntity):
     entity_description: UnifiButtonEntityDescription[HandlerT, ApiItemT]

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -1,3 +1,4 @@
+from . import UnifiConfigEntry as UnifiConfigEntry
 from .const import 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_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DOMAIN as UNIFI_DOMAIN
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
 from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
@@ -34,7 +35,7 @@ class UnifiOptionsFlowHandler(OptionsFlow):
     hub: UnifiHub
     config_entry: Incomplete
     options: Incomplete
-    def __init__(self, config_entry: ConfigEntry) -> None: ...
+    def __init__(self, config_entry: UnifiConfigEntry) -> None: ...
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_simple_options(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_configure_entity_sources(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -1,4 +1,5 @@
 import aiounifi
+from . import UnifiConfigEntry as UnifiConfigEntry
 from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn
 from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
@@ -9,7 +10,6 @@ from collections.abc import Callable as Callable, Mapping
 from dataclasses import dataclass
 from datetime import timedelta
 from homeassistant.components.device_tracker import DOMAIN as DOMAIN, ScannerEntity as ScannerEntity, SourceType as SourceType
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import Event as core_Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -40,8 +40,8 @@ class UnifiTrackerEntityDescription(UnifiEntityDescription[HandlerT, ApiItemT]):
 
 ENTITY_DESCRIPTIONS: tuple[UnifiTrackerEntityDescription, ...]
 
-def async_update_unique_id(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UnifiScannerEntity(UnifiEntity[HandlerT, ApiItemT], ScannerEntity):
     entity_description: UnifiTrackerEntityDescription

homeassistant-stubs/components/unifi/diagnostics.pyi

@@ -1,8 +1,7 @@
-from .hub import UnifiHub as UnifiHub
+from . import UnifiConfigEntry as UnifiConfigEntry
 from _typeshed import Incomplete
 from collections.abc import Mapping
 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_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import format_mac as format_mac
@@ -16,4 +15,4 @@ REDACT_WLANS: Incomplete
 
 def async_replace_dict_data(data: Mapping, to_replace: dict[str, str]) -> dict[str, Any]: ...
 def async_replace_list_data(data: list | set | tuple, to_replace: dict[str, str]) -> list[Any]: ...
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> dict[str, Any]: ...

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

@@ -6,7 +6,6 @@ from homeassistant.const import Platform as Platform
 from homeassistant.core import callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 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
 
 CHECK_HEARTBEAT_INTERVAL: Incomplete
 

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

@@ -1,4 +1,5 @@
 import aiounifi
+from .. import UnifiConfigEntry as UnifiConfigEntry
 from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_SITE_ID as CONF_SITE_ID, PLATFORMS as PLATFORMS
 from .config import UnifiConfig as UnifiConfig
 from .entity_helper import UnifiEntityHelper as UnifiEntityHelper
@@ -6,7 +7,6 @@ from .entity_loader import UnifiEntityLoader as UnifiEntityLoader
 from .websocket import UnifiWebsocket as UnifiWebsocket
 from _typeshed import Incomplete
 from datetime import datetime
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry, DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
@@ -20,9 +20,7 @@ class UnifiHub:
     websocket: Incomplete
     site: Incomplete
     is_admin: bool
-    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, api: aiounifi.Controller) -> None: ...
-    @staticmethod
-    def get_hub(hass: HomeAssistant, config_entry: ConfigEntry) -> UnifiHub: ...
+    def __init__(self, hass: HomeAssistant, config_entry: UnifiConfigEntry, api: aiounifi.Controller) -> None: ...
     @property
     def available(self) -> bool: ...
     @property
@@ -39,6 +37,6 @@ class UnifiHub:
     def device_info(self) -> DeviceInfo: ...
     def async_update_device_registry(self) -> DeviceEntry: ...
     @staticmethod
-    async def async_config_entry_updated(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
+    async def async_config_entry_updated(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> None: ...
     def shutdown(self, event: Event) -> None: ...
     async def async_reset(self) -> bool: ...

homeassistant-stubs/components/unifi/image.pyi

@@ -1,3 +1,4 @@
+from . import UnifiConfigEntry as UnifiConfigEntry
 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
@@ -7,7 +8,6 @@ from aiounifi.models.wlan import Wlan
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.image import ImageEntity as ImageEntity, ImageEntityDescription as ImageEntityDescription
-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.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -22,7 +22,7 @@ class UnifiImageEntityDescription(ImageEntityDescription, UnifiEntityDescription
 
 ENTITY_DESCRIPTIONS: tuple[UnifiImageEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UnifiImageEntity(UnifiEntity[HandlerT, ApiItemT], ImageEntity):
     entity_description: UnifiImageEntityDescription[HandlerT, ApiItemT]

homeassistant-stubs/components/unifi/sensor.pyi

@@ -1,3 +1,4 @@
+from . import UnifiConfigEntry as UnifiConfigEntry
 from .const import DEVICE_STATES as DEVICE_STATES
 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
@@ -12,7 +13,6 @@ from dataclasses import dataclass
 from datetime import date, datetime
 from decimal import Decimal
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass, UnitOfTemperature as UnitOfTemperature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfDataRate as UnitOfDataRate, UnitOfPower as UnitOfPower
 from homeassistant.core import Event as core_Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -42,7 +42,7 @@ class UnifiSensorEntityDescription(SensorEntityDescription, UnifiEntityDescripti
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UnifiSensorEntity(UnifiEntity[HandlerT, ApiItemT], SensorEntity):
     entity_description: UnifiSensorEntityDescription[HandlerT, ApiItemT]

homeassistant-stubs/components/unifi/services.pyi

@@ -12,6 +12,5 @@ SUPPORTED_SERVICES: Incomplete
 SERVICE_TO_SCHEMA: Incomplete
 
 def async_setup_services(hass: HomeAssistant) -> None: ...
-def async_unload_services(hass: HomeAssistant) -> None: ...
 async def async_reconnect_client(hass: HomeAssistant, data: Mapping[str, Any]) -> None: ...
 async def async_remove_clients(hass: HomeAssistant, data: Mapping[str, Any]) -> None: ...

homeassistant-stubs/components/unifi/switch.pyi

@@ -1,4 +1,5 @@
 import aiounifi
+from . import UnifiConfigEntry as UnifiConfigEntry
 from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER
 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
@@ -10,7 +11,6 @@ from aiounifi.models.event import Event as Event
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.switch import DOMAIN as DOMAIN, SwitchDeviceClass as SwitchDeviceClass, SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
-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.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -42,8 +42,8 @@ class UnifiSwitchEntityDescription(SwitchEntityDescription, UnifiEntityDescripti
 
 ENTITY_DESCRIPTIONS: tuple[UnifiSwitchEntityDescription, ...]
 
-def async_update_unique_id(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UnifiSwitchEntity(UnifiEntity[HandlerT, ApiItemT], SwitchEntity):
     entity_description: UnifiSwitchEntityDescription[HandlerT, ApiItemT]

homeassistant-stubs/components/unifi/update.pyi

@@ -1,6 +1,6 @@
 import aiounifi
+from . import UnifiConfigEntry as UnifiConfigEntry
 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
@@ -8,7 +8,6 @@ from aiounifi.models.device import Device
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.update import UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityDescription as UpdateEntityDescription, UpdateEntityFeature as UpdateEntityFeature
-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
 from typing import Any, TypeVar
@@ -27,7 +26,7 @@ class UnifiUpdateEntityDescription(UpdateEntityDescription, UnifiEntityDescripti
 
 ENTITY_DESCRIPTIONS: tuple[UnifiUpdateEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UnifiConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UnifiDeviceUpdateEntity(UnifiEntity[_HandlerT, _DataT], UpdateEntity):
     entity_description: UnifiUpdateEntityDescription[_HandlerT, _DataT]

homeassistant-stubs/components/unifiprotect/entity.pyi

@@ -31,7 +31,7 @@ class ProtectDeviceEntity(Entity):
     _attr_available: Incomplete
     def _async_update_device_from_protect(self, device: ProtectModelWithId) -> None: ...
     def _async_get_state_attrs(self) -> tuple[Any, ...]: ...
-    def _async_updated_event(self, device: ProtectModelWithId) -> None: ...
+    def _async_updated_event(self, device: ProtectAdoptableDeviceModel | NVR) -> None: ...
     async def async_added_to_hass(self) -> None: ...
 
 class ProtectNVREntity(ProtectDeviceEntity):

homeassistant-stubs/components/unifiprotect/migrate.pyi

@@ -8,7 +8,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity
 from pyunifiprotect import ProtectApiClient as ProtectApiClient
 from pyunifiprotect.data import Bootstrap as Bootstrap
-from typing_extensions import TypedDict
+from typing import TypedDict
 
 _LOGGER: Incomplete
 

homeassistant-stubs/components/upcloud/__init__.pyi

@@ -1,15 +1,15 @@
 import dataclasses
 import upcloud_api
 from .const import CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE as CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE, DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, DOMAIN as DOMAIN
+from .coordinator import UpCloudDataUpdateCoordinator as UpCloudDataUpdateCoordinator
 from _typeshed import Incomplete
-from datetime import timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_SCAN_INTERVAL as CONF_SCAN_INTERVAL, CONF_USERNAME as CONF_USERNAME, Platform as Platform, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, STATE_PROBLEM as STATE_PROBLEM
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers.device_registry import 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.update_coordinator import CoordinatorEntity as CoordinatorEntity, DataUpdateCoordinator as DataUpdateCoordinator
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from typing import Any
 
 _LOGGER: Incomplete
@@ -26,13 +26,6 @@ PLATFORMS: Incomplete
 SIGNAL_UPDATE_UPCLOUD: str
 STATE_MAP: Incomplete
 
-class UpCloudDataUpdateCoordinator(DataUpdateCoordinator[dict[str, upcloud_api.Server]]):
-    cloud_manager: Incomplete
-    def __init__(self, hass: HomeAssistant, *, cloud_manager: upcloud_api.CloudManager, update_interval: timedelta, username: str) -> None: ...
-    update_interval: Incomplete
-    async def async_update_config(self, config_entry: ConfigEntry) -> None: ...
-    async def _async_update_data(self) -> dict[str, upcloud_api.Server]: ...
-
 @dataclasses.dataclass
 class UpCloudHassData:
     coordinators: dict[str, UpCloudDataUpdateCoordinator] = ...

homeassistant-stubs/components/upcloud/coordinator.pyi

@@ -0,0 +1,16 @@
+import upcloud_api
+from _typeshed import Incomplete
+from datetime import timedelta
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_SCAN_INTERVAL as CONF_SCAN_INTERVAL
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+
+_LOGGER: Incomplete
+
+class UpCloudDataUpdateCoordinator(DataUpdateCoordinator[dict[str, upcloud_api.Server]]):
+    cloud_manager: Incomplete
+    def __init__(self, hass: HomeAssistant, *, cloud_manager: upcloud_api.CloudManager, update_interval: timedelta, username: str) -> None: ...
+    update_interval: Incomplete
+    async def async_update_config(self, config_entry: ConfigEntry) -> None: ...
+    async def _async_update_data(self) -> dict[str, upcloud_api.Server]: ...

homeassistant-stubs/components/vacuum/__init__.pyi

@@ -1,6 +1,5 @@
-from .const import STATE_CLEANING as STATE_CLEANING, STATE_DOCKED as STATE_DOCKED, STATE_ERROR as STATE_ERROR, STATE_RETURNING as STATE_RETURNING
+from .const import DOMAIN as DOMAIN, STATE_CLEANING as STATE_CLEANING, STATE_DOCKED as STATE_DOCKED, STATE_ERROR as STATE_ERROR, STATE_RETURNING as STATE_RETURNING
 from _typeshed import Incomplete
-from collections.abc import Mapping
 from enum import IntFlag
 from functools import cached_property as cached_property
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -15,7 +14,6 @@ from homeassistant.loader import bind_hass as bind_hass
 from typing import Any
 
 _LOGGER: Incomplete
-DOMAIN: str
 ENTITY_ID_FORMAT: Incomplete
 SCAN_INTERVAL: Incomplete
 ATTR_BATTERY_ICON: str
@@ -91,7 +89,7 @@ class StateVacuumEntity(Entity, cached_properties=STATE_VACUUM_CACHED_PROPERTIES
     @property
     def battery_icon(self) -> str: ...
     @property
-    def capability_attributes(self) -> Mapping[str, Any] | None: ...
+    def capability_attributes(self) -> dict[str, Any] | None: ...
     @cached_property
     def fan_speed(self) -> str | None: ...
     @cached_property

homeassistant-stubs/components/vacuum/const.pyi

@@ -1,5 +1,6 @@
 from _typeshed import Incomplete
 
+DOMAIN: str
 STATE_CLEANING: str
 STATE_DOCKED: str
 STATE_RETURNING: str