homeassistant-stubs 2024.2.4__py3-none-any.whl → 2024.3.0__py3-none-any.whl

homeassistant-stubs/components/systemmonitor/coordinator.pyi

@@ -1,11 +1,12 @@
 import abc
-import psutil
+import psutil_home_assistant as ha_psutil
 from _typeshed import Incomplete
 from abc import abstractmethod
 from datetime import datetime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_component import DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+from homeassistant.helpers.update_coordinator import TimestampDataUpdateCoordinator as TimestampDataUpdateCoordinator, UpdateFailed as UpdateFailed
+from psutil import Process
 from psutil._common import sdiskusage, shwtemp, snetio, snicaddr, sswap
 from typing import NamedTuple, TypeVar
 
@@ -17,17 +18,18 @@ class VirtualMemory(NamedTuple):
     percent: float
     used: float
     free: float
-dataT = TypeVar('dataT', bound=datetime | dict[str, list[shwtemp]] | dict[str, list[snicaddr]] | dict[str, snetio] | float | list[psutil.Process] | sswap | VirtualMemory | tuple[float, float, float] | sdiskusage | None)
+dataT = TypeVar('dataT', bound=datetime | dict[str, list[shwtemp]] | dict[str, list[snicaddr]] | dict[str, snetio] | float | list[Process] | sswap | VirtualMemory | tuple[float, float, float] | sdiskusage | None)
 
-class MonitorCoordinator(DataUpdateCoordinator[dataT], metaclass=abc.ABCMeta):
-    def __init__(self, hass: HomeAssistant, name: str) -> None: ...
+class MonitorCoordinator(TimestampDataUpdateCoordinator[dataT], metaclass=abc.ABCMeta):
+    _psutil: Incomplete
+    def __init__(self, hass: HomeAssistant, psutil_wrapper: ha_psutil.PsutilWrapper, name: str) -> None: ...
     async def _async_update_data(self) -> dataT: ...
     @abstractmethod
     def update_data(self) -> dataT: ...
 
 class SystemMonitorDiskCoordinator(MonitorCoordinator[sdiskusage]):
     _argument: Incomplete
-    def __init__(self, hass: HomeAssistant, name: str, argument: str) -> None: ...
+    def __init__(self, hass: HomeAssistant, psutil_wrapper: ha_psutil.PsutilWrapper, name: str, argument: str) -> None: ...
     def update_data(self) -> sdiskusage: ...
 
 class SystemMonitorSwapCoordinator(MonitorCoordinator[sswap]):
@@ -42,17 +44,19 @@ class SystemMonitorNetIOCoordinator(MonitorCoordinator[dict[str, snetio]]):
 class SystemMonitorNetAddrCoordinator(MonitorCoordinator[dict[str, list[snicaddr]]]):
     def update_data(self) -> dict[str, list[snicaddr]]: ...
 
-class SystemMonitorLoadCoordinator(MonitorCoordinator[tuple[float, float, float]]):
-    def update_data(self) -> tuple[float, float, float]: ...
+class SystemMonitorLoadCoordinator(MonitorCoordinator[tuple[float, float, float] | None]):
+    def update_data(self) -> tuple[float, float, float] | None: ...
+    async def _async_update_data(self) -> tuple[float, float, float] | None: ...
 
 class SystemMonitorProcessorCoordinator(MonitorCoordinator[float | None]):
     def update_data(self) -> float | None: ...
+    async def _async_update_data(self) -> float | None: ...
 
 class SystemMonitorBootTimeCoordinator(MonitorCoordinator[datetime]):
     def update_data(self) -> datetime: ...
 
-class SystemMonitorProcessCoordinator(MonitorCoordinator[list[psutil.Process]]):
-    def update_data(self) -> list[psutil.Process]: ...
+class SystemMonitorProcessCoordinator(MonitorCoordinator[list[Process]]):
+    def update_data(self) -> list[Process]: ...
 
 class SystemMonitorCPUtempCoordinator(MonitorCoordinator[dict[str, list[shwtemp]]]):
     def update_data(self) -> dict[str, list[shwtemp]]: ...

homeassistant-stubs/components/systemmonitor/diagnostics.pyi

@@ -0,0 +1,7 @@
+from .const import DOMAIN_COORDINATORS as DOMAIN_COORDINATORS
+from .coordinator import MonitorCoordinator as MonitorCoordinator
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from typing import Any
+
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/systemmonitor/repairs.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from homeassistant import data_entry_flow as data_entry_flow
+from homeassistant.components.repairs import ConfirmRepairFlow as ConfirmRepairFlow, RepairsFlow as RepairsFlow
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from typing import Any
+
+class ProcessFixFlow(RepairsFlow):
+    entry: Incomplete
+    _processes: Incomplete
+    def __init__(self, entry: ConfigEntry, processes: list[str]) -> None: ...
+    async def async_step_init(self, user_input: dict[str, str] | None = None) -> data_entry_flow.FlowResult: ...
+    async def async_step_migrate_process_sensor(self, user_input: dict[str, Any] | None = None) -> data_entry_flow.FlowResult: ...
+
+async def async_create_fix_flow(hass: HomeAssistant, issue_id: str, data: dict[str, Any] | None) -> RepairsFlow: ...

homeassistant-stubs/components/systemmonitor/sensor.pyi

@@ -1,5 +1,4 @@
-import psutil
-from .const import CONF_PROCESS as CONF_PROCESS, DOMAIN as DOMAIN, NET_IO_TYPES as NET_IO_TYPES
+from .const import CONF_PROCESS as CONF_PROCESS, DOMAIN as DOMAIN, DOMAIN_COORDINATORS as DOMAIN_COORDINATORS, NET_IO_TYPES as NET_IO_TYPES
 from .coordinator import MonitorCoordinator as MonitorCoordinator, SystemMonitorBootTimeCoordinator as SystemMonitorBootTimeCoordinator, SystemMonitorCPUtempCoordinator as SystemMonitorCPUtempCoordinator, SystemMonitorDiskCoordinator as SystemMonitorDiskCoordinator, SystemMonitorLoadCoordinator as SystemMonitorLoadCoordinator, SystemMonitorMemoryCoordinator as SystemMonitorMemoryCoordinator, SystemMonitorNetAddrCoordinator as SystemMonitorNetAddrCoordinator, SystemMonitorNetIOCoordinator as SystemMonitorNetIOCoordinator, SystemMonitorProcessCoordinator as SystemMonitorProcessCoordinator, SystemMonitorProcessorCoordinator as SystemMonitorProcessorCoordinator, SystemMonitorSwapCoordinator as SystemMonitorSwapCoordinator, VirtualMemory as VirtualMemory, dataT as dataT
 from .util import get_all_disk_mounts as get_all_disk_mounts, get_all_network_interfaces as get_all_network_interfaces, read_cpu_temperature as read_cpu_temperature
 from _typeshed import Incomplete
@@ -9,12 +8,14 @@ from datetime import datetime
 from homeassistant.components.sensor import PLATFORM_SCHEMA as PLATFORM_SCHEMA, SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, SOURCE_IMPORT as SOURCE_IMPORT
 from homeassistant.const import CONF_RESOURCES as CONF_RESOURCES, CONF_TYPE as CONF_TYPE, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, UnitOfDataRate as UnitOfDataRate, UnitOfInformation as UnitOfInformation, UnitOfTemperature as UnitOfTemperature
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+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, StateType as StateType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
 from homeassistant.util import slugify as slugify
+from psutil import Process
 from psutil._common import shwtemp, snetio, snicaddr
 from typing import Any, Generic, Literal
 
@@ -29,7 +30,7 @@ SIGNAL_SYSTEMMONITOR_UPDATE: str
 
 def get_cpu_icon() -> Literal['mdi:cpu-64-bit', 'mdi:cpu-32-bit']: ...
 def get_processor_temperature(entity: SystemMonitorSensor[dict[str, list[shwtemp]]]) -> float | None: ...
-def get_process(entity: SystemMonitorSensor[list[psutil.Process]]) -> str: ...
+def get_process(entity: SystemMonitorSensor[list[Process]]) -> str: ...
 def get_network(entity: SystemMonitorSensor[dict[str, snetio]]) -> float | None: ...
 def get_packets(entity: SystemMonitorSensor[dict[str, snetio]]) -> float | None: ...
 def get_throughput(entity: SystemMonitorSensor[dict[str, snetio]]) -> float | None: ...

homeassistant-stubs/components/systemmonitor/util.pyi

@@ -1,11 +1,12 @@
-from .const import CPU_SENSOR_PREFIXES as CPU_SENSOR_PREFIXES
+from .const import CPU_SENSOR_PREFIXES as CPU_SENSOR_PREFIXES, DOMAIN as DOMAIN
 from _typeshed import Incomplete
+from homeassistant.core import HomeAssistant as HomeAssistant
 from psutil._common import shwtemp as shwtemp
 
 _LOGGER: Incomplete
 SKIP_DISK_TYPES: Incomplete
 
-def get_all_disk_mounts() -> set[str]: ...
-def get_all_network_interfaces() -> set[str]: ...
-def get_all_running_processes() -> set[str]: ...
-def read_cpu_temperature(temps: dict[str, list[shwtemp]] | None = None) -> float | None: ...
+def get_all_disk_mounts(hass: HomeAssistant) -> set[str]: ...
+def get_all_network_interfaces(hass: HomeAssistant) -> set[str]: ...
+def get_all_running_processes(hass: HomeAssistant) -> set[str]: ...
+def read_cpu_temperature(hass: HomeAssistant, temps: dict[str, list[shwtemp]] | None = None) -> float | None: ...

homeassistant-stubs/components/technove/helpers.pyi

@@ -0,0 +1,9 @@
+from .entity import TechnoVEEntity as TechnoVEEntity
+from collections.abc import Callable as Callable, Coroutine
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from typing import Any, Concatenate, ParamSpec, TypeVar
+
+_TechnoVEEntityT = TypeVar('_TechnoVEEntityT', bound=TechnoVEEntity)
+_P = ParamSpec('_P')
+
+def technove_exception_handler(func: Callable[Concatenate[_TechnoVEEntityT, _P], Coroutine[Any, Any, Any]]) -> Callable[Concatenate[_TechnoVEEntityT, _P], Coroutine[Any, Any, None]]: ...

homeassistant-stubs/components/technove/switch.pyi

@@ -0,0 +1,33 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import TechnoVEDataUpdateCoordinator as TechnoVEDataUpdateCoordinator
+from .entity import TechnoVEEntity as TechnoVEEntity
+from .helpers import technove_exception_handler as technove_exception_handler
+from _typeshed import Incomplete
+from collections.abc import Awaitable, 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.const import EntityCategory as EntityCategory
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from technove import Station as TechnoVEStation, TechnoVE as TechnoVE
+from typing import Any
+
+@dataclass(frozen=True, kw_only=True)
+class TechnoVESwitchDescription(SwitchEntityDescription):
+    is_on_fn: Callable[[TechnoVEStation], bool]
+    turn_on_fn: Callable[[TechnoVE], Awaitable[dict[str, Any]]]
+    turn_off_fn: Callable[[TechnoVE], Awaitable[dict[str, Any]]]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, is_on_fn, turn_on_fn, turn_off_fn) -> None: ...
+
+SWITCHES: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TechnoVESwitchEntity(TechnoVEEntity, SwitchEntity):
+    entity_description: TechnoVESwitchDescription
+    def __init__(self, coordinator: TechnoVEDataUpdateCoordinator, description: TechnoVESwitchDescription) -> None: ...
+    @property
+    def is_on(self) -> bool: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/tedee/config_flow.pyi

@@ -10,3 +10,4 @@ class TedeeConfigFlow(ConfigFlow, domain=DOMAIN):
     reauth_entry: ConfigEntry | None
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult: ...
+    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...

homeassistant-stubs/components/tedee/sensor.pyi

@@ -4,7 +4,7 @@ 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 PERCENTAGE as PERCENTAGE, UnitOfTime as UnitOfTime
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from pytedee_async import TedeeLock as TedeeLock

homeassistant-stubs/components/tibber/sensor.pyi

@@ -21,12 +21,15 @@ ICON: str
 SCAN_INTERVAL: Incomplete
 MIN_TIME_BETWEEN_UPDATES: Incomplete
 PARALLEL_UPDATES: int
+RT_SENSORS_UNIQUE_ID_MIGRATION: Incomplete
+RT_SENSORS_UNIQUE_ID_MIGRATION_SIMPLE: Incomplete
 RT_SENSORS: tuple[SensorEntityDescription, ...]
 SENSORS: tuple[SensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TibberSensor(SensorEntity):
+    _attr_has_entity_name: bool
     _tibber_home: Incomplete
     _home_name: Incomplete
     _device_name: Incomplete
@@ -36,13 +39,13 @@ class TibberSensor(SensorEntity):
     def device_info(self) -> DeviceInfo: ...
 
 class TibberSensorElPrice(TibberSensor):
+    _attr_state_class: Incomplete
+    _attr_translation_key: str
     _last_updated: Incomplete
     _spread_load_constant: Incomplete
     _attr_available: bool
     _attr_extra_state_attributes: Incomplete
     _attr_icon: Incomplete
-    _attr_name: Incomplete
-    _attr_state_class: Incomplete
     _attr_unique_id: Incomplete
     _model: str
     _device_name: Incomplete
@@ -54,7 +57,6 @@ class TibberSensorElPrice(TibberSensor):
 class TibberDataSensor(TibberSensor, CoordinatorEntity['TibberDataCoordinator']):
     entity_description: Incomplete
     _attr_unique_id: Incomplete
-    _attr_name: Incomplete
     _attr_native_unit_of_measurement: Incomplete
     _device_name: Incomplete
     def __init__(self, tibber_home: tibber.TibberHome, coordinator: TibberDataCoordinator, entity_description: SensorEntityDescription) -> None: ...
@@ -65,7 +67,6 @@ class TibberSensorRT(TibberSensor, CoordinatorEntity['TibberRtDataCoordinator'])
     entity_description: Incomplete
     _model: str
     _device_name: Incomplete
-    _attr_name: Incomplete
     _attr_native_value: Incomplete
     _attr_unique_id: Incomplete
     _attr_native_unit_of_measurement: Incomplete
@@ -81,8 +82,10 @@ class TibberRtDataCoordinator(DataUpdateCoordinator):
     hass: Incomplete
     _added_sensors: 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 _handle_ha_stop(self, _event: Event) -> None: ...
+    def _migrate_unique_id(self, sensor_description: SensorEntityDescription) -> None: ...
     def _add_sensors(self) -> None: ...
     def get_live_measurement(self) -> Any: ...
 

homeassistant-stubs/components/tplink/__init__.pyi

@@ -6,7 +6,7 @@ from aiohttp import ClientSession as ClientSession
 from homeassistant import config_entries as config_entries
 from homeassistant.components import network as network
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_ALIAS as CONF_ALIAS, CONF_AUTHENTICATION as CONF_AUTHENTICATION, 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, EVENT_HOMEASSISTANT_STARTED as EVENT_HOMEASSISTANT_STARTED
+from homeassistant.const import CONF_ALIAS as CONF_ALIAS, CONF_AUTHENTICATION as CONF_AUTHENTICATION, 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 HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers import discovery_flow as discovery_flow

homeassistant-stubs/components/tplink/config_flow.pyi

@@ -7,7 +7,7 @@ from homeassistant.components import dhcp as dhcp
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, SOURCE_REAUTH as SOURCE_REAUTH
 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.data_entry_flow import AbortFlow as AbortFlow, FlowResult as FlowResult
+from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
 from kasa import Credentials, SmartDevice
 from typing import Any
@@ -23,7 +23,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
     def __init__(self) -> None: ...
     async def async_step_dhcp(self, discovery_info: dhcp.DhcpServiceInfo) -> FlowResult: ...
     async def async_step_integration_discovery(self, discovery_info: DiscoveryInfoType) -> FlowResult: ...
-    def _update_config_if_entry_in_setup_error(self, entry: ConfigEntry, host: str, config: dict) -> None: ...
+    def _update_config_if_entry_in_setup_error(self, entry: ConfigEntry, host: str, config: dict) -> FlowResult | None: ...
     async def _async_handle_discovery(self, host: str, formatted_mac: str, config: dict | None = None) -> FlowResult: ...
     async def async_step_discovery_auth_confirm(self, user_input: dict[str, Any] | None = None) -> FlowResult: ...
     def _async_make_placeholders_from_discovery(self) -> dict[str, str]: ...

homeassistant-stubs/components/tplink/light.pyi

@@ -31,6 +31,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
 class TPLinkSmartBulb(CoordinatedTPLinkEntity, LightEntity):
     _attr_supported_features: Incomplete
     _attr_name: Incomplete
+    _fixed_color_mode: ColorMode | None
     device: SmartBulb
     _attr_unique_id: Incomplete
     _attr_min_color_temp_kelvin: Incomplete

homeassistant-stubs/components/tplink_omada/binary_sensor.pyi

@@ -7,7 +7,7 @@ from homeassistant.components.binary_sensor import BinarySensorDeviceClass as Bi
 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 tplink_omada_client.devices import OmadaDevice as OmadaDevice, OmadaGateway, OmadaGatewayPort as OmadaGatewayPort
+from tplink_omada_client.devices import OmadaDevice as OmadaDevice, OmadaGateway, OmadaGatewayPortStatus as OmadaGatewayPortStatus
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 def get_gateway_port_status_sensors(gateway: OmadaGateway, hass: HomeAssistant, coordinator: OmadaGatewayCoordinator) -> Generator[BinarySensorEntity, None, None]: ...
@@ -17,7 +17,7 @@ class GatewayPortBinarySensorConfig:
     id_suffix: str
     name_suffix: str
     device_class: BinarySensorDeviceClass
-    update_func: Callable[[OmadaGatewayPort], bool]
+    update_func: Callable[[OmadaGatewayPortStatus], bool]
     def __init__(self, port_number, id_suffix, name_suffix, device_class, update_func) -> None: ...
     def __lt__(self, other): ...
     def __le__(self, other): ...

homeassistant-stubs/components/tplink_omada/config_flow.pyi

@@ -7,7 +7,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.data_entry_flow import FlowResult as FlowResult
 from homeassistant.helpers import selector as selector
 from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession, async_get_clientsession as async_get_clientsession
-from tplink_omada_client.omadaclient import OmadaClient, OmadaSite as OmadaSite
+from tplink_omada_client import OmadaClient, OmadaSite as OmadaSite
 from types import MappingProxyType
 from typing import Any, NamedTuple
 

homeassistant-stubs/components/tplink_omada/controller.pyi

@@ -1,8 +1,8 @@
 from .coordinator import OmadaCoordinator as OmadaCoordinator
 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
-from tplink_omada_client.omadasiteclient import OmadaSiteClient as OmadaSiteClient
 
 POLL_SWITCH_PORT: int
 POLL_GATEWAY: int

homeassistant-stubs/components/tplink_omada/coordinator.pyi

@@ -1,7 +1,7 @@
 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.omadaclient import OmadaSiteClient as OmadaSiteClient
+from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
 from typing import Generic, TypeVar
 
 _LOGGER: Incomplete

homeassistant-stubs/components/tplink_omada/update.pyi

@@ -8,8 +8,8 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from tplink_omada_client import OmadaSiteClient as OmadaSiteClient
 from tplink_omada_client.devices import OmadaFirmwareUpdate as OmadaFirmwareUpdate, OmadaListDevice as OmadaListDevice
-from tplink_omada_client.omadasiteclient import OmadaSiteClient as OmadaSiteClient
 from typing import Any, NamedTuple
 
 POLL_DELAY_IDLE: Incomplete

homeassistant-stubs/components/tractive/device_tracker.pyi

@@ -12,7 +12,6 @@ from typing import Any
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TractiveDeviceTracker(TractiveEntity, TrackerEntity):
-    _attr_icon: str
     _attr_translation_key: str
     _battery_level: Incomplete
     _latitude: Incomplete

homeassistant-stubs/components/unifi/__init__.pyi

@@ -1,6 +1,6 @@
 from .const import PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
-from .controller import UniFiController as UniFiController, get_unifi_controller as get_unifi_controller
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
 from .services import async_setup_services as async_setup_services, async_unload_services as async_unload_services
 from _typeshed import Incomplete
 from aiounifi.models.client import Client as Client

homeassistant-stubs/components/unifi/button.pyi

@@ -1,6 +1,6 @@
 import aiounifi
-from .controller import UniFiController as UniFiController
 from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn, async_device_device_info_fn as async_device_device_info_fn
+from .hub import UnifiHub as UnifiHub
 from aiounifi.interfaces.api_handlers import ItemEvent as ItemEvent
 from aiounifi.models.api import ApiItemT
 from collections.abc import Callable as Callable, Coroutine

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -1,6 +1,6 @@
 from .const import CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DOMAIN as UNIFI_DOMAIN
-from .controller import UniFiController as UniFiController, get_unifi_controller as get_unifi_controller
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
 from _typeshed import Incomplete
 from aiounifi.interfaces.sites import Sites as Sites
 from collections.abc import Mapping
@@ -32,7 +32,7 @@ class UnifiFlowHandler(config_entries.ConfigFlow, domain=UNIFI_DOMAIN):
     async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> FlowResult: ...
 
 class UnifiOptionsFlowHandler(config_entries.OptionsFlow):
-    controller: UniFiController
+    hub: UnifiHub
     config_entry: Incomplete
     options: Incomplete
     def __init__(self, config_entry: config_entries.ConfigEntry) -> None: ...

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -1,6 +1,6 @@
 import aiounifi
-from .controller import UNIFI_DOMAIN as UNIFI_DOMAIN, UniFiController as UniFiController
 from .entity import HandlerT as HandlerT, UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription, async_device_available_fn as async_device_available_fn
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from aiounifi.interfaces.api_handlers import ItemEvent
 from aiounifi.models.api import ApiItemT
@@ -26,15 +26,15 @@ WIRED_DISCONNECTION: Incomplete
 WIRELESS_CONNECTION: Incomplete
 WIRELESS_DISCONNECTION: Incomplete
 
-def async_client_allowed_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_client_is_connected_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_device_heartbeat_timedelta_fn(controller: UniFiController, obj_id: str) -> timedelta: ...
+def async_client_allowed_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_client_is_connected_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_device_heartbeat_timedelta_fn(hub: UnifiHub, obj_id: str) -> timedelta: ...
 
 @dataclass(frozen=True)
 class UnifiEntityTrackerDescriptionMixin(Generic[HandlerT, ApiItemT]):
-    heartbeat_timedelta_fn: Callable[[UniFiController, str], timedelta]
+    heartbeat_timedelta_fn: Callable[[UnifiHub, str], timedelta]
     ip_address_fn: Callable[[aiounifi.Controller, str], str | None]
-    is_connected_fn: Callable[[UniFiController, str], bool]
+    is_connected_fn: Callable[[UnifiHub, str], bool]
     hostname_fn: Callable[[aiounifi.Controller, str], str | None]
     def __init__(self, heartbeat_timedelta_fn, ip_address_fn, is_connected_fn, hostname_fn) -> None: ...
 

homeassistant-stubs/components/unifi/diagnostics.pyi

@@ -1,4 +1,4 @@
-from .controller import UniFiController as UniFiController
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from collections.abc import Mapping
 from homeassistant.components.diagnostics import REDACTED as REDACTED, async_redact_data as async_redact_data

homeassistant-stubs/components/unifi/entity.pyi

@@ -1,7 +1,7 @@
 import abc
 import aiounifi
 from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, DOMAIN as DOMAIN
-from .controller import UniFiController as UniFiController
+from .hub import UnifiHub as UnifiHub
 from _typeshed import Incomplete
 from abc import abstractmethod
 from aiounifi.interfaces.api_handlers import APIHandler, CallbackType, ItemEvent, UnsubscribeType
@@ -18,25 +18,25 @@ from typing import Generic, TypeVar
 HandlerT = TypeVar('HandlerT', bound=APIHandler)
 SubscriptionT = Callable[[CallbackType, ItemEvent], UnsubscribeType]
 
-def async_device_available_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_wlan_available_fn(controller: UniFiController, obj_id: str) -> bool: ...
-def async_device_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
-def async_wlan_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
-def async_client_device_info_fn(controller: UniFiController, obj_id: str) -> DeviceInfo: ...
+def async_device_available_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_wlan_available_fn(hub: UnifiHub, obj_id: str) -> bool: ...
+def async_device_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+def async_wlan_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
+def async_client_device_info_fn(hub: UnifiHub, obj_id: str) -> DeviceInfo: ...
 
 @dataclass(frozen=True)
 class UnifiDescription(Generic[HandlerT, ApiItemT]):
-    allowed_fn: Callable[[UniFiController, str], bool]
+    allowed_fn: Callable[[UnifiHub, str], bool]
     api_handler_fn: Callable[[aiounifi.Controller], HandlerT]
-    available_fn: Callable[[UniFiController, str], bool]
-    device_info_fn: Callable[[UniFiController, str], DeviceInfo | None]
+    available_fn: Callable[[UnifiHub, str], bool]
+    device_info_fn: Callable[[UnifiHub, str], DeviceInfo | None]
     event_is_on: tuple[EventKey, ...] | None
     event_to_subscribe: tuple[EventKey, ...] | None
     name_fn: Callable[[ApiItemT], str | None]
     object_fn: Callable[[aiounifi.Controller, str], ApiItemT]
     should_poll: bool
-    supported_fn: Callable[[UniFiController, str], bool | None]
-    unique_id_fn: Callable[[UniFiController, str], str]
+    supported_fn: Callable[[UnifiHub, str], bool | None]
+    unique_id_fn: Callable[[UnifiHub, str], str]
     def __init__(self, allowed_fn, api_handler_fn, available_fn, device_info_fn, event_is_on, event_to_subscribe, name_fn, object_fn, should_poll, supported_fn, unique_id_fn) -> None: ...
 
 @dataclass(frozen=True)
@@ -47,13 +47,13 @@ class UnifiEntity(Entity, Generic[HandlerT, ApiItemT], metaclass=abc.ABCMeta):
     entity_description: UnifiEntityDescription[HandlerT, ApiItemT]
     _attr_unique_id: str
     _obj_id: Incomplete
-    controller: Incomplete
+    hub: Incomplete
     _removed: bool
     _attr_available: Incomplete
     _attr_device_info: Incomplete
     _attr_should_poll: Incomplete
     _attr_name: Incomplete
-    def __init__(self, obj_id: str, controller: UniFiController, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
+    def __init__(self, obj_id: str, hub: UnifiHub, description: UnifiEntityDescription[HandlerT, ApiItemT]) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     def async_signalling_callback(self, event: ItemEvent, obj_id: str) -> None: ...
     def async_signal_reachable_callback(self) -> None: ...

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

@@ -0,0 +1,2 @@
+from .api import get_unifi_api as get_unifi_api
+from .hub import UnifiHub as UnifiHub

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

@@ -0,0 +1,10 @@
+import aiounifi
+from ..const import CONF_SITE_ID as CONF_SITE_ID, LOGGER as LOGGER
+from ..errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+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
+from homeassistant.helpers import aiohttp_client as aiohttp_client
+from types import MappingProxyType
+from typing import Any
+
+async def get_unifi_api(hass: HomeAssistant, config: MappingProxyType[str, Any]) -> aiounifi.Controller: ...

homeassistant-stubs/components/unifi/{controller.pyi → hub/hub.pyi}

@@ -1,32 +1,25 @@
 import aiounifi
-from .const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_ALLOW_BANDWIDTH_SENSORS as DEFAULT_ALLOW_BANDWIDTH_SENSORS, DEFAULT_ALLOW_UPTIME_SENSORS as DEFAULT_ALLOW_UPTIME_SENSORS, DEFAULT_DETECTION_TIME as DEFAULT_DETECTION_TIME, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DEFAULT_IGNORE_WIRED_BUG as DEFAULT_IGNORE_WIRED_BUG, DEFAULT_TRACK_CLIENTS as DEFAULT_TRACK_CLIENTS, DEFAULT_TRACK_DEVICES as DEFAULT_TRACK_DEVICES, DEFAULT_TRACK_WIRED_CLIENTS as DEFAULT_TRACK_WIRED_CLIENTS, LOGGER as LOGGER, PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
-from .entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
-from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
+from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, CONF_ALLOW_BANDWIDTH_SENSORS as CONF_ALLOW_BANDWIDTH_SENSORS, CONF_ALLOW_UPTIME_SENSORS as CONF_ALLOW_UPTIME_SENSORS, CONF_BLOCK_CLIENT as CONF_BLOCK_CLIENT, CONF_CLIENT_SOURCE as CONF_CLIENT_SOURCE, CONF_DETECTION_TIME as CONF_DETECTION_TIME, CONF_DPI_RESTRICTIONS as CONF_DPI_RESTRICTIONS, CONF_IGNORE_WIRED_BUG as CONF_IGNORE_WIRED_BUG, CONF_SITE_ID as CONF_SITE_ID, CONF_SSID_FILTER as CONF_SSID_FILTER, CONF_TRACK_CLIENTS as CONF_TRACK_CLIENTS, CONF_TRACK_DEVICES as CONF_TRACK_DEVICES, CONF_TRACK_WIRED_CLIENTS as CONF_TRACK_WIRED_CLIENTS, DEFAULT_ALLOW_BANDWIDTH_SENSORS as DEFAULT_ALLOW_BANDWIDTH_SENSORS, DEFAULT_ALLOW_UPTIME_SENSORS as DEFAULT_ALLOW_UPTIME_SENSORS, DEFAULT_DETECTION_TIME as DEFAULT_DETECTION_TIME, DEFAULT_DPI_RESTRICTIONS as DEFAULT_DPI_RESTRICTIONS, DEFAULT_IGNORE_WIRED_BUG as DEFAULT_IGNORE_WIRED_BUG, DEFAULT_TRACK_CLIENTS as DEFAULT_TRACK_CLIENTS, DEFAULT_TRACK_DEVICES as DEFAULT_TRACK_DEVICES, DEFAULT_TRACK_WIRED_CLIENTS as DEFAULT_TRACK_WIRED_CLIENTS, PLATFORMS as PLATFORMS, UNIFI_WIRELESS_CLIENTS as UNIFI_WIRELESS_CLIENTS
+from ..entity import UnifiEntity as UnifiEntity, UnifiEntityDescription as UnifiEntityDescription
+from .websocket import UnifiWebsocket as UnifiWebsocket
 from _typeshed import Incomplete
 from datetime import datetime
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL, Platform as Platform
+from homeassistant.const import CONF_HOST as CONF_HOST, Platform as Platform
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers import aiohttp_client as aiohttp_client
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry, DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.entity_registry import async_entries_for_config_entry as async_entries_for_config_entry
 from homeassistant.helpers.event import async_call_later as async_call_later, async_track_time_interval as async_track_time_interval
-from types import MappingProxyType
-from typing import Any
 
-RETRY_TIMER: int
 CHECK_HEARTBEAT_INTERVAL: Incomplete
-CHECK_WEBSOCKET_INTERVAL: Incomplete
 
-class UniFiController:
+class UnifiHub:
     hass: Incomplete
     config_entry: Incomplete
     api: Incomplete
-    ws_task: Incomplete
-    _cancel_websocket_check: Incomplete
-    available: bool
+    websocket: Incomplete
     wireless_clients: Incomplete
     site: Incomplete
     is_admin: bool
@@ -51,8 +44,11 @@ class UniFiController:
     def load_config_entry_options(self) -> None: ...
     @property
     def host(self) -> str: ...
+    @property
+    def available(self) -> bool: ...
     @staticmethod
     def register_platform(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, entity_class: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], requires_admin: bool = False) -> None: ...
+    def _async_should_add_entity(self, description: UnifiEntityDescription, obj_id: str) -> bool: ...
     def register_platform_add_entities(self, unifi_platform_entity: type[UnifiEntity], descriptions: tuple[UnifiEntityDescription, ...], async_add_entities: AddEntitiesCallback) -> None: ...
     @property
     def signal_reachable(self) -> str: ...
@@ -69,11 +65,5 @@ class UniFiController:
     def async_update_device_registry(self) -> DeviceEntry: ...
     @staticmethod
     async def async_config_entry_updated(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
-    def start_websocket(self) -> None: ...
-    def reconnect(self, log: bool = False) -> None: ...
-    async def async_reconnect(self) -> None: ...
-    def _async_watch_websocket(self, now: datetime) -> None: ...
     def shutdown(self, event: Event) -> None: ...
     async def async_reset(self) -> bool: ...
-
-async def get_unifi_controller(hass: HomeAssistant, config: MappingProxyType[str, Any]) -> aiounifi.Controller: ...

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

@@ -0,0 +1,25 @@
+import aiounifi
+from ..const import LOGGER as LOGGER
+from _typeshed import Incomplete
+from datetime import datetime
+from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
+from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
+
+RETRY_TIMER: int
+CHECK_WEBSOCKET_INTERVAL: Incomplete
+
+class UnifiWebsocket:
+    hass: Incomplete
+    api: Incomplete
+    signal: Incomplete
+    ws_task: Incomplete
+    _cancel_websocket_check: Incomplete
+    available: bool
+    def __init__(self, hass: HomeAssistant, api: aiounifi.Controller, signal: str) -> None: ...
+    def start(self) -> None: ...
+    def stop(self) -> None: ...
+    async def stop_and_wait(self) -> None: ...
+    def start_websocket(self) -> None: ...
+    def reconnect(self, log: bool = False) -> None: ...
+    def _async_watch_websocket(self, now: datetime) -> None: ...