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

homeassistant-stubs/components/threshold/binary_sensor.pyi

@@ -1,9 +1,11 @@
 from .const import CONF_HYSTERESIS as CONF_HYSTERESIS, CONF_LOWER as CONF_LOWER, CONF_UPPER as CONF_UPPER
 from _typeshed import Incomplete
-from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA, PLATFORM_SCHEMA as PLATFORM_SCHEMA
+from collections.abc import Callable as Callable, Mapping
+from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, DEVICE_CLASSES_SCHEMA as DEVICE_CLASSES_SCHEMA
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, CONF_DEVICE_CLASS as CONF_DEVICE_CLASS, CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME, 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.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.device import async_device_info_to_link_from_entity as async_device_info_to_link_from_entity
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 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
@@ -26,6 +28,7 @@ POSITION_UNKNOWN: str
 TYPE_LOWER: str
 TYPE_RANGE: str
 TYPE_UPPER: str
+PLATFORM_SCHEMA: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 async def async_setup_platform(hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
@@ -33,6 +36,7 @@ def _threshold_type(lower: float | None, upper: float | None) -> str: ...
 
 class ThresholdSensor(BinarySensorEntity):
     _attr_should_poll: bool
+    _preview_callback: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
     _entity_id: Incomplete
@@ -45,9 +49,13 @@ class ThresholdSensor(BinarySensorEntity):
     _state_position: Incomplete
     _state: Incomplete
     sensor_value: Incomplete
-    def __init__(self, hass: HomeAssistant, entity_id: str, name: str, lower: float | None, upper: float | None, hysteresis: float, device_class: BinarySensorDeviceClass | None, unique_id: str | None, device_info: DeviceInfo | None = None) -> None: ...
+    def __init__(self, entity_id: str, name: str, lower: float | None, upper: float | None, hysteresis: float, device_class: BinarySensorDeviceClass | None, unique_id: str | None, device_info: DeviceInfo | None = None) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    def _async_setup_sensor(self) -> None: ...
     @property
     def is_on(self) -> bool | None: ...
     @property
     def extra_state_attributes(self) -> dict[str, Any]: ...
     def _update_state(self) -> None: ...
+    _attr_available: bool
+    def async_start_preview(self, preview_callback: Callable[[str, Mapping[str, Any]], None]) -> CALLBACK_TYPE: ...

homeassistant-stubs/components/threshold/config_flow.pyi

@@ -1,7 +1,11 @@
+from .binary_sensor import ThresholdSensor as ThresholdSensor
 from .const import CONF_HYSTERESIS as CONF_HYSTERESIS, CONF_LOWER as CONF_LOWER, CONF_UPPER as CONF_UPPER, DEFAULT_HYSTERESIS as DEFAULT_HYSTERESIS, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from collections.abc import Mapping
+from homeassistant.components import websocket_api as websocket_api
 from homeassistant.const import CONF_ENTITY_ID as CONF_ENTITY_ID, CONF_NAME as CONF_NAME
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import selector as selector
 from homeassistant.helpers.schema_config_entry_flow import SchemaCommonFlowHandler as SchemaCommonFlowHandler, SchemaConfigFlowHandler as SchemaConfigFlowHandler, SchemaFlowError as SchemaFlowError, SchemaFlowFormStep as SchemaFlowFormStep
 from typing import Any
@@ -17,3 +21,7 @@ class ConfigFlowHandler(SchemaConfigFlowHandler, domain=DOMAIN):
     config_flow = CONFIG_FLOW
     options_flow = OPTIONS_FLOW
     def async_config_entry_title(self, options: Mapping[str, Any]) -> str: ...
+    @staticmethod
+    async def async_setup_preview(hass: HomeAssistant) -> None: ...
+
+def ws_start_preview(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/tibber/__init__.pyi

@@ -1,4 +1,5 @@
 from .const import DATA_HASS_CONFIG as DATA_HASS_CONFIG, DOMAIN as DOMAIN
+from .services import async_setup_services as async_setup_services
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ACCESS_TOKEN as CONF_ACCESS_TOKEN, CONF_NAME as CONF_NAME, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform

homeassistant-stubs/components/tibber/sensor.pyi

@@ -2,11 +2,13 @@ import tibber
 from .const import MANUFACTURER as MANUFACTURER
 from .coordinator import TibberDataCoordinator as TibberDataCoordinator
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable
 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 EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, SIGNAL_STRENGTH_DECIBELS as SIGNAL_STRENGTH_DECIBELS, UnitOfElectricCurrent as UnitOfElectricCurrent, UnitOfElectricPotential as UnitOfElectricPotential, UnitOfEnergy as UnitOfEnergy, UnitOfPower as UnitOfPower
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import PlatformNotReady as PlatformNotReady
+from homeassistant.helpers import entity_registry as er
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
@@ -74,15 +76,19 @@ class TibberSensorRT(TibberSensor, CoordinatorEntity['TibberRtDataCoordinator'])
     _attr_last_reset: Incomplete
     def _handle_coordinator_update(self) -> None: ...
 
-class TibberRtDataCoordinator(DataUpdateCoordinator):
+class TibberRtEntityCreator:
     _async_add_entities: Incomplete
     _tibber_home: Incomplete
-    hass: Incomplete
     _added_sensors: Incomplete
+    _entity_registry: Incomplete
+    def __init__(self, async_add_entities: AddEntitiesCallback, tibber_home: tibber.TibberHome, entity_registry: er.EntityRegistry) -> None: ...
+    def _migrate_unique_id(self, sensor_description: SensorEntityDescription) -> None: ...
+    def add_sensors(self, coordinator: TibberRtDataCoordinator, live_measurement: Any) -> None: ...
+
+class TibberRtDataCoordinator(DataUpdateCoordinator):
+    _add_sensor_callback: Incomplete
     _async_remove_device_updates_handler: Incomplete
-    entity_registry: Incomplete
-    def __init__(self, async_add_entities: AddEntitiesCallback, tibber_home: tibber.TibberHome, hass: HomeAssistant) -> None: ...
+    def __init__(self, add_sensor_callback: Callable[[TibberRtDataCoordinator, Any], None], tibber_home: tibber.TibberHome, hass: HomeAssistant) -> None: ...
     def _handle_ha_stop(self, _event: Event) -> None: ...
-    def _migrate_unique_id(self, sensor_description: SensorEntityDescription) -> None: ...
-    def _add_sensors(self) -> None: ...
+    def _data_updated(self) -> None: ...
     def get_live_measurement(self) -> Any: ...

homeassistant-stubs/components/tibber/services.pyi

@@ -0,0 +1,15 @@
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from datetime import date, datetime
+from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, ServiceResponse as ServiceResponse, SupportsResponse as SupportsResponse, callback as callback
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from typing import Final
+
+PRICE_SERVICE_NAME: str
+ATTR_START: Final[str]
+ATTR_END: Final[str]
+SERVICE_SCHEMA: Final[Incomplete]
+
+async def __get_prices(call: ServiceCall, *, hass: HomeAssistant) -> ServiceResponse: ...
+def __get_date(date_input: str | None, mode: str | None) -> date | datetime: ...
+def async_setup_services(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/time_date/sensor.pyi

@@ -1,19 +1,19 @@
-from .const import DOMAIN as DOMAIN, OPTION_TYPES as OPTION_TYPES
+from .const import OPTION_TYPES as OPTION_TYPES
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Mapping
 from datetime import datetime
-from homeassistant.components.sensor import ENTITY_ID_FORMAT as ENTITY_ID_FORMAT, PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorEntity as SensorEntity
+from homeassistant.components.sensor import ENTITY_ID_FORMAT as ENTITY_ID_FORMAT, SensorEntity as SensorEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_DISPLAY_OPTIONS as CONF_DISPLAY_OPTIONS, EVENT_CORE_CONFIG_UPDATE as EVENT_CORE_CONFIG_UPDATE
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.event import async_track_point_in_utc_time as async_track_point_in_utc_time
-from homeassistant.helpers.issue_registry import IssueSeverity as IssueSeverity, async_create_issue as async_create_issue
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
 from typing import Any
 
 _LOGGER: Incomplete
 TIME_STR_FORMAT: str
+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: ...

homeassistant-stubs/components/timer/__init__.pyi

@@ -9,7 +9,7 @@ from homeassistant.helpers.entity_component import EntityComponent as EntityComp
 from homeassistant.helpers.event import async_track_point_in_utc_time as async_track_point_in_utc_time
 from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
 from homeassistant.helpers.storage import Store as Store
-from homeassistant.helpers.typing import ConfigType as ConfigType
+from homeassistant.helpers.typing import ConfigType as ConfigType, VolDictType as VolDictType
 from typing import Any, Self
 
 _LOGGER: Incomplete
@@ -40,7 +40,7 @@ SERVICE_CHANGE: str
 SERVICE_FINISH: str
 STORAGE_KEY = DOMAIN
 STORAGE_VERSION: int
-STORAGE_FIELDS: Incomplete
+STORAGE_FIELDS: VolDictType
 
 def _format_timedelta(delta: timedelta) -> str: ...
 def _none_to_empty_dict(value: _T | None) -> _T | dict[Any, Any]: ...

homeassistant-stubs/components/todo/__init__.pyi

@@ -10,7 +10,6 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_ENTITY_ID as CONF_ENTITY_ID
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, SupportsResponse as SupportsResponse, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
-from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.entity import Entity as Entity
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
 from homeassistant.helpers.typing import ConfigType as ConfigType
@@ -18,8 +17,10 @@ from homeassistant.util.json import JsonValueType as JsonValueType
 from typing import Any
 
 _LOGGER: Incomplete
-SCAN_INTERVAL: Incomplete
 ENTITY_ID_FORMAT: Incomplete
+PLATFORM_SCHEMA: Incomplete
+PLATFORM_SCHEMA_BASE: Incomplete
+SCAN_INTERVAL: Incomplete
 
 @dataclasses.dataclass
 class TodoItemFieldDescription:

homeassistant-stubs/components/tplink/__init__.pyi

@@ -13,19 +13,22 @@ from homeassistant.helpers import discovery_flow as discovery_flow
 from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession
 from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
 from homeassistant.helpers.typing import ConfigType as ConfigType
-from kasa import Credentials, SmartDevice
+from kasa import Credentials, Device
 
+TPLinkConfigEntry = ConfigEntry[TPLinkData]
 DISCOVERY_INTERVAL: Incomplete
 CONFIG_SCHEMA: Incomplete
 _LOGGER: Incomplete
 
 def create_async_tplink_clientsession(hass: HomeAssistant) -> ClientSession: ...
-def async_trigger_discovery(hass: HomeAssistant, discovered_devices: dict[str, SmartDevice]) -> None: ...
-async def async_discover_devices(hass: HomeAssistant) -> dict[str, SmartDevice]: ...
+def async_trigger_discovery(hass: HomeAssistant, discovered_devices: dict[str, Device]) -> None: ...
+async def async_discover_devices(hass: HomeAssistant) -> dict[str, Device]: ...
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-def legacy_device_id(device: SmartDevice) -> str: ...
+async def async_setup_entry(hass: HomeAssistant, entry: TPLinkConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: TPLinkConfigEntry) -> bool: ...
+def legacy_device_id(device: Device) -> str: ...
+def get_device_name(device: Device, parent: Device | None = None) -> str: ...
 async def get_credentials(hass: HomeAssistant) -> Credentials | None: ...
 async def set_credentials(hass: HomeAssistant, username: str, password: str) -> None: ...
 def mac_alias(mac: str) -> str: ...
+async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/tplink/binary_sensor.pyi

@@ -0,0 +1,22 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Final
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkBinarySensorEntityDescription(BinarySensorEntityDescription, 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: ...
+
+BINARY_SENSOR_DESCRIPTIONS: Final[Incomplete]
+BINARYSENSOR_DESCRIPTIONS_MAP: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TPLinkBinarySensorEntity(CoordinatedTPLinkFeatureEntity, BinarySensorEntity):
+    entity_description: TPLinkBinarySensorEntityDescription
+    _attr_is_on: Incomplete
+    def _async_update_attrs(self) -> None: ...

homeassistant-stubs/components/tplink/button.pyi

@@ -0,0 +1,22 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Final
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkButtonEntityDescription(ButtonEntityDescription, 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: ...
+
+BUTTON_DESCRIPTIONS: Final[Incomplete]
+BUTTON_DESCRIPTIONS_MAP: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TPLinkButtonEntity(CoordinatedTPLinkFeatureEntity, ButtonEntity):
+    entity_description: TPLinkButtonEntityDescription
+    async def async_press(self) -> None: ...
+    def _async_update_attrs(self) -> None: ...

homeassistant-stubs/components/tplink/climate.pyi

@@ -0,0 +1,42 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .const import UNIT_MAPPING as UNIT_MAPPING
+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.climate import ATTR_TEMPERATURE as ATTR_TEMPERATURE, ClimateEntity as ClimateEntity, ClimateEntityFeature as ClimateEntityFeature, HVACAction as HVACAction, HVACMode as HVACMode
+from homeassistant.const import PRECISION_WHOLE as PRECISION_WHOLE
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from kasa import Device as Device
+from typing import Any
+
+STATE_TO_ACTION: Incomplete
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TPLinkClimateEntity(CoordinatedTPLinkEntity, ClimateEntity):
+    _attr_name: Incomplete
+    _attr_supported_features: Incomplete
+    _attr_hvac_modes: Incomplete
+    _attr_precision = PRECISION_WHOLE
+    _enable_turn_on_off_backwards_compatibility: bool
+    _state_feature: Incomplete
+    _mode_feature: Incomplete
+    _temp_feature: Incomplete
+    _target_feature: Incomplete
+    _attr_min_temp: Incomplete
+    _attr_max_temp: Incomplete
+    _attr_temperature_unit: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, parent: Device) -> None: ...
+    async def async_set_temperature(self, **kwargs: Any) -> None: ...
+    async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None: ...
+    async def async_turn_on(self) -> None: ...
+    async def async_turn_off(self) -> None: ...
+    _attr_current_temperature: Incomplete
+    _attr_target_temperature: Incomplete
+    _attr_hvac_mode: Incomplete
+    _attr_hvac_action: Incomplete
+    def _async_update_attrs(self) -> None: ...
+    def _get_unique_id(self) -> str: ...

homeassistant-stubs/components/tplink/config_flow.pyi

@@ -7,7 +7,7 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntry
 from homeassistant.const import CONF_ALIAS as CONF_ALIAS, CONF_DEVICE as CONF_DEVICE, CONF_HOST as CONF_HOST, CONF_MAC as CONF_MAC, CONF_MODEL as CONF_MODEL, CONF_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import callback as callback
 from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
-from kasa import Credentials, SmartDevice
+from kasa import Credentials, Device
 from typing import Any
 
 STEP_AUTH_DATA_SCHEMA: Incomplete
@@ -30,8 +30,8 @@ class TPLinkConfigFlow(ConfigFlow, domain=DOMAIN):
     async def async_step_user_auth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_pick_device(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def _async_reload_requires_auth_entries(self) -> None: ...
-    def _async_create_entry_from_device(self, device: SmartDevice) -> ConfigFlowResult: ...
-    async def _async_try_discover_and_update(self, host: str, credentials: Credentials | None, raise_on_progress: bool) -> SmartDevice: ...
-    async def _async_try_connect(self, discovered_device: SmartDevice, credentials: Credentials | None) -> SmartDevice: ...
+    def _async_create_entry_from_device(self, device: Device) -> ConfigFlowResult: ...
+    async def _async_try_discover_and_update(self, host: str, credentials: Credentials | None, raise_on_progress: bool) -> Device: ...
+    async def _async_try_connect(self, discovered_device: Device, credentials: Credentials | None) -> Device: ...
     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/tplink/const.pyi

@@ -1,13 +1,15 @@
 from _typeshed import Incomplete
-from homeassistant.const import Platform as Platform
+from homeassistant.const import Platform as Platform, UnitOfTemperature as UnitOfTemperature
 from typing import Final
 
 DOMAIN: str
 DISCOVERY_TIMEOUT: int
 CONNECT_TIMEOUT: int
+PRIMARY_STATE_ID: str
 ATTR_CURRENT_A: Final[str]
 ATTR_CURRENT_POWER_W: Final[str]
 ATTR_TODAY_ENERGY_KWH: Final[str]
 ATTR_TOTAL_ENERGY_KWH: Final[str]
 CONF_DEVICE_CONFIG: Final[str]
 PLATFORMS: Final[Incomplete]
+UNIT_MAPPING: Incomplete

homeassistant-stubs/components/tplink/coordinator.pyi

@@ -5,7 +5,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.debounce import Debouncer as Debouncer
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from kasa import SmartDevice as SmartDevice
+from kasa import Device as Device
 
 _LOGGER: Incomplete
 REQUEST_REFRESH_DELAY: float
@@ -13,5 +13,5 @@ REQUEST_REFRESH_DELAY: float
 class TPLinkDataUpdateCoordinator(DataUpdateCoordinator[None]):
     config_entry: config_entries.ConfigEntry
     device: Incomplete
-    def __init__(self, hass: HomeAssistant, device: SmartDevice, update_interval: timedelta) -> None: ...
+    def __init__(self, hass: HomeAssistant, device: Device, update_interval: timedelta) -> None: ...
     async def _async_update_data(self) -> None: ...

homeassistant-stubs/components/tplink/diagnostics.pyi

@@ -1,12 +1,10 @@
-from .const import DOMAIN as DOMAIN
-from .models import TPLinkData as TPLinkData
+from . import TPLinkConfigEntry as TPLinkConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import format_mac as format_mac
 from typing import Any
 
 TO_REDACT: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: TPLinkConfigEntry) -> dict[str, Any]: ...

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
+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: ...
+    async def async_added_to_hass(self) -> None: ...
+    @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,10 +51,10 @@ 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_effect_list: Incomplete

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: ...