homeassistant-stubs 2025.9.0b0__py3-none-any.whl → 2026.1.0b0__py3-none-any.whl

homeassistant-stubs/components/tolo/climate.pyi

@@ -1,16 +1,14 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 from _typeshed import Incomplete
 from homeassistant.components.climate import ClimateEntity as ClimateEntity, ClimateEntityFeature as ClimateEntityFeature, FAN_OFF as FAN_OFF, FAN_ON as FAN_ON, HVACAction as HVACAction, HVACMode as HVACMode
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_TEMPERATURE as ATTR_TEMPERATURE, PRECISION_WHOLE as PRECISION_WHOLE, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from tololib import TARGET_HUMIDITY_MAX, TARGET_HUMIDITY_MIN, TARGET_TEMPERATURE_MAX, TARGET_TEMPERATURE_MIN
 from typing import Any
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class SaunaClimate(ToloSaunaCoordinatorEntity, ClimateEntity):
     _attr_fan_modes: Incomplete
@@ -25,7 +23,7 @@ class SaunaClimate(ToloSaunaCoordinatorEntity, ClimateEntity):
     _attr_target_temperature_step: int
     _attr_temperature_unit: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry) -> None: ...
     @property
     def current_temperature(self) -> int: ...
     @property

homeassistant-stubs/components/tolo/config_flow.pyi

@@ -1,18 +1,20 @@
 from .const import DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, SOURCE_RECONFIGURE as SOURCE_RECONFIGURE
 from homeassistant.const import CONF_HOST as CONF_HOST
 from homeassistant.helpers.device_registry import format_mac as format_mac
 from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo as DhcpServiceInfo
 from typing import Any
 
 _LOGGER: Incomplete
+CONFIG_SCHEMA: Incomplete
 
-class ToloSaunaConfigFlow(ConfigFlow, domain=DOMAIN):
+class ToloConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    _discovered_host: str
+    _dhcp_discovery_info: DhcpServiceInfo | None
     @staticmethod
     def _check_device_availability(host: str) -> bool: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_dhcp(self, discovery_info: DhcpServiceInfo) -> ConfigFlowResult: ...
     async def async_step_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/tolo/coordinator.pyi

@@ -8,14 +8,15 @@ from tololib import ToloSettings as ToloSettings, ToloStatus as ToloStatus
 from typing import NamedTuple
 
 _LOGGER: Incomplete
+type ToloConfigEntry = ConfigEntry[ToloSaunaUpdateCoordinator]
 
 class ToloSaunaData(NamedTuple):
     status: ToloStatus
     settings: ToloSettings
 
 class ToloSaunaUpdateCoordinator(DataUpdateCoordinator[ToloSaunaData]):
-    config_entry: ConfigEntry
+    config_entry: ToloConfigEntry
     client: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+    def __init__(self, hass: HomeAssistant, entry: ToloConfigEntry) -> None: ...
     async def _async_update_data(self) -> ToloSaunaData: ...
     def _get_tolo_sauna_data(self) -> ToloSaunaData: ...

homeassistant-stubs/components/tolo/entity.pyi

@@ -1,11 +1,10 @@
 from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 
 class ToloSaunaCoordinatorEntity(CoordinatorEntity[ToloSaunaUpdateCoordinator]):
     _attr_has_entity_name: bool
     _attr_device_info: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry) -> None: ...

homeassistant-stubs/components/tolo/fan.pyi

@@ -1,20 +1,18 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 from _typeshed import Incomplete
 from homeassistant.components.fan import FanEntity as FanEntity, FanEntityFeature as FanEntityFeature
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ToloFan(ToloSaunaCoordinatorEntity, FanEntity):
     _attr_translation_key: str
     _attr_supported_features: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry) -> None: ...
     @property
     def is_on(self) -> bool: ...
     def turn_on(self, percentage: int | None = None, preset_mode: str | None = None, **kwargs: Any) -> None: ...

homeassistant-stubs/components/tolo/light.pyi

@@ -1,21 +1,19 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 from _typeshed import Incomplete
 from homeassistant.components.light import ColorMode as ColorMode, LightEntity as LightEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from typing import Any
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ToloLight(ToloSaunaCoordinatorEntity, LightEntity):
     _attr_color_mode: Incomplete
     _attr_translation_key: str
     _attr_supported_color_modes: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry) -> None: ...
     @property
     def is_on(self) -> bool: ...
     def turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/tolo/number.pyi

@@ -1,11 +1,9 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -22,12 +20,12 @@ class ToloNumberEntityDescription(NumberEntityDescription):
 
 NUMBERS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ToloNumberEntity(ToloSaunaCoordinatorEntity, NumberEntity):
     entity_description: ToloNumberEntityDescription
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry, entity_description: ToloNumberEntityDescription) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry, entity_description: ToloNumberEntityDescription) -> None: ...
     @property
     def native_value(self) -> float: ...
     def set_native_value(self, value: float) -> None: ...

homeassistant-stubs/components/tolo/select.pyi

@@ -1,11 +1,10 @@
-from .const import AromaTherapySlot as AromaTherapySlot, DOMAIN as DOMAIN, LampMode as LampMode
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .const import AromaTherapySlot as AromaTherapySlot, LampMode as LampMode
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -19,13 +18,13 @@ class ToloSelectEntityDescription(SelectEntityDescription):
 
 SELECTS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ToloSelectEntity(ToloSaunaCoordinatorEntity, SelectEntity):
     _attr_entity_category: Incomplete
     entity_description: ToloSelectEntityDescription
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry, entity_description: ToloSelectEntityDescription) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry, entity_description: ToloSelectEntityDescription) -> None: ...
     @property
     def options(self) -> list[str]: ...
     @property

homeassistant-stubs/components/tolo/sensor.pyi

@@ -1,11 +1,9 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 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, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -19,12 +17,12 @@ class ToloSensorEntityDescription(SensorEntityDescription):
 
 SENSORS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ToloSensorEntity(ToloSaunaCoordinatorEntity, SensorEntity):
     entity_description: ToloSensorEntityDescription
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry, entity_description: ToloSensorEntityDescription) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry, entity_description: ToloSensorEntityDescription) -> None: ...
     @property
     def available(self) -> bool: ...
     @property

homeassistant-stubs/components/tolo/switch.pyi

@@ -1,11 +1,9 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
+from .coordinator import ToloConfigEntry as ToloConfigEntry, ToloSaunaUpdateCoordinator as ToloSaunaUpdateCoordinator
 from .entity import ToloSaunaCoordinatorEntity as ToloSaunaCoordinatorEntity
 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.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from tololib import ToloClient as ToloClient, ToloStatus as ToloStatus
@@ -18,12 +16,12 @@ class ToloSwitchEntityDescription(SwitchEntityDescription):
 
 SWITCHES: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class ToloSwitchEntity(ToloSaunaCoordinatorEntity, SwitchEntity):
     entity_description: ToloSwitchEntityDescription
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry, entity_description: ToloSwitchEntityDescription) -> None: ...
+    def __init__(self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry, entity_description: ToloSwitchEntityDescription) -> None: ...
     @property
     def is_on(self) -> bool: ...
     def turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/tplink/number.pyi

@@ -2,7 +2,7 @@ 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.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription, NumberMode as NumberMode
+from homeassistant.components.number import NumberDeviceClass as NumberDeviceClass, 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 AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from kasa import Device as Device, Feature

homeassistant-stubs/components/tplink_omada/__init__.pyi

@@ -1,15 +1,20 @@
 from .config_flow import CONF_SITE as CONF_SITE, create_omada_client as create_omada_client
 from .const import DOMAIN as DOMAIN
 from .controller import OmadaSiteController as OmadaSiteController
+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 Platform as Platform
-from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall
+from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.typing import ConfigType as ConfigType
 from tplink_omada_client.devices import OmadaListDevice as OmadaListDevice
 
 PLATFORMS: list[Platform]
+CONFIG_SCHEMA: Incomplete
 type OmadaConfigEntry = ConfigEntry[OmadaSiteController]
 
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
 async def async_setup_entry(hass: HomeAssistant, entry: OmadaConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: OmadaConfigEntry) -> bool: ...
 def _remove_old_devices(hass: HomeAssistant, entry: OmadaConfigEntry, omada_devices: dict[str, OmadaListDevice]) -> None: ...

homeassistant-stubs/components/tplink_omada/binary_sensor.pyi

@@ -6,8 +6,9 @@ from collections.abc import Callable as Callable
 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 import Entity as Entity
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from tplink_omada_client.devices import OmadaDevice as OmadaDevice, OmadaGatewayPortConfig as OmadaGatewayPortConfig, OmadaGatewayPortStatus as OmadaGatewayPortStatus
+from tplink_omada_client.devices import OmadaDevice as OmadaDevice, OmadaGatewayPortConfig as OmadaGatewayPortConfig, OmadaGatewayPortStatus as OmadaGatewayPortStatus, OmadaListDevice as OmadaListDevice
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: OmadaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/tplink_omada/controller.pyi

@@ -1,9 +1,10 @@
 from . import OmadaConfigEntry as OmadaConfigEntry
 from .coordinator import OmadaClientsCoordinator as OmadaClientsCoordinator, OmadaDevicesCoordinator as OmadaDevicesCoordinator, OmadaGatewayCoordinator as OmadaGatewayCoordinator, OmadaSwitchPortCoordinator as OmadaSwitchPortCoordinator
 from _typeshed import Incomplete
-from homeassistant.core import HomeAssistant as HomeAssistant
+from collections.abc import Awaitable, Callable as Callable
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
-from tplink_omada_client.devices import OmadaSwitch as OmadaSwitch
+from tplink_omada_client.devices import OmadaListDevice as OmadaListDevice, OmadaSwitch as OmadaSwitch
 
 class OmadaSiteController:
     _gateway_coordinator: OmadaGatewayCoordinator | None
@@ -15,6 +16,7 @@ class OmadaSiteController:
     _clients_coordinator: Incomplete
     def __init__(self, hass: HomeAssistant, config_entry: OmadaConfigEntry, omada_client: OmadaSiteClient) -> None: ...
     async def initialize_first_refresh(self) -> None: ...
+    async def async_register_device_entities(self, device_filter: Callable[[OmadaListDevice], bool], entity_callback: Callable[[OmadaListDevice], Awaitable[None]]) -> None: ...
     @property
     def omada_client(self) -> OmadaSiteClient: ...
     def get_switch_port_coordinator(self, switch: OmadaSwitch) -> OmadaSwitchPortCoordinator: ...

homeassistant-stubs/components/tplink_omada/coordinator.pyi

@@ -1,16 +1,18 @@
 from . import OmadaConfigEntry as OmadaConfigEntry
 from _typeshed import Incomplete
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 from tplink_omada_client import OmadaSiteClient as OmadaSiteClient, OmadaSwitchPortDetails
 from tplink_omada_client.clients import OmadaWirelessClient
-from tplink_omada_client.devices import OmadaGateway, OmadaListDevice, OmadaSwitch as OmadaSwitch
+from tplink_omada_client.devices import OmadaFirmwareUpdate as OmadaFirmwareUpdate, OmadaGateway, OmadaListDevice, OmadaSwitch as OmadaSwitch
+from typing import NamedTuple
 
 _LOGGER: Incomplete
 POLL_SWITCH_PORT: int
 POLL_GATEWAY: int
 POLL_CLIENTS: int
 POLL_DEVICES: int
+POLL_UPGRADE: int
 
 class OmadaCoordinator[_T](DataUpdateCoordinator[dict[str, _T]]):
     config_entry: OmadaConfigEntry
@@ -36,3 +38,16 @@ class OmadaDevicesCoordinator(OmadaCoordinator[OmadaListDevice]):
 class OmadaClientsCoordinator(OmadaCoordinator[OmadaWirelessClient]):
     def __init__(self, hass: HomeAssistant, config_entry: OmadaConfigEntry, omada_client: OmadaSiteClient) -> None: ...
     async def poll_update(self) -> dict[str, OmadaWirelessClient]: ...
+
+class FirmwareUpdateStatus(NamedTuple):
+    device: OmadaListDevice
+    firmware: OmadaFirmwareUpdate | None
+
+class OmadaFirmwareUpdateCoordinator(OmadaCoordinator[FirmwareUpdateStatus]):
+    _devices_coordinator: Incomplete
+    _config_entry: Incomplete
+    def __init__(self, hass: HomeAssistant, config_entry: OmadaConfigEntry, omada_client: OmadaSiteClient, devices_coordinator: OmadaDevicesCoordinator) -> None: ...
+    async def _get_firmware_updates(self) -> list[FirmwareUpdateStatus]: ...
+    async def poll_update(self) -> dict[str, FirmwareUpdateStatus]: ...
+    @callback
+    def _handle_devices_update(self) -> None: ...

homeassistant-stubs/components/tplink_omada/services.pyi

@@ -0,0 +1,22 @@
+from .const import DOMAIN as DOMAIN
+from .controller import OmadaSiteController as OmadaSiteController
+from _typeshed import Incomplete
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState
+from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError
+from homeassistant.helpers import selector as selector
+
+SERVICE_RECONNECT_CLIENT: str
+ATTR_CONFIG_ENTRY_ID: str
+ATTR_MAC: str
+
+def _get_controller(call: ServiceCall) -> OmadaSiteController: ...
+
+SCHEMA_RECONNECT_CLIENT: Incomplete
+
+async def _handle_reconnect_client(call: ServiceCall) -> None: ...
+
+SERVICES: Incomplete
+
+@callback
+def async_setup_services(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/tplink_omada/switch.pyi

@@ -8,9 +8,10 @@ from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.const import EntityCategory as EntityCategory
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity import Entity as Entity
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
-from tplink_omada_client.devices import OmadaDevice as OmadaDevice, OmadaGateway, OmadaGatewayPortConfig, OmadaGatewayPortStatus, OmadaSwitch, OmadaSwitchPortDetails
+from tplink_omada_client.devices import OmadaDevice as OmadaDevice, OmadaGateway, OmadaGatewayPortConfig, OmadaGatewayPortStatus, OmadaListDevice as OmadaListDevice, OmadaSwitch, OmadaSwitchPortDetails
 from typing import Any, Generic, TypeVar
 
 TPort = TypeVar('TPort')

homeassistant-stubs/components/tplink_omada/update.pyi

@@ -1,29 +1,13 @@
 from . import OmadaConfigEntry as OmadaConfigEntry
-from .coordinator import OmadaCoordinator as OmadaCoordinator, OmadaDevicesCoordinator as OmadaDevicesCoordinator, POLL_DEVICES as POLL_DEVICES
+from .coordinator import OmadaFirmwareUpdateCoordinator as OmadaFirmwareUpdateCoordinator
 from .entity import OmadaDeviceEntity as OmadaDeviceEntity
 from _typeshed import Incomplete
 from homeassistant.components.update import UpdateDeviceClass as UpdateDeviceClass, UpdateEntity as UpdateEntity, UpdateEntityFeature as UpdateEntityFeature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
-from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
-from tplink_omada_client.devices import OmadaFirmwareUpdate as OmadaFirmwareUpdate, OmadaListDevice as OmadaListDevice
-from typing import Any, NamedTuple
-
-POLL_DELAY_UPGRADE: int
-
-class FirmwareUpdateStatus(NamedTuple):
-    device: OmadaListDevice
-    firmware: OmadaFirmwareUpdate | None
-
-class OmadaFirmwareUpdateCoordinator(OmadaCoordinator[FirmwareUpdateStatus]):
-    _devices_coordinator: Incomplete
-    _config_entry: Incomplete
-    def __init__(self, hass: HomeAssistant, config_entry: OmadaConfigEntry, omada_client: OmadaSiteClient, devices_coordinator: OmadaDevicesCoordinator) -> None: ...
-    async def _get_firmware_updates(self) -> list[FirmwareUpdateStatus]: ...
-    async def poll_update(self) -> dict[str, FirmwareUpdateStatus]: ...
-    @callback
-    def _handle_devices_update(self) -> None: ...
+from tplink_omada_client.devices import OmadaListDevice as OmadaListDevice
+from typing import Any
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: OmadaConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 

homeassistant-stubs/components/tractive/switch.pyi

@@ -1,5 +1,5 @@
 from . import Trackables as Trackables, TractiveClient as TractiveClient, TractiveConfigEntry as TractiveConfigEntry
-from .const import ATTR_BUZZER as ATTR_BUZZER, ATTR_LED as ATTR_LED, ATTR_LIVE_TRACKING as ATTR_LIVE_TRACKING, TRACKER_SWITCH_STATUS_UPDATED as TRACKER_SWITCH_STATUS_UPDATED
+from .const import ATTR_BUZZER as ATTR_BUZZER, ATTR_LED as ATTR_LED, ATTR_LIVE_TRACKING as ATTR_LIVE_TRACKING, ATTR_POWER_SAVING as ATTR_POWER_SAVING, TRACKER_SWITCH_STATUS_UPDATED as TRACKER_SWITCH_STATUS_UPDATED
 from .entity import TractiveEntity as TractiveEntity
 from _typeshed import Incomplete
 from dataclasses import dataclass
@@ -25,7 +25,7 @@ class TractiveSwitch(TractiveEntity, SwitchEntity):
     _tracker: Incomplete
     _method: Incomplete
     def __init__(self, client: TractiveClient, item: Trackables, description: TractiveSwitchEntityDescription) -> None: ...
-    _attr_available: bool
+    _attr_available: Incomplete
     _attr_is_on: Incomplete
     @callback
     def handle_status_update(self, event: dict[str, Any]) -> None: ...

homeassistant-stubs/components/transmission/__init__.pyi

@@ -1,30 +1,23 @@
 import transmission_rpc
-from .const import ATTR_DELETE_DATA as ATTR_DELETE_DATA, ATTR_DOWNLOAD_PATH as ATTR_DOWNLOAD_PATH, 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 .const import DEFAULT_PATH as DEFAULT_PATH, DEFAULT_SSL as DEFAULT_SSL, DOMAIN as DOMAIN
 from .coordinator import TransmissionConfigEntry as TransmissionConfigEntry, TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
 from .errors import AuthenticationError as AuthenticationError, CannotConnect as CannotConnect, UnknownError as UnknownError
+from .services import async_setup_services as async_setup_services
 from _typeshed import Incomplete
-from homeassistant.config_entries import 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, HomeAssistantError as HomeAssistantError
-from homeassistant.helpers import selector as selector
+from homeassistant.const import CONF_HOST as CONF_HOST, 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, callback as callback
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from typing import Any
 
 _LOGGER: Incomplete
 PLATFORMS: Incomplete
 MIGRATION_NAME_TO_KEY: Incomplete
-SERVICE_BASE_SCHEMA: Incomplete
-SERVICE_ADD_TORRENT_SCHEMA: Incomplete
-SERVICE_REMOVE_TORRENT_SCHEMA: Incomplete
-SERVICE_START_TORRENT_SCHEMA: Incomplete
-SERVICE_STOP_TORRENT_SCHEMA: Incomplete
 CONFIG_SCHEMA: Incomplete
 
 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: TransmissionConfigEntry) -> bool: ...
 async def async_migrate_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry) -> 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/const.pyi

@@ -20,9 +20,13 @@ DEFAULT_PATH: str
 DEFAULT_SCAN_INTERVAL: int
 STATE_ATTR_TORRENT_INFO: str
 ATTR_DELETE_DATA: str
+ATTR_LABELS: str
 ATTR_TORRENT: str
+ATTR_TORRENTS: str
 ATTR_DOWNLOAD_PATH: str
+ATTR_TORRENT_FILTER: str
 SERVICE_ADD_TORRENT: str
+SERVICE_GET_TORRENTS: str
 SERVICE_REMOVE_TORRENT: str
 SERVICE_START_TORRENT: str
 SERVICE_STOP_TORRENT: str
@@ -32,3 +36,4 @@ EVENT_DOWNLOADED_TORRENT: str
 STATE_UP_DOWN: str
 STATE_SEEDING: str
 STATE_DOWNLOADING: str
+FILTER_MODES: dict[str, list[str] | None]

homeassistant-stubs/components/transmission/entity.pyi

@@ -0,0 +1,13 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
+from _typeshed import Incomplete
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity import EntityDescription as EntityDescription
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+
+class TransmissionEntity(CoordinatorEntity[TransmissionDataUpdateCoordinator]):
+    _attr_has_entity_name: bool
+    entity_description: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, coordinator: TransmissionDataUpdateCoordinator, entity_description: EntityDescription) -> None: ...

homeassistant-stubs/components/transmission/helpers.pyi

@@ -0,0 +1,6 @@
+from transmission_rpc.torrent import Torrent as Torrent
+from typing import Any
+
+def format_torrent(torrent: Torrent) -> dict[str, Any]: ...
+def filter_torrents(torrents: list[Torrent], statuses: list[str] | None = None) -> list[Torrent]: ...
+def format_torrents(torrents: list[Torrent]) -> dict[str, dict[str, Any]]: ...

homeassistant-stubs/components/transmission/sensor.pyi

@@ -1,19 +1,17 @@
-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 .const import FILTER_MODES as FILTER_MODES, 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 TransmissionConfigEntry as TransmissionConfigEntry, TransmissionDataUpdateCoordinator as TransmissionDataUpdateCoordinator
-from _typeshed import Incomplete
+from .entity import TransmissionEntity as TransmissionEntity
+from .helpers import filter_torrents as filter_torrents
 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.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
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
-from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from transmission_rpc.torrent import Torrent as Torrent
 from typing import Any
 
-MODES: dict[str, list[str] | None]
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class TransmissionSensorEntityDescription(SensorEntityDescription):
@@ -24,17 +22,12 @@ SENSOR_TYPES: tuple[TransmissionSensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: TransmissionConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
-class TransmissionSensor(CoordinatorEntity[TransmissionDataUpdateCoordinator], SensorEntity):
+class TransmissionSensor(TransmissionEntity, SensorEntity):
     entity_description: TransmissionSensorEntityDescription
-    _attr_has_entity_name: bool
-    _attr_unique_id: Incomplete
-    _attr_device_info: Incomplete
-    def __init__(self, coordinator: TransmissionDataUpdateCoordinator, entity_description: TransmissionSensorEntityDescription) -> None: ...
     @property
     def native_value(self) -> StateType: ...
     @property
     def extra_state_attributes(self) -> dict[str, Any] | None: ...
 
 def get_state(upload: int, download: int) -> str: ...
-def _filter_torrents(torrents: list[Torrent], statuses: list[str] | None = None) -> list[Torrent]: ...
 def _torrents_info_attr(coordinator: TransmissionDataUpdateCoordinator, key: str) -> dict[str, Any]: ...

homeassistant-stubs/components/transmission/services.pyi

@@ -0,0 +1,37 @@
+from .const import ATTR_DELETE_DATA as ATTR_DELETE_DATA, ATTR_DOWNLOAD_PATH as ATTR_DOWNLOAD_PATH, ATTR_LABELS as ATTR_LABELS, ATTR_TORRENT as ATTR_TORRENT, ATTR_TORRENTS as ATTR_TORRENTS, ATTR_TORRENT_FILTER as ATTR_TORRENT_FILTER, CONF_ENTRY_ID as CONF_ENTRY_ID, DEFAULT_DELETE_DATA as DEFAULT_DELETE_DATA, DOMAIN as DOMAIN, FILTER_MODES as FILTER_MODES, SERVICE_ADD_TORRENT as SERVICE_ADD_TORRENT, SERVICE_GET_TORRENTS as SERVICE_GET_TORRENTS, 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 .helpers import filter_torrents as filter_torrents, format_torrents as format_torrents
+from _typeshed import Incomplete
+from enum import StrEnum
+from homeassistant.config_entries import ConfigEntryState as ConfigEntryState
+from homeassistant.const import CONF_ID as CONF_ID
+from homeassistant.core import HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, SupportsResponse as SupportsResponse, callback as callback
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from homeassistant.helpers import selector as selector
+from transmission_rpc import Torrent as Torrent
+from typing import Any
+
+_LOGGER: Incomplete
+
+class TorrentFilter(StrEnum):
+    ALL = 'all'
+    STARTED = 'started'
+    COMPLETED = 'completed'
+    PAUSED = 'paused'
+    ACTIVE = 'active'
+
+SERVICE_BASE_SCHEMA: Incomplete
+SERVICE_ADD_TORRENT_SCHEMA: Incomplete
+SERVICE_GET_TORRENTS_SCHEMA: Incomplete
+SERVICE_REMOVE_TORRENT_SCHEMA: Incomplete
+SERVICE_START_TORRENT_SCHEMA: Incomplete
+SERVICE_STOP_TORRENT_SCHEMA: Incomplete
+
+def _get_coordinator_from_service_data(call: ServiceCall) -> TransmissionDataUpdateCoordinator: ...
+async def _async_add_torrent(service: ServiceCall) -> None: ...
+async def _async_get_torrents(service: ServiceCall) -> dict[str, Any] | None: ...
+async def _async_start_torrent(service: ServiceCall) -> None: ...
+async def _async_stop_torrent(service: ServiceCall) -> None: ...
+async def _async_remove_torrent(service: ServiceCall) -> None: ...
+@callback
+def async_setup_services(hass: HomeAssistant) -> None: ...