homeassistant-stubs 2024.3.3__py3-none-any.whl → 2024.4.0b0__py3-none-any.whl

homeassistant-stubs/components/trafikverket_ferry/sensor.pyi

@@ -19,18 +19,13 @@ ATTR_FROM: str
 ATTR_TO: str
 ATTR_MODIFIED_TIME: str
 ATTR_OTHER_INFO: str
-ICON: str
 SCAN_INTERVAL: Incomplete
 
-@dataclass(frozen=True)
-class TrafikverketRequiredKeysMixin:
+@dataclass(frozen=True, kw_only=True)
+class TrafikverketSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[dict[str, Any]], StateType | datetime]
     info_fn: Callable[[dict[str, Any]], StateType | list] | None
-    def __init__(self, value_fn, info_fn) -> None: ...
-
-@dataclass(frozen=True)
-class TrafikverketSensorEntityDescription(SensorEntityDescription, TrafikverketRequiredKeysMixin):
-    def __init__(self, value_fn, info_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) -> None: ...
+    def __init__(self, *, 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, value_fn, info_fn) -> None: ...
 
 SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...]
 

homeassistant-stubs/components/trafikverket_train/config_flow.pyi

@@ -2,10 +2,9 @@ from .const import CONF_FILTER_PRODUCT as CONF_FILTER_PRODUCT, CONF_FROM as CONF
 from .util import create_unique_id as create_unique_id, next_departuredate as next_departuredate
 from _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant import config_entries as config_entries
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlowWithConfigEntry as OptionsFlowWithConfigEntry
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_NAME as CONF_NAME, CONF_WEEKDAY as CONF_WEEKDAY, WEEKDAYS as WEEKDAYS
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.selector import SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TextSelector as TextSelector, TimeSelector as TimeSelector
 from typing import Any
@@ -17,14 +16,14 @@ DATA_SCHEMA_REAUTH: Incomplete
 
 async def validate_input(hass: HomeAssistant, api_key: str, train_from: str, train_to: str, train_time: str | None, weekdays: list[str], product_filter: str | None) -> dict[str, str]: ...
 
-class TVTrainConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
+class TVTrainConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    entry: config_entries.ConfigEntry | None
+    entry: ConfigEntry | None
     @staticmethod
-    def async_get_options_flow(config_entry: config_entries.ConfigEntry) -> TVTrainOptionsFlowHandler: ...
-    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult: ...
-    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    def async_get_options_flow(config_entry: ConfigEntry) -> TVTrainOptionsFlowHandler: ...
+    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
+    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
 
-class TVTrainOptionsFlowHandler(config_entries.OptionsFlowWithConfigEntry):
-    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+class TVTrainOptionsFlowHandler(OptionsFlowWithConfigEntry):
+    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/trafikverket_train/sensor.pyi

@@ -16,14 +16,10 @@ from typing import Any
 
 ATTR_PRODUCT_FILTER: str
 
-@dataclass(frozen=True)
-class TrafikverketRequiredKeysMixin:
+@dataclass(frozen=True, kw_only=True)
+class TrafikverketSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[TrainData], StateType | datetime]
-    def __init__(self, value_fn) -> None: ...
-
-@dataclass(frozen=True)
-class TrafikverketSensorEntityDescription(SensorEntityDescription, TrafikverketRequiredKeysMixin):
-    def __init__(self, value_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) -> None: ...
+    def __init__(self, *, 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, value_fn) -> None: ...
 
 SENSOR_TYPES: tuple[TrafikverketSensorEntityDescription, ...]
 

homeassistant-stubs/components/trafikverket_weatherstation/config_flow.pyi

@@ -1,15 +1,14 @@
 from .const import CONF_STATION as CONF_STATION, DOMAIN as DOMAIN
 from collections.abc import Mapping
-from homeassistant import config_entries as config_entries
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY
-from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from typing import Any
 
-class TVWeatherConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
+class TVWeatherConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    entry: config_entries.ConfigEntry | None
+    entry: ConfigEntry | None
     async def validate_input(self, sensor_api: str, station: str) -> None: ...
-    async def async_step_user(self, user_input: dict[str, str] | None = None) -> FlowResult: ...
-    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult: ...
-    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def async_step_user(self, user_input: dict[str, str] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
+    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/trafikverket_weatherstation/sensor.pyi

@@ -16,14 +16,10 @@ from pytrafikverket.trafikverket_weather import WeatherStationInfo as WeatherSta
 
 PRECIPITATION_TYPE: Incomplete
 
-@dataclass(frozen=True)
-class TrafikverketRequiredKeysMixin:
+@dataclass(frozen=True, kw_only=True)
+class TrafikverketSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[WeatherStationInfo], StateType | datetime]
-    def __init__(self, value_fn) -> None: ...
-
-@dataclass(frozen=True)
-class TrafikverketSensorEntityDescription(SensorEntityDescription, TrafikverketRequiredKeysMixin):
-    def __init__(self, value_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) -> None: ...
+    def __init__(self, *, 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, value_fn) -> None: ...
 
 def add_utc_timezone(date_time: datetime | None) -> datetime | None: ...
 

homeassistant-stubs/components/transmission/__init__.pyi

@@ -1,10 +1,10 @@
 import transmission_rpc
-from .const import ATTR_DELETE_DATA as ATTR_DELETE_DATA, ATTR_TORRENT as ATTR_TORRENT, CONF_ENTRY_ID as CONF_ENTRY_ID, DEFAULT_DELETE_DATA as DEFAULT_DELETE_DATA, DOMAIN as DOMAIN, SERVICE_ADD_TORRENT as SERVICE_ADD_TORRENT, SERVICE_REMOVE_TORRENT as SERVICE_REMOVE_TORRENT, SERVICE_START_TORRENT as SERVICE_START_TORRENT, SERVICE_STOP_TORRENT as SERVICE_STOP_TORRENT
+from .const import ATTR_DELETE_DATA as ATTR_DELETE_DATA, ATTR_TORRENT as ATTR_TORRENT, CONF_ENTRY_ID as CONF_ENTRY_ID, DEFAULT_DELETE_DATA as DEFAULT_DELETE_DATA, DEFAULT_PATH as DEFAULT_PATH, DEFAULT_SSL as DEFAULT_SSL, DOMAIN as DOMAIN, SERVICE_ADD_TORRENT as SERVICE_ADD_TORRENT, SERVICE_REMOVE_TORRENT as SERVICE_REMOVE_TORRENT, SERVICE_START_TORRENT as SERVICE_START_TORRENT, SERVICE_STOP_TORRENT as SERVICE_STOP_TORRENT
 from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from .errors import AuthenticationError as AuthenticationError, CannotConnect as CannotConnect, UnknownError as UnknownError
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_ID as CONF_ID, CONF_NAME as CONF_NAME, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, Platform as Platform
+from homeassistant.const import CONF_HOST as CONF_HOST, CONF_ID as CONF_ID, CONF_NAME as CONF_NAME, CONF_PASSWORD as CONF_PASSWORD, CONF_PATH as CONF_PATH, CONF_PORT as CONF_PORT, CONF_SSL as CONF_SSL, CONF_USERNAME as CONF_USERNAME, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers import selector as selector
@@ -21,4 +21,5 @@ SERVICE_STOP_TORRENT_SCHEMA: Incomplete
 
 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_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
 async def get_api(hass: HomeAssistant, entry: dict[str, Any]) -> transmission_rpc.Client: ...

homeassistant-stubs/components/transmission/config_flow.pyi

@@ -1,26 +1,26 @@
 from . import get_api as get_api
-from .const import CONF_LIMIT as CONF_LIMIT, CONF_ORDER as CONF_ORDER, DEFAULT_LIMIT as DEFAULT_LIMIT, DEFAULT_NAME as DEFAULT_NAME, DEFAULT_ORDER as DEFAULT_ORDER, DEFAULT_PORT as DEFAULT_PORT, DOMAIN as DOMAIN, SUPPORTED_ORDER_MODES as SUPPORTED_ORDER_MODES
+from .const import CONF_LIMIT as CONF_LIMIT, CONF_ORDER as CONF_ORDER, DEFAULT_LIMIT as DEFAULT_LIMIT, DEFAULT_NAME as DEFAULT_NAME, DEFAULT_ORDER as DEFAULT_ORDER, DEFAULT_PATH as DEFAULT_PATH, DEFAULT_PORT as DEFAULT_PORT, DEFAULT_SSL as DEFAULT_SSL, DOMAIN as DOMAIN, SUPPORTED_ORDER_MODES as SUPPORTED_ORDER_MODES
 from .errors import AuthenticationError as AuthenticationError, CannotConnect as CannotConnect, UnknownError as UnknownError
 from _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant import config_entries as config_entries
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
+from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PATH as CONF_PATH, CONF_PORT as CONF_PORT, CONF_SSL as CONF_SSL, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import callback as callback
-from homeassistant.data_entry_flow import FlowResult as FlowResult
 from typing import Any
 
 DATA_SCHEMA: Incomplete
 
-class TransmissionFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
+class TransmissionFlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    _reauth_entry: config_entries.ConfigEntry | None
+    MINOR_VERSION: int
+    _reauth_entry: ConfigEntry | None
     @staticmethod
-    def async_get_options_flow(config_entry: config_entries.ConfigEntry) -> TransmissionOptionsFlowHandler: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult: ...
-    async def async_step_reauth_confirm(self, user_input: dict[str, str] | None = None) -> FlowResult: ...
+    def async_get_options_flow(config_entry: ConfigEntry) -> TransmissionOptionsFlowHandler: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
+    async def async_step_reauth_confirm(self, user_input: dict[str, str] | None = None) -> ConfigFlowResult: ...
 
-class TransmissionOptionsFlowHandler(config_entries.OptionsFlow):
+class TransmissionOptionsFlowHandler(OptionsFlow):
     config_entry: Incomplete
-    def __init__(self, config_entry: config_entries.ConfigEntry) -> None: ...
-    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    def __init__(self, config_entry: ConfigEntry) -> None: ...
+    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/transmission/const.pyi

@@ -14,7 +14,9 @@ DEFAULT_DELETE_DATA: bool
 DEFAULT_LIMIT: int
 DEFAULT_ORDER = ORDER_OLDEST_FIRST
 DEFAULT_NAME: str
+DEFAULT_SSL: bool
 DEFAULT_PORT: int
+DEFAULT_PATH: str
 DEFAULT_SCAN_INTERVAL: int
 STATE_ATTR_TORRENT_INFO: str
 ATTR_DELETE_DATA: str

homeassistant-stubs/components/transmission/switch.pyi

@@ -13,16 +13,12 @@ from typing import Any
 
 _LOGGING: Incomplete
 
-@dataclass(frozen=True)
-class TransmissionSwitchEntityDescriptionMixin:
+@dataclass(frozen=True, kw_only=True)
+class TransmissionSwitchEntityDescription(SwitchEntityDescription):
     is_on_func: Callable[[TransmissionDataUpdateCoordinator], bool | None]
     on_func: Callable[[TransmissionDataUpdateCoordinator], None]
     off_func: Callable[[TransmissionDataUpdateCoordinator], None]
-    def __init__(self, is_on_func, on_func, off_func) -> None: ...
-
-@dataclass(frozen=True)
-class TransmissionSwitchEntityDescription(SwitchEntityDescription, TransmissionSwitchEntityDescriptionMixin):
-    def __init__(self, is_on_func, on_func, off_func, *, 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) -> None: ...
+    def __init__(self, *, 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, is_on_func, on_func, off_func) -> None: ...
 
 SWITCH_TYPES: tuple[TransmissionSwitchEntityDescription, ...]
 

homeassistant-stubs/components/trend/binary_sensor.pyi

@@ -5,13 +5,13 @@ from collections.abc import Mapping
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, ENTITY_ID_FORMAT as ENTITY_ID_FORMAT, PLATFORM_SCHEMA as PLATFORM_SCHEMA
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME as ATTR_FRIENDLY_NAME, CONF_ATTRIBUTE as CONF_ATTRIBUTE, CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_FRIENDLY_NAME as CONF_FRIENDLY_NAME, CONF_SENSORS as CONF_SENSORS, STATE_ON as STATE_ON, STATE_UNAVAILABLE as STATE_UNAVAILABLE, STATE_UNKNOWN as STATE_UNKNOWN
-from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity import generate_entity_id as generate_entity_id
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.event import EventStateChangedData as EventStateChangedData, async_track_state_change_event as async_track_state_change_event
 from homeassistant.helpers.reload import async_setup_reload_service as async_setup_reload_service
 from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
-from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType, EventType as EventType
+from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from homeassistant.util.dt import utcnow as utcnow
 from typing import Any
 

homeassistant-stubs/components/tts/__init__.pyi

@@ -62,6 +62,7 @@ class SpeechManager:
     file_cache: Incomplete
     mem_cache: Incomplete
     def __init__(self, hass: HomeAssistant, use_cache: bool, cache_dir: str, time_memory: int) -> None: ...
+    def _init_cache(self) -> dict[str, str]: ...
     async def async_init_cache(self) -> None: ...
     async def async_clear_cache(self) -> None: ...
     def async_register_legacy_engine(self, engine: str, provider: Provider, config: ConfigType) -> None: ...

homeassistant-stubs/components/tts/legacy.pyi

@@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as Se
 from homeassistant.helpers import discovery as discovery
 from homeassistant.helpers.service import async_set_service_schema as async_set_service_schema
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
-from homeassistant.setup import async_prepare_setup_platform as async_prepare_setup_platform
+from homeassistant.setup import SetupPhases as SetupPhases, async_prepare_setup_platform as async_prepare_setup_platform, async_start_setup as async_start_setup
 from homeassistant.util.yaml import load_yaml_dict as load_yaml_dict
 from typing import Any
 

homeassistant-stubs/components/twentemilieu/calendar.pyi

@@ -14,8 +14,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
 
 class TwenteMilieuCalendar(TwenteMilieuEntity, CalendarEntity):
     _attr_has_entity_name: bool
-    _attr_icon: str
     _attr_name: Incomplete
+    _attr_translation_key: str
     _attr_unique_id: Incomplete
     _event: Incomplete
     def __init__(self, coordinator: DataUpdateCoordinator[dict[WasteType, list[date]]], entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/twentemilieu/config_flow.pyi

@@ -1,11 +1,10 @@
 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
-from homeassistant.config_entries import ConfigFlow as ConfigFlow
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import CONF_ID as CONF_ID
-from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from typing import Any
 
 class TwenteMilieuFlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    async def _show_setup_form(self, errors: dict[str, str] | None = None) -> FlowResult: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
+    async def _show_setup_form(self, errors: dict[str, str] | None = None) -> ConfigFlowResult: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/unifi/button.pyi

@@ -10,19 +10,15 @@ 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
-from typing import Any, Generic
+from typing import Any
 
 async def async_restart_device_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
 async def async_power_cycle_port_control_fn(api: aiounifi.Controller, obj_id: str) -> None: ...
 
-@dataclass(frozen=True)
-class UnifiButtonEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
+@dataclass(frozen=True, kw_only=True)
+class UnifiButtonEntityDescription(ButtonEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     control_fn: Callable[[aiounifi.Controller, str], Coroutine[Any, Any, None]]
-    def __init__(self, control_fn) -> None: ...
-
-@dataclass(frozen=True)
-class UnifiButtonEntityDescription(ButtonEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT], UnifiButtonEntityDescriptionMixin[HandlerT, ApiItemT]):
-    def __init__(self, control_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) -> None: ...
+    def __init__(self, *, 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, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, control_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiButtonEntityDescription, ...]
 

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -4,11 +4,10 @@ from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
 from _typeshed import Incomplete
 from aiounifi.interfaces.sites import Sites as Sites
 from collections.abc import Mapping
-from homeassistant import config_entries as config_entries
 from homeassistant.components import ssdp as ssdp
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
 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, callback as callback
-from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.helpers.device_registry import format_mac as format_mac
 from typing import Any
 
@@ -17,31 +16,31 @@ DEFAULT_SITE_ID: str
 DEFAULT_VERIFY_SSL: bool
 MODEL_PORTS: Incomplete
 
-class UnifiFlowHandler(config_entries.ConfigFlow, domain=UNIFI_DOMAIN):
+class UnifiFlowHandler(ConfigFlow, domain=UNIFI_DOMAIN):
     VERSION: int
     sites: Sites
     @staticmethod
-    def async_get_options_flow(config_entry: config_entries.ConfigEntry) -> UnifiOptionsFlowHandler: ...
+    def async_get_options_flow(config_entry: ConfigEntry) -> UnifiOptionsFlowHandler: ...
     config: Incomplete
     reauth_config_entry: Incomplete
     reauth_schema: Incomplete
     def __init__(self) -> None: ...
-    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_site(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult: ...
-    async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> FlowResult: ...
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_site(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
+    async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> ConfigFlowResult: ...
 
-class UnifiOptionsFlowHandler(config_entries.OptionsFlow):
+class UnifiOptionsFlowHandler(OptionsFlow):
     hub: UnifiHub
     config_entry: Incomplete
     options: Incomplete
-    def __init__(self, config_entry: config_entries.ConfigEntry) -> None: ...
-    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_simple_options(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_configure_entity_sources(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_device_tracker(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_client_control(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def async_step_statistics_sensors(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
-    async def _update_options(self) -> FlowResult: ...
+    def __init__(self, config_entry: ConfigEntry) -> 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: ...
+    async def async_step_device_tracker(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_client_control(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_statistics_sensors(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def _update_options(self) -> ConfigFlowResult: ...
 
 async def _async_discover_unifi(hass: HomeAssistant) -> str | None: ...

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -13,7 +13,7 @@ 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
-from typing import Any, Generic
+from typing import Any
 
 LOGGER: Incomplete
 CLIENT_TRACKER: str
@@ -30,17 +30,13 @@ def async_client_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_client_is_connected_fn(hub: UnifiHub, obj_id: str) -> bool: ...
 def async_device_heartbeat_timedelta_fn(hub: UnifiHub, obj_id: str) -> timedelta: ...
 
-@dataclass(frozen=True)
-class UnifiEntityTrackerDescriptionMixin(Generic[HandlerT, ApiItemT]):
+@dataclass(frozen=True, kw_only=True)
+class UnifiTrackerEntityDescription(UnifiEntityDescription[HandlerT, ApiItemT]):
     heartbeat_timedelta_fn: Callable[[UnifiHub, str], timedelta]
     ip_address_fn: Callable[[aiounifi.Controller, str], str | None]
     is_connected_fn: Callable[[UnifiHub, str], bool]
     hostname_fn: Callable[[aiounifi.Controller, str], str | None]
-    def __init__(self, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn) -> None: ...
-
-@dataclass(frozen=True)
-class UnifiTrackerEntityDescription(UnifiEntityDescription[HandlerT, ApiItemT], UnifiEntityTrackerDescriptionMixin[HandlerT, ApiItemT]):
-    def __init__(self, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_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) -> None: ...
+    def __init__(self, *, 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, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiTrackerEntityDescription, ...]
 

homeassistant-stubs/components/unifi/entity.pyi

@@ -5,7 +5,7 @@ from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from abc import abstractmethod
 from aiounifi.interfaces.api_handlers import APIHandler, CallbackType, ItemEvent, UnsubscribeType
-from aiounifi.models.api import ApiItemT
+from aiounifi.models.api import ApiItemT as ApiItemT
 from aiounifi.models.event import Event as Event, EventKey as EventKey
 from collections.abc import Callable
 from dataclasses import dataclass
@@ -24,24 +24,21 @@ 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[[UnifiHub, str], bool]
+@dataclass(frozen=True, kw_only=True)
+class UnifiEntityDescription(EntityDescription, Generic[HandlerT, ApiItemT]):
     api_handler_fn: Callable[[aiounifi.Controller], HandlerT]
-    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[[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)
-class UnifiEntityDescription(EntityDescription, UnifiDescription[HandlerT, ApiItemT]):
-    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, *, 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) -> None: ...
+    allowed_fn: Callable[[UnifiHub, str], bool] = ...
+    available_fn: Callable[[UnifiHub, str], bool] = ...
+    name_fn: Callable[[ApiItemT], str | None] = ...
+    supported_fn: Callable[[UnifiHub, str], bool] = ...
+    has_entity_name = ...
+    event_is_on: tuple[EventKey, ...] | None = ...
+    event_to_subscribe: tuple[EventKey, ...] | None = ...
+    should_poll: bool = ...
+    def __init__(self, *, 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, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll) -> None: ...
 
 class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
     entity_description: UnifiEntityDescription[HandlerT, ApiItemT]

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

@@ -0,0 +1,31 @@
+import ssl
+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_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
+from dataclasses import dataclass
+from datetime import timedelta
+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
+from typing import Literal, Self
+
+@dataclass
+class UnifiConfig:
+    entry: ConfigEntry
+    host: str
+    port: int
+    username: str
+    password: str
+    site: str
+    ssl_context: ssl.SSLContext | Literal[False]
+    option_supported_clients: list[str]
+    option_track_clients: list[str]
+    option_track_wired_clients: list[str]
+    option_track_devices: bool
+    option_ssid_filter: set[str]
+    option_detection_time: timedelta
+    option_ignore_wired_bug: bool
+    option_block_clients: list[str]
+    option_dpi_restrictions: bool
+    option_allow_bandwidth_sensors: bool
+    option_allow_uptime_sensors: bool
+    @classmethod
+    def from_config_entry(cls, config_entry: ConfigEntry) -> Self: ...
+    def __init__(self, entry, host, port, username, password, site, ssl_context, option_supported_clients, option_track_clients, option_track_wired_clients, option_track_devices, option_ssid_filter, option_detection_time, option_ignore_wired_bug, option_block_clients, option_dpi_restrictions, option_allow_bandwidth_sensors, option_allow_uptime_sensors) -> None: ...

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

@@ -0,0 +1,26 @@
+from ..const import LOGGER as LOGGER, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
+from ..entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
+from .hub import UnifiHub as UnifiHub
+from _typeshed import Incomplete
+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
+
+class UnifiEntityLoader:
+    hub: Incomplete
+    api_updaters: Incomplete
+    wireless_clients: Incomplete
+    platforms: Incomplete
+    known_objects: Incomplete
+    def __init__(self, hub: UnifiHub) -> None: ...
+    async def initialize(self) -> None: ...
+    async def _refresh_api_data(self) -> None: ...
+    def _restore_inactive_clients(self) -> None: ...
+    def register_platform(self, async_add_entities: AddEntitiesCallback, entity_class: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], requires_admin: bool = False) -> None: ...
+    def load_entities(self) -> None: ...
+    def _should_add_entity(self, description: UnifiEntityDescription, obj_id: str) -> bool: ...
+    def _load_entities(self, unifi_platform_entity: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], async_add_entities: AddEntitiesCallback) -> None: ...

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

@@ -1,55 +1,35 @@
 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, PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
-from ..entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
+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_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.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.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.dispatcher import async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.event import async_call_later as async_call_later, async_track_time_interval as async_track_time_interval
 
 CHECK_HEARTBEAT_INTERVAL: Incomplete
 
 class UnifiHub:
     hass: Incomplete
-    config_entry: Incomplete
     api: Incomplete
+    config: Incomplete
+    entity_loader: Incomplete
     websocket: Incomplete
-    wireless_clients: Incomplete
     site: Incomplete
     is_admin: bool
     _cancel_heartbeat_check: Incomplete
     _heartbeat_time: Incomplete
-    entities: Incomplete
-    known_objects: Incomplete
     poe_command_queue: Incomplete
     _cancel_poe_command: Incomplete
     def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, api: aiounifi.Controller) -> None: ...
-    option_supported_clients: Incomplete
-    option_track_clients: Incomplete
-    option_track_wired_clients: Incomplete
-    option_track_devices: Incomplete
-    option_ssid_filter: Incomplete
-    option_detection_time: Incomplete
-    option_ignore_wired_bug: Incomplete
-    option_block_clients: Incomplete
-    option_dpi_restrictions: Incomplete
-    option_allow_bandwidth_sensors: Incomplete
-    option_allow_uptime_sensors: Incomplete
-    def load_config_entry_options(self) -> None: ...
-    @property
-    def host(self) -> str: ...
+    @staticmethod
+    def get_hub(hass: HomeAssistant, config_entry: ConfigEntry) -> UnifiHub: ...
     @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: ...
     @property

homeassistant-stubs/components/unifi/image.pyi

@@ -11,19 +11,14 @@ 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
-from typing import Generic
 
 def async_wlan_qr_code_image_fn(hub: UnifiHub, wlan: Wlan) -> bytes: ...
 
-@dataclass(frozen=True)
-class UnifiImageEntityDescriptionMixin(Generic[HandlerT, ApiItemT]):
+@dataclass(frozen=True, kw_only=True)
+class UnifiImageEntityDescription(ImageEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT]):
     image_fn: Callable[[UnifiHub, ApiItemT], bytes]
     value_fn: Callable[[ApiItemT], str | None]
-    def __init__(self, image_fn, value_fn) -> None: ...
-
-@dataclass(frozen=True)
-class UnifiImageEntityDescription(ImageEntityDescription, UnifiEntityDescription[HandlerT, ApiItemT], UnifiImageEntityDescriptionMixin[HandlerT, ApiItemT]):
-    def __init__(self, image_fn, 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) -> None: ...
+    def __init__(self, *, 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, api_handler_fn, device_info_fn, object_fn, unique_id_fn, allowed_fn, available_fn, name_fn, supported_fn, event_is_on, event_to_subscribe, should_poll, image_fn, value_fn) -> None: ...
 
 ENTITY_DESCRIPTIONS: tuple[UnifiImageEntityDescription, ...]