homeassistant-stubs 2024.2.5__py3-none-any.whl → 2024.3.0b0__py3-none-any.whl

homeassistant-stubs/components/shelly/sensor.pyi

@@ -1,7 +1,7 @@
 from .const import CONF_SLEEP_PERIOD as CONF_SLEEP_PERIOD, SHAIR_MAX_WORK_HOURS as SHAIR_MAX_WORK_HOURS
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyRpcCoordinator as ShellyRpcCoordinator
 from .entity import BlockEntityDescription as BlockEntityDescription, RestEntityDescription as RestEntityDescription, RpcEntityDescription as RpcEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, ShellyRestAttributeEntity as ShellyRestAttributeEntity, ShellyRpcAttributeEntity as ShellyRpcAttributeEntity, ShellySleepingBlockAttributeEntity as ShellySleepingBlockAttributeEntity, ShellySleepingRpcAttributeEntity as ShellySleepingRpcAttributeEntity, async_setup_entry_attribute_entities as async_setup_entry_attribute_entities, async_setup_entry_rest as async_setup_entry_rest, async_setup_entry_rpc as async_setup_entry_rpc
-from .utils import get_device_entry_gen as get_device_entry_gen, get_device_uptime as get_device_uptime
+from .utils import get_device_entry_gen as get_device_entry_gen, get_device_uptime as get_device_uptime, is_rpc_wifi_stations_disabled as is_rpc_wifi_stations_disabled
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from dataclasses import dataclass
@@ -16,7 +16,7 @@ from typing import Final
 
 @dataclass(frozen=True)
 class BlockSensorDescription(BlockEntityDescription, SensorEntityDescription):
-    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, icon_fn, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
 
 @dataclass(frozen=True)
 class RpcSensorDescription(RpcEntityDescription, SensorEntityDescription):

homeassistant-stubs/components/shelly/switch.pyi

@@ -1,7 +1,7 @@
-from .const import DOMAIN as DOMAIN, GAS_VALVE_OPEN_STATES as GAS_VALVE_OPEN_STATES, MODEL_WALL_DISPLAY as MODEL_WALL_DISPLAY
+from .const import DOMAIN as DOMAIN, GAS_VALVE_OPEN_STATES as GAS_VALVE_OPEN_STATES
 from .coordinator import ShellyBlockCoordinator as ShellyBlockCoordinator, ShellyRpcCoordinator as ShellyRpcCoordinator, get_entry_data as get_entry_data
 from .entity import BlockEntityDescription as BlockEntityDescription, ShellyBlockAttributeEntity as ShellyBlockAttributeEntity, ShellyBlockEntity as ShellyBlockEntity, ShellyRpcEntity as ShellyRpcEntity, async_setup_block_attribute_entities as async_setup_block_attribute_entities
-from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, is_block_channel_type_light as is_block_channel_type_light, is_rpc_channel_type_light as is_rpc_channel_type_light
+from .utils import async_remove_shelly_entity as async_remove_shelly_entity, get_device_entry_gen as get_device_entry_gen, get_rpc_key_ids as get_rpc_key_ids, is_block_channel_type_light as is_block_channel_type_light, is_rpc_channel_type_light as is_rpc_channel_type_light, is_rpc_thermostat_internal_actuator as is_rpc_thermostat_internal_actuator
 from _typeshed import Incomplete
 from aioshelly.block_device import Block as Block
 from dataclasses import dataclass
@@ -16,7 +16,7 @@ from typing import Any
 
 @dataclass(frozen=True)
 class BlockSwitchDescription(BlockEntityDescription, SwitchEntityDescription):
-    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, icon_fn, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
 
 GAS_VALVE_SWITCH: Incomplete
 
@@ -26,12 +26,11 @@ def async_setup_rpc_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_
 
 class BlockValveSwitch(ShellyBlockAttributeEntity, SwitchEntity):
     entity_description: BlockSwitchDescription
+    _attr_translation_key: str
     control_result: Incomplete
     def __init__(self, coordinator: ShellyBlockCoordinator, block: Block, attribute: str, description: BlockSwitchDescription) -> None: ...
     @property
     def is_on(self) -> bool: ...
-    @property
-    def icon(self) -> str: ...
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     async def async_added_to_hass(self) -> None: ...

homeassistant-stubs/components/shelly/utils.pyi

@@ -45,6 +45,7 @@ def get_rpc_key_ids(keys_dict: dict[str, Any], key: str) -> list[int]: ...
 def is_rpc_momentary_input(config: dict[str, Any], status: dict[str, Any], key: str) -> bool: ...
 def is_block_channel_type_light(settings: dict[str, Any], channel: int) -> bool: ...
 def is_rpc_channel_type_light(config: dict[str, Any], channel: int) -> bool: ...
+def is_rpc_thermostat_internal_actuator(status: dict[str, Any]) -> bool: ...
 def get_rpc_input_triggers(device: RpcDevice) -> list[tuple[str, str]]: ...
 def update_device_fw_info(hass: HomeAssistant, shellydevice: BlockDevice | RpcDevice, entry: ConfigEntry) -> None: ...
 def brightness_to_percentage(brightness: int) -> int: ...
@@ -52,3 +53,4 @@ def percentage_to_brightness(percentage: int) -> int: ...
 def mac_address_from_name(name: str) -> str | None: ...
 def get_release_url(gen: int, model: str, beta: bool) -> str | None: ...
 def async_create_issue_unsupported_firmware(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+def is_rpc_wifi_stations_disabled(config: dict[str, Any], _status: dict[str, Any], key: str) -> bool: ...

homeassistant-stubs/components/shelly/valve.pyi

@@ -12,7 +12,7 @@ from typing import Any
 
 @dataclass(kw_only=True, frozen=True)
 class BlockValveDescription(BlockEntityDescription, ValveEntityDescription):
-    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, reports_position, icon_fn, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, reports_position, unit_fn, value, available, removal_condition, extra_state_attributes) -> None: ...
 
 GAS_VALVE: Incomplete
 

homeassistant-stubs/components/ssdp/__init__.pyi

@@ -17,6 +17,7 @@ from homeassistant.helpers.network import NoURLAvailableError as NoURLAvailableE
 from homeassistant.helpers.system_info import async_get_system_info as async_get_system_info
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import async_get_ssdp as async_get_ssdp, bind_hass as bind_hass
+from homeassistant.util.async_ import create_eager_task as create_eager_task
 from ipaddress import IPv4Address, IPv6Address
 from typing import Any
 

homeassistant-stubs/components/statistics/sensor.pyi

@@ -56,6 +56,7 @@ STATS_BINARY_PERCENTAGE: Incomplete
 CONF_STATE_CHARACTERISTIC: str
 CONF_SAMPLES_MAX_BUFFER_SIZE: str
 CONF_MAX_AGE: str
+CONF_KEEP_LAST_SAMPLE: str
 CONF_PRECISION: str
 CONF_PERCENTILE: str
 DEFAULT_NAME: str
@@ -64,6 +65,7 @@ ICON: str
 
 def valid_state_characteristic_configuration(config: dict[str, Any]) -> dict[str, Any]: ...
 def valid_boundary_configuration(config: dict[str, Any]) -> dict[str, Any]: ...
+def valid_keep_last_sample(config: dict[str, Any]) -> dict[str, Any]: ...
 
 _PLATFORM_SCHEMA_BASE: Incomplete
 
@@ -79,6 +81,7 @@ class StatisticsSensor(SensorEntity):
     _state_characteristic: Incomplete
     _samples_max_buffer_size: Incomplete
     _samples_max_age: Incomplete
+    samples_keep_last: Incomplete
     _precision: Incomplete
     _percentile: Incomplete
     _value: Incomplete
@@ -89,7 +92,7 @@ class StatisticsSensor(SensorEntity):
     attributes: Incomplete
     _state_characteristic_fn: Incomplete
     _update_listener: Incomplete
-    def __init__(self, source_entity_id: str, name: str, unique_id: str | None, state_characteristic: str, samples_max_buffer_size: int | None, samples_max_age: timedelta | None, precision: int, percentile: int) -> None: ...
+    def __init__(self, source_entity_id: str, name: str, unique_id: str | None, state_characteristic: str, samples_max_buffer_size: int | None, samples_max_age: timedelta | None, samples_keep_last: bool, precision: int, percentile: int) -> None: ...
     async def async_added_to_hass(self) -> None: ...
     def _add_state_to_queue(self, new_state: State) -> None: ...
     def _derive_unit_of_measurement(self, new_state: State) -> str | None: ...

homeassistant-stubs/components/system_health/__init__.pyi

@@ -18,7 +18,7 @@ class SystemHealthProtocol(Protocol):
 
 def async_register_info(hass: HomeAssistant, domain: str, info_callback: Callable[[HomeAssistant], Awaitable[dict]]) -> None: ...
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
-async def _register_system_health_platform(hass: HomeAssistant, integration_domain: str, platform: SystemHealthProtocol) -> None: ...
+def _register_system_health_platform(hass: HomeAssistant, integration_domain: str, platform: SystemHealthProtocol) -> None: ...
 async def get_integration_info(hass: HomeAssistant, registration: SystemHealthRegistration) -> dict[str, Any]: ...
 def _format_value(val: Any) -> Any: ...
 async def handle_info(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/systemmonitor/__init__.pyi

@@ -1,10 +1,13 @@
+from .const import DOMAIN as DOMAIN
 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
 
+_LOGGER: Incomplete
 PLATFORMS: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/systemmonitor/binary_sensor.pyi

@@ -0,0 +1,48 @@
+from .const import CONF_PROCESS as CONF_PROCESS, DOMAIN as DOMAIN
+from .coordinator import MonitorCoordinator as MonitorCoordinator, SystemMonitorProcessCoordinator as SystemMonitorProcessCoordinator, dataT as dataT
+from _typeshed import Incomplete
+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.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import EntityCategory as EntityCategory
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+from homeassistant.util import slugify as slugify
+from psutil import Process
+from typing import Generic, Literal
+
+_LOGGER: Incomplete
+CONF_ARG: str
+SENSOR_TYPE_NAME: int
+SENSOR_TYPE_UOM: int
+SENSOR_TYPE_ICON: int
+SENSOR_TYPE_DEVICE_CLASS: int
+SENSOR_TYPE_MANDATORY_ARG: int
+SIGNAL_SYSTEMMONITOR_UPDATE: str
+
+def get_cpu_icon() -> Literal['mdi:cpu-64-bit', 'mdi:cpu-32-bit']: ...
+def get_process(entity: SystemMonitorSensor[list[Process]]) -> bool: ...
+
+@dataclass(frozen=True, kw_only=True)
+class SysMonitorBinarySensorEntityDescription(BinarySensorEntityDescription, Generic[dataT]):
+    value_fn: Callable[[SystemMonitorSensor[dataT]], bool]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, value_fn) -> None: ...
+
+SENSOR_TYPES: tuple[SysMonitorBinarySensorEntityDescription[list[Process]], ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class SystemMonitorSensor(CoordinatorEntity[MonitorCoordinator[dataT]], BinarySensorEntity):
+    _attr_has_entity_name: bool
+    _attr_entity_category: Incomplete
+    entity_description: SysMonitorBinarySensorEntityDescription[dataT]
+    _attr_translation_placeholders: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
+    argument: Incomplete
+    def __init__(self, coordinator: MonitorCoordinator[dataT], sensor_description: SysMonitorBinarySensorEntityDescription[dataT], entry_id: str, argument: str) -> None: ...
+    @property
+    def is_on(self) -> bool | None: ...

homeassistant-stubs/components/systemmonitor/config_flow.pyi

@@ -22,5 +22,7 @@ OPTIONS_FLOW: Incomplete
 class SystemMonitorConfigFlowHandler(SchemaConfigFlowHandler, domain=DOMAIN):
     config_flow = CONFIG_FLOW
     options_flow = OPTIONS_FLOW
+    VERSION: int
+    MINOR_VERSION: int
     def async_config_entry_title(self, options: Mapping[str, Any]) -> str: ...
     def async_create_entry(self, data: Mapping[str, Any], **kwargs: Any) -> FlowResult: ...

homeassistant-stubs/components/systemmonitor/const.pyi

@@ -1,6 +1,7 @@
 from _typeshed import Incomplete
 
 DOMAIN: str
+DOMAIN_COORDINATORS: str
 CONF_INDEX: str
 CONF_PROCESS: str
 NET_IO_TYPES: Incomplete

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

@@ -27,6 +27,7 @@ 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 +37,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 +55,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 +65,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

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