homeassistant-stubs 2024.6.4__py3-none-any.whl → 2024.7.0__py3-none-any.whl

homeassistant-stubs/components/tplink/entity.pyi

@@ -1,18 +1,60 @@
-from .const import DOMAIN as DOMAIN
+import abc
+from . import get_device_name as get_device_name, legacy_device_id as legacy_device_id
+from .const import ATTR_CURRENT_A as ATTR_CURRENT_A, ATTR_CURRENT_POWER_W as ATTR_CURRENT_POWER_W, ATTR_TODAY_ENERGY_KWH as ATTR_TODAY_ENERGY_KWH, ATTR_TOTAL_ENERGY_KWH as ATTR_TOTAL_ENERGY_KWH, DOMAIN as DOMAIN, PRIMARY_STATE_ID as PRIMARY_STATE_ID
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
 from _typeshed import Incomplete
-from collections.abc import Awaitable as Awaitable, Callable as Callable, Coroutine
+from abc import ABC, abstractmethod
+from collections.abc import Awaitable as Awaitable, Callable as Callable, Coroutine, Mapping
+from dataclasses import dataclass
+from homeassistant.const import EntityCategory as EntityCategory
+from homeassistant.core import callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity import EntityDescription as EntityDescription
+from homeassistant.helpers.typing import UNDEFINED as UNDEFINED, UndefinedType as UndefinedType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from kasa import SmartDevice as SmartDevice
+from kasa import Device, Feature
 from typing import Any, Concatenate
 
+_LOGGER: Incomplete
+FEATURE_CATEGORY_TO_ENTITY_CATEGORY: Incomplete
+DEVICETYPES_WITH_SPECIALIZED_PLATFORMS: Incomplete
+FEATURES_ALLOW_LIST: Incomplete
+EXCLUDED_FEATURES: Incomplete
+LEGACY_KEY_MAPPING: Incomplete
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkFeatureEntityDescription(EntityDescription):
+    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) -> None: ...
+
 def async_refresh_after(func: Callable[Concatenate[_T, _P], Awaitable[None]]) -> Callable[Concatenate[_T, _P], Coroutine[Any, Any, None]]: ...
 
-class CoordinatedTPLinkEntity(CoordinatorEntity[TPLinkDataUpdateCoordinator]):
+class CoordinatedTPLinkEntity(CoordinatorEntity[TPLinkDataUpdateCoordinator], ABC, metaclass=abc.ABCMeta):
     _attr_has_entity_name: bool
-    device: Incomplete
-    _attr_unique_id: Incomplete
+    _device: Device
+    _feature: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
+    _attr_unique_id: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature: Feature | None = None, parent: Device | None = None) -> None: ...
+    def _get_unique_id(self) -> str: ...
+    @abstractmethod
+    def _async_update_attrs(self) -> None: ...
+    _attr_available: bool
+    def _async_call_update_attrs(self) -> None: ...
+    def _handle_coordinator_update(self) -> None: ...
+    @property
+    def available(self) -> bool: ...
+
+class CoordinatedTPLinkFeatureEntity(CoordinatedTPLinkEntity, ABC, metaclass=abc.ABCMeta):
+    entity_description: TPLinkFeatureEntityDescription
+    _feature: Feature
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature: Feature, description: TPLinkFeatureEntityDescription, parent: Device | None = None) -> None: ...
+    def _get_unique_id(self) -> str: ...
+    @classmethod
+    def _category_for_feature(cls, feature: Feature | None) -> EntityCategory | None: ...
+    @classmethod
+    def _description_for_feature(cls, feature: Feature, descriptions: Mapping[str, _D], *, device: Device, parent: Device | None = None) -> _D | None: ...
+    @classmethod
+    def _entities_for_device(cls, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature_type: Feature.Type, entity_class: type[_E], descriptions: Mapping[str, _D], parent: Device | None = None) -> list[_E]: ...
+    @classmethod
+    def entities_for_device_and_its_children(cls, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature_type: Feature.Type, entity_class: type[_E], descriptions: Mapping[str, _D], child_coordinators: list[TPLinkDataUpdateCoordinator] | None = None) -> list[_E]: ...

homeassistant-stubs/components/tplink/fan.pyi

@@ -0,0 +1,31 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
+from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity, async_refresh_after as async_refresh_after
+from _typeshed import Incomplete
+from homeassistant.components.fan import FanEntity as FanEntity, FanEntityFeature as FanEntityFeature
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.util.percentage import percentage_to_ranged_value as percentage_to_ranged_value, ranged_value_to_percentage as ranged_value_to_percentage
+from homeassistant.util.scaling import int_states_in_range as int_states_in_range
+from kasa import Device as Device
+from kasa.interfaces import Fan as FanInterface
+from typing import Any
+
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+SPEED_RANGE: Incomplete
+
+class TPLinkFanEntity(CoordinatedTPLinkEntity, FanEntity):
+    _attr_speed_count: Incomplete
+    _attr_supported_features: Incomplete
+    fan_module: Incomplete
+    _attr_name: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, fan_module: FanInterface, parent: Device | None = None) -> None: ...
+    async def async_turn_on(self, percentage: int | None = None, preset_mode: str | None = None, **kwargs: Any) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_set_percentage(self, percentage: int) -> None: ...
+    _attr_is_on: Incomplete
+    _attr_percentage: Incomplete
+    def _async_update_attrs(self) -> None: ...

homeassistant-stubs/components/tplink/light.pyi

@@ -1,17 +1,16 @@
-from . import legacy_device_id as legacy_device_id
-from .const import DOMAIN as DOMAIN
+from . import TPLinkConfigEntry as TPLinkConfigEntry, legacy_device_id as legacy_device_id
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
 from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity, async_refresh_after as async_refresh_after
-from .models import TPLinkData as TPLinkData
 from _typeshed import Incomplete
 from collections.abc import Sequence
-from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP_KELVIN as ATTR_COLOR_TEMP_KELVIN, ATTR_EFFECT as ATTR_EFFECT, ATTR_HS_COLOR as ATTR_HS_COLOR, ATTR_TRANSITION as ATTR_TRANSITION, 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.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP_KELVIN as ATTR_COLOR_TEMP_KELVIN, ATTR_EFFECT as ATTR_EFFECT, ATTR_HS_COLOR as ATTR_HS_COLOR, ATTR_TRANSITION as ATTR_TRANSITION, ColorMode as ColorMode, EFFECT_OFF as EFFECT_OFF, LightEntity as LightEntity, LightEntityFeature as LightEntityFeature, filter_supported_color_modes as filter_supported_color_modes
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from kasa import SmartBulb, SmartLightStrip
-from typing import Any, Final
+from homeassistant.helpers.typing import VolDictType as VolDictType
+from kasa import Device as Device
+from kasa.interfaces import Light as Light, LightEffect
+from typing import Any
 
 _LOGGER: Incomplete
 SERVICE_RANDOM_EFFECT: str
@@ -21,23 +20,24 @@ SAT: Incomplete
 VAL: Incomplete
 TRANSITION: Incomplete
 HSV_SEQUENCE: Incomplete
-BASE_EFFECT_DICT: Final[Incomplete]
-SEQUENCE_EFFECT_DICT: Final[Incomplete]
-RANDOM_EFFECT_DICT: Final[Incomplete]
+BASE_EFFECT_DICT: VolDictType
+SEQUENCE_EFFECT_DICT: VolDictType
+RANDOM_EFFECT_DICT: VolDictType
 
 def _async_build_base_effect(brightness: int, duration: int, transition: int, segments: list[int]) -> dict[str, Any]: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class TPLinkSmartBulb(CoordinatedTPLinkEntity, LightEntity):
+class TPLinkLightEntity(CoordinatedTPLinkEntity, LightEntity):
     _attr_supported_features: Incomplete
-    _attr_name: Incomplete
     _fixed_color_mode: ColorMode | None
-    device: SmartBulb
-    _attr_unique_id: Incomplete
+    _parent: Incomplete
+    _light_module: Incomplete
+    _attr_name: Incomplete
     _attr_min_color_temp_kelvin: Incomplete
     _attr_max_color_temp_kelvin: Incomplete
     _attr_supported_color_modes: Incomplete
-    def __init__(self, device: SmartBulb, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, light_module: Light, parent: Device | None = None) -> None: ...
+    def _get_unique_id(self) -> str: ...
     def _async_extract_brightness_transition(self, **kwargs: Any) -> tuple[int | None, int | None]: ...
     async def _async_set_hsv(self, hs_color: tuple[int, int], brightness: int | None, transition: int | None) -> None: ...
     async def _async_set_color_temp(self, color_temp: float, brightness: int | None, transition: int | None) -> None: ...
@@ -51,12 +51,13 @@ class TPLinkSmartBulb(CoordinatedTPLinkEntity, LightEntity):
     _attr_color_temp_kelvin: Incomplete
     _attr_hs_color: Incomplete
     def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...
 
-class TPLinkSmartLightStrip(TPLinkSmartBulb):
-    device: SmartLightStrip
+class TPLinkLightEffectEntity(TPLinkLightEntity):
+    _effect_module: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, light_module: Light, effect_module: LightEffect) -> None: ...
     _attr_supported_features: Incomplete
     _attr_effect: Incomplete
+    _attr_color_mode: Incomplete
     _attr_effect_list: Incomplete
     def _async_update_attrs(self) -> None: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/tplink/number.pyi

@@ -0,0 +1,27 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription, async_refresh_after as async_refresh_after
+from _typeshed import Incomplete
+from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription, NumberMode as NumberMode
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from kasa import Device as Device, Feature
+from typing import Final
+
+_LOGGER: Incomplete
+
+class TPLinkNumberEntityDescription(NumberEntityDescription, TPLinkFeatureEntityDescription):
+    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, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step) -> None: ...
+
+NUMBER_DESCRIPTIONS: Final[Incomplete]
+NUMBER_DESCRIPTIONS_MAP: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TPLinkNumberEntity(CoordinatedTPLinkFeatureEntity, NumberEntity):
+    entity_description: TPLinkNumberEntityDescription
+    _attr_native_min_value: Incomplete
+    _attr_native_max_value: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature: Feature, description: TPLinkFeatureEntityDescription, parent: Device | None = None) -> None: ...
+    async def async_set_native_value(self, value: float) -> None: ...
+    _attr_native_value: Incomplete
+    def _async_update_attrs(self) -> None: ...

homeassistant-stubs/components/tplink/select.pyi

@@ -0,0 +1,26 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription, async_refresh_after as async_refresh_after
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from kasa import Device as Device, Feature
+from typing import Final
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkSelectEntityDescription(SelectEntityDescription, TPLinkFeatureEntityDescription):
+    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, options) -> None: ...
+
+SELECT_DESCRIPTIONS: Final[Incomplete]
+SELECT_DESCRIPTIONS_MAP: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TPLinkSelectEntity(CoordinatedTPLinkFeatureEntity, SelectEntity):
+    entity_description: TPLinkSelectEntityDescription
+    _attr_options: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, feature: Feature, description: TPLinkFeatureEntityDescription, parent: Device | None = None) -> None: ...
+    async def async_select_option(self, option: str) -> None: ...
+    _attr_current_option: Incomplete
+    def _async_update_attrs(self) -> None: ...

homeassistant-stubs/components/tplink/sensor.pyi

@@ -1,35 +1,24 @@
-from . import legacy_device_id as legacy_device_id
-from .const import ATTR_CURRENT_A as ATTR_CURRENT_A, ATTR_CURRENT_POWER_W as ATTR_CURRENT_POWER_W, ATTR_TODAY_ENERGY_KWH as ATTR_TODAY_ENERGY_KWH, ATTR_TOTAL_ENERGY_KWH as ATTR_TOTAL_ENERGY_KWH, DOMAIN as DOMAIN
-from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
-from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity
-from .models import TPLinkData as TPLinkData
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .const import UNIT_MAPPING as UNIT_MAPPING
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import ATTR_VOLTAGE as ATTR_VOLTAGE, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfPower as UnitOfPower
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from kasa import SmartDevice as SmartDevice
 
-@dataclass(frozen=True)
-class TPLinkSensorEntityDescription(SensorEntityDescription):
-    emeter_attr: str | None = ...
-    precision: int | 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, emeter_attr, precision) -> None: ...
+@dataclass(frozen=True, kw_only=True)
+class TPLinkSensorEntityDescription(SensorEntityDescription, TPLinkFeatureEntityDescription):
+    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) -> None: ...
 
-ENERGY_SENSORS: tuple[TPLinkSensorEntityDescription, ...]
+SENSOR_DESCRIPTIONS: tuple[TPLinkSensorEntityDescription, ...]
+SENSOR_DESCRIPTIONS_MAP: Incomplete
 
-def async_emeter_from_device(device: SmartDevice, description: TPLinkSensorEntityDescription) -> float | None: ...
-def _async_sensors_for_device(device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, has_parent: bool = False) -> list[SmartPlugSensor]: ...
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class SmartPlugSensor(CoordinatedTPLinkEntity, SensorEntity):
+class TPLinkSensorEntity(CoordinatedTPLinkFeatureEntity, SensorEntity):
     entity_description: TPLinkSensorEntityDescription
-    _attr_unique_id: Incomplete
-    _attr_translation_placeholders: Incomplete
-    _attr_translation_key: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkSensorEntityDescription, has_parent: bool = False) -> None: ...
+    _attr_suggested_display_precision: Incomplete
     _attr_native_value: Incomplete
+    _attr_native_unit_of_measurement: Incomplete
     def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...

homeassistant-stubs/components/tplink/switch.pyi

@@ -1,48 +1,25 @@
-from . import legacy_device_id as legacy_device_id
-from .const import DOMAIN as DOMAIN
-from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
-from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity, async_refresh_after as async_refresh_after
-from .models import TPLinkData as TPLinkData
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription, async_refresh_after as async_refresh_after
 from _typeshed import Incomplete
-from homeassistant.components.switch import SwitchEntity as SwitchEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import EntityCategory as EntityCategory
+from dataclasses import dataclass
+from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from kasa import SmartDevice as SmartDevice, SmartPlug
 from typing import Any
 
 _LOGGER: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+@dataclass(frozen=True, kw_only=True)
+class TPLinkSwitchEntityDescription(SwitchEntityDescription, TPLinkFeatureEntityDescription):
+    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) -> None: ...
 
-class SmartPlugLedSwitch(CoordinatedTPLinkEntity, SwitchEntity):
-    device: SmartPlug
-    _attr_translation_key: str
-    _attr_entity_category: Incomplete
-    _attr_unique_id: Incomplete
-    def __init__(self, device: SmartPlug, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-    _attr_is_on: Incomplete
-    def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...
+SWITCH_DESCRIPTIONS: tuple[TPLinkSwitchEntityDescription, ...]
+SWITCH_DESCRIPTIONS_MAP: Incomplete
 
-class SmartPlugSwitch(CoordinatedTPLinkEntity, SwitchEntity):
-    _attr_name: str | None
-    _attr_unique_id: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
-    async def async_turn_on(self, **kwargs: Any) -> None: ...
-    async def async_turn_off(self, **kwargs: Any) -> None: ...
-    _attr_is_on: Incomplete
-    def _async_update_attrs(self) -> None: ...
-    def _handle_coordinator_update(self) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class SmartPlugSwitchChild(SmartPlugSwitch):
-    _plug: Incomplete
-    _attr_unique_id: Incomplete
-    _attr_name: Incomplete
-    def __init__(self, device: SmartDevice, coordinator: TPLinkDataUpdateCoordinator, plug: SmartDevice) -> None: ...
+class TPLinkSwitch(CoordinatedTPLinkFeatureEntity, SwitchEntity):
+    entity_description: TPLinkSwitchEntityDescription
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     _attr_is_on: Incomplete

homeassistant-stubs/components/tplink_omada/controller.pyi

@@ -1,25 +1,13 @@
-from .coordinator import OmadaCoordinator as OmadaCoordinator
+from .coordinator import OmadaClientsCoordinator as OmadaClientsCoordinator, OmadaGatewayCoordinator as OmadaGatewayCoordinator, OmadaSwitchPortCoordinator as OmadaSwitchPortCoordinator
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
 from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
-from tplink_omada_client.devices import OmadaGateway, OmadaSwitch as OmadaSwitch, OmadaSwitchPortDetails
-
-POLL_SWITCH_PORT: int
-POLL_GATEWAY: int
-
-class OmadaSwitchPortCoordinator(OmadaCoordinator[OmadaSwitchPortDetails]):
-    _network_switch: Incomplete
-    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, network_switch: OmadaSwitch) -> None: ...
-    async def poll_update(self) -> dict[str, OmadaSwitchPortDetails]: ...
-
-class OmadaGatewayCoordinator(OmadaCoordinator[OmadaGateway]):
-    mac: Incomplete
-    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, mac: str) -> None: ...
-    async def poll_update(self) -> dict[str, OmadaGateway]: ...
+from tplink_omada_client.devices import OmadaSwitch as OmadaSwitch
 
 class OmadaSiteController:
     _gateway_coordinator: OmadaGatewayCoordinator | None
     _initialized_gateway_coordinator: bool
+    _clients_coordinator: OmadaClientsCoordinator | None
     _hass: Incomplete
     _omada_client: Incomplete
     _switch_port_coordinators: Incomplete
@@ -28,3 +16,4 @@ class OmadaSiteController:
     def omada_client(self) -> OmadaSiteClient: ...
     def get_switch_port_coordinator(self, switch: OmadaSwitch) -> OmadaSwitchPortCoordinator: ...
     async def get_gateway_coordinator(self) -> OmadaGatewayCoordinator | None: ...
+    def get_clients_coordinator(self) -> OmadaClientsCoordinator: ...

homeassistant-stubs/components/tplink_omada/coordinator.pyi

@@ -1,12 +1,31 @@
 from _typeshed import Incomplete
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
+from tplink_omada_client import OmadaSiteClient as OmadaSiteClient, OmadaSwitchPortDetails
+from tplink_omada_client.clients import OmadaWirelessClient
+from tplink_omada_client.devices import OmadaGateway, OmadaSwitch as OmadaSwitch
 
 _LOGGER: Incomplete
+POLL_SWITCH_PORT: int
+POLL_GATEWAY: int
+POLL_CLIENTS: int
 
 class OmadaCoordinator(DataUpdateCoordinator[dict[str, _T]]):
     omada_client: Incomplete
     def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, name: str, poll_delay: int = 300) -> None: ...
     async def _async_update_data(self) -> dict[str, _T]: ...
     async def poll_update(self) -> dict[str, _T]: ...
+
+class OmadaSwitchPortCoordinator(OmadaCoordinator[OmadaSwitchPortDetails]):
+    _network_switch: Incomplete
+    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, network_switch: OmadaSwitch) -> None: ...
+    async def poll_update(self) -> dict[str, OmadaSwitchPortDetails]: ...
+
+class OmadaGatewayCoordinator(OmadaCoordinator[OmadaGateway]):
+    mac: Incomplete
+    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient, mac: str) -> None: ...
+    async def poll_update(self) -> dict[str, OmadaGateway]: ...
+
+class OmadaClientsCoordinator(OmadaCoordinator[OmadaWirelessClient]):
+    def __init__(self, hass: HomeAssistant, omada_client: OmadaSiteClient) -> None: ...
+    async def poll_update(self) -> dict[str, OmadaWirelessClient]: ...

homeassistant-stubs/components/tplink_omada/device_tracker.pyi

@@ -0,0 +1,36 @@
+from .config_flow import CONF_SITE as CONF_SITE
+from .const import DOMAIN as DOMAIN
+from .controller import OmadaClientsCoordinator as OmadaClientsCoordinator, OmadaSiteController as OmadaSiteController
+from _typeshed import Incomplete
+from homeassistant.components.device_tracker import ScannerEntity as ScannerEntity, SourceType as SourceType
+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 homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+from tplink_omada_client.clients import OmadaWirelessClient
+
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class OmadaClientScannerEntity(CoordinatorEntity[OmadaClientsCoordinator], ScannerEntity):
+    _client_details: OmadaWirelessClient | None
+    _site_id: Incomplete
+    _client_id: Incomplete
+    _attr_name: Incomplete
+    def __init__(self, site_id: str, client_id: str, display_name: str, coordinator: OmadaClientsCoordinator) -> None: ...
+    @property
+    def source_type(self) -> SourceType: ...
+    def _do_update(self) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    def _handle_coordinator_update(self) -> None: ...
+    @property
+    def ip_address(self) -> str | None: ...
+    @property
+    def mac_address(self) -> str | None: ...
+    @property
+    def hostname(self) -> str | None: ...
+    @property
+    def is_connected(self) -> bool: ...
+    @property
+    def unique_id(self) -> str | None: ...

homeassistant-stubs/components/transmission/__init__.pyi

@@ -3,11 +3,12 @@ from .const import ATTR_DELETE_DATA as ATTR_DELETE_DATA, ATTR_TORRENT as ATTR_TO
 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.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
 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.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady, HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import selector as selector
+from homeassistant.helpers.typing import ConfigType as ConfigType
 from typing import Any
 
 _LOGGER: Incomplete
@@ -18,8 +19,13 @@ SERVICE_ADD_TORRENT_SCHEMA: Incomplete
 SERVICE_REMOVE_TORRENT_SCHEMA: Incomplete
 SERVICE_START_TORRENT_SCHEMA: Incomplete
 SERVICE_STOP_TORRENT_SCHEMA: Incomplete
+CONFIG_SCHEMA: Incomplete
+TransmissionConfigEntry = ConfigEntry[TransmissionDataUpdateCoordinator]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
 async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+def _get_coordinator_from_service_data(hass: HomeAssistant, entry_id: str) -> TransmissionDataUpdateCoordinator: ...
+def setup_hass_services(hass: HomeAssistant) -> None: ...
 async def get_api(hass: HomeAssistant, entry: dict[str, Any]) -> transmission_rpc.Client: ...

homeassistant-stubs/components/transmission/sensor.pyi

@@ -1,10 +1,10 @@
+from . import TransmissionConfigEntry as TransmissionConfigEntry
 from .const import DOMAIN as DOMAIN, STATE_ATTR_TORRENT_INFO as STATE_ATTR_TORRENT_INFO, STATE_DOWNLOADING as STATE_DOWNLOADING, STATE_SEEDING as STATE_SEEDING, STATE_UP_DOWN as STATE_UP_DOWN, SUPPORTED_ORDER_MODES as SUPPORTED_ORDER_MODES
 from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 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 STATE_IDLE as STATE_IDLE, UnitOfDataRate as UnitOfDataRate
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
@@ -24,7 +24,7 @@ class TransmissionSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_TYPES: tuple[TransmissionSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TransmissionSensor(CoordinatorEntity[TransmissionDataUpdateCoordinator], SensorEntity):
     entity_description: TransmissionSensorEntityDescription

homeassistant-stubs/components/transmission/switch.pyi

@@ -1,18 +1,16 @@
+from . import TransmissionConfigEntry as TransmissionConfigEntry
 from .const import DOMAIN as DOMAIN
 from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from typing import Any
 
-_LOGGING: Incomplete
-
 @dataclass(frozen=True, kw_only=True)
 class TransmissionSwitchEntityDescription(SwitchEntityDescription):
     is_on_func: Callable[[TransmissionDataUpdateCoordinator], bool | None]
@@ -22,7 +20,7 @@ class TransmissionSwitchEntityDescription(SwitchEntityDescription):
 
 SWITCH_TYPES: tuple[TransmissionSwitchEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TransmissionSwitch(CoordinatorEntity[TransmissionDataUpdateCoordinator], SwitchEntity):
     entity_description: TransmissionSwitchEntityDescription

homeassistant-stubs/components/trend/__init__.pyi

@@ -1,7 +1,8 @@
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import Platform as Platform
+from homeassistant.const import CONF_ENTITY_ID as CONF_ENTITY_ID, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device import async_remove_stale_devices_links_keep_entity_device as async_remove_stale_devices_links_keep_entity_device
 
 PLATFORMS: Incomplete
 

homeassistant-stubs/components/trend/binary_sensor.pyi

@@ -2,10 +2,12 @@ from . import PLATFORMS as PLATFORMS
 from .const import ATTR_GRADIENT as ATTR_GRADIENT, ATTR_INVERT as ATTR_INVERT, ATTR_MIN_GRADIENT as ATTR_MIN_GRADIENT, ATTR_SAMPLE_COUNT as ATTR_SAMPLE_COUNT, ATTR_SAMPLE_DURATION as ATTR_SAMPLE_DURATION, CONF_INVERT as CONF_INVERT, CONF_MAX_SAMPLES as CONF_MAX_SAMPLES, CONF_MIN_GRADIENT as CONF_MIN_GRADIENT, CONF_MIN_SAMPLES as CONF_MIN_SAMPLES, CONF_SAMPLE_DURATION as CONF_SAMPLE_DURATION, DEFAULT_MAX_SAMPLES as DEFAULT_MAX_SAMPLES, DEFAULT_MIN_GRADIENT as DEFAULT_MIN_GRADIENT, DEFAULT_MIN_SAMPLES as DEFAULT_MIN_SAMPLES, DEFAULT_SAMPLE_DURATION as DEFAULT_SAMPLE_DURATION, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 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.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, ENTITY_ID_FORMAT as ENTITY_ID_FORMAT
 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 Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.device import async_device_info_to_link_from_entity as async_device_info_to_link_from_entity
 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 async_track_state_change_event as async_track_state_change_event
@@ -20,6 +22,7 @@ _LOGGER: Incomplete
 def _validate_min_max(data: dict[str, Any]) -> dict[str, Any]: ...
 
 SENSOR_SCHEMA: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
@@ -38,8 +41,9 @@ class SensorTrend(BinarySensorEntity, RestoreEntity):
     _attr_name: Incomplete
     _attr_device_class: Incomplete
     _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
     entity_id: Incomplete
-    def __init__(self, name: str, entity_id: str, attribute: str | None, invert: bool, sample_duration: int, min_gradient: float, min_samples: int, max_samples: int, unique_id: str | None = None, device_class: BinarySensorDeviceClass | None = None, sensor_entity_id: str | None = None) -> None: ...
+    def __init__(self, name: str, entity_id: str, attribute: str | None, invert: bool, sample_duration: int, min_gradient: float, min_samples: int, max_samples: int, unique_id: str | None = None, device_class: BinarySensorDeviceClass | None = None, sensor_entity_id: str | None = None, device_info: dr.DeviceInfo | None = None) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
     @property

homeassistant-stubs/components/tts/notify.pyi

@@ -1,6 +1,6 @@
 from . import ATTR_LANGUAGE as ATTR_LANGUAGE, ATTR_MEDIA_PLAYER_ENTITY_ID as ATTR_MEDIA_PLAYER_ENTITY_ID, ATTR_MESSAGE as ATTR_MESSAGE, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.components.notify import BaseNotificationService as BaseNotificationService, PLATFORM_SCHEMA as PLATFORM_SCHEMA
+from homeassistant.components.notify import BaseNotificationService as BaseNotificationService
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant, split_entity_id as split_entity_id
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
@@ -10,6 +10,7 @@ CONF_MEDIA_PLAYER: str
 CONF_TTS_SERVICE: str
 ENTITY_LEGACY_PROVIDER_GROUP: str
 _LOGGER: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 async def async_get_service(hass: HomeAssistant, config: ConfigType, discovery_info: DiscoveryInfoType | None = None) -> TTSNotificationService: ...
 

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -6,7 +6,7 @@ from _typeshed import Incomplete
 from aiounifi.interfaces.sites import Sites as Sites
 from collections.abc import Mapping
 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.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, 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.helpers.device_registry import format_mac as format_mac

homeassistant-stubs/components/unifi/sensor.pyi

@@ -25,8 +25,9 @@ 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_clients_value_fn(hub: UnifiHub, device: Device) -> 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_uptime_value_changed_fn(old: StateType | date | datetime | Decimal, new: datetime | float | str | None) -> 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 device_system_stats_supported_fn(stat_index: int, hub: UnifiHub, obj_id: str) -> bool: ...

homeassistant-stubs/components/unifi/services.pyi

@@ -1,5 +1,6 @@
 from _typeshed import Incomplete
 from collections.abc import Mapping
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
 from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC