homeassistant-stubs 2024.5.5__py3-none-any.whl → 2024.6.0b0__py3-none-any.whl

homeassistant-stubs/components/{poolsense → apsystems}/config_flow.pyi

@@ -1,13 +1,12 @@
 from .const import DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
-from homeassistant.const import CONF_EMAIL as CONF_EMAIL, CONF_PASSWORD as CONF_PASSWORD
-from homeassistant.helpers import aiohttp_client as aiohttp_client
+from homeassistant.const import CONF_IP_ADDRESS as CONF_IP_ADDRESS
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from typing import Any
 
-_LOGGER: Incomplete
+DATA_SCHEMA: Incomplete
 
-class PoolSenseConfigFlow(ConfigFlow, domain=DOMAIN):
+class APsystemsLocalAPIFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
-    def __init__(self) -> None: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/apsystems/const.pyi

@@ -0,0 +1,4 @@
+from logging import Logger
+
+LOGGER: Logger
+DOMAIN: str

homeassistant-stubs/components/apsystems/coordinator.pyi

@@ -0,0 +1,10 @@
+from .const import LOGGER as LOGGER
+from APsystemsEZ1 import APsystemsEZ1M as APsystemsEZ1M, ReturnOutputData
+from _typeshed import Incomplete
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+
+class ApSystemsDataCoordinator(DataUpdateCoordinator[ReturnOutputData]):
+    api: Incomplete
+    def __init__(self, hass: HomeAssistant, api: APsystemsEZ1M) -> None: ...
+    async def _async_update_data(self) -> ReturnOutputData: ...

homeassistant-stubs/components/apsystems/sensor.pyi

@@ -0,0 +1,31 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import ApSystemsDataCoordinator as ApSystemsDataCoordinator
+from APsystemsEZ1 import ReturnOutputData as ReturnOutputData
+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, StateType as StateType
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import UnitOfEnergy as UnitOfEnergy, UnitOfPower as UnitOfPower
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+
+@dataclass(frozen=True, kw_only=True)
+class ApsystemsLocalApiSensorDescription(SensorEntityDescription):
+    value_fn: Callable[[ReturnOutputData], float | 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, value_fn) -> None: ...
+
+SENSORS: tuple[ApsystemsLocalApiSensorDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
+
+class ApSystemsSensorWithDescription(CoordinatorEntity[ApSystemsDataCoordinator], SensorEntity):
+    entity_description: ApsystemsLocalApiSensorDescription
+    _attr_unique_id: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, coordinator: ApSystemsDataCoordinator, entity_description: ApsystemsLocalApiSensorDescription, device_id: str) -> None: ...
+    @property
+    def native_value(self) -> StateType: ...

homeassistant-stubs/components/aquostv/media_player.pyi

@@ -6,10 +6,8 @@ from homeassistant.const import CONF_HOST as CONF_HOST, CONF_NAME as CONF_NAME,
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfoType as DiscoveryInfoType
-from typing import Any, Concatenate, ParamSpec, TypeVar
+from typing import Any, Concatenate
 
-_SharpAquosTVDeviceT = TypeVar('_SharpAquosTVDeviceT', bound='SharpAquosTVDevice')
-_P = ParamSpec('_P')
 _LOGGER: Incomplete
 DEFAULT_NAME: str
 DEFAULT_PORT: int

homeassistant-stubs/components/aranet/const.pyi

@@ -1 +1,2 @@
 DOMAIN: str
+ARANET_MANUFACTURER_NAME: str

homeassistant-stubs/components/aranet/sensor.pyi

@@ -1,30 +1,32 @@
-from .const import DOMAIN as DOMAIN
+from .const import ARANET_MANUFACTURER_NAME as ARANET_MANUFACTURER_NAME, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from aranet4.client import Aranet4Advertisement as Aranet4Advertisement
+from aranet4.client import Aranet4Advertisement
 from bleak.backends.device import BLEDevice as BLEDevice
 from dataclasses import dataclass
 from homeassistant import config_entries as config_entries
 from homeassistant.components.bluetooth.passive_update_processor import PassiveBluetoothDataProcessor as PassiveBluetoothDataProcessor, PassiveBluetoothDataUpdate as PassiveBluetoothDataUpdate, PassiveBluetoothEntityKey as PassiveBluetoothEntityKey, PassiveBluetoothProcessorCoordinator as PassiveBluetoothProcessorCoordinator, PassiveBluetoothProcessorEntity as PassiveBluetoothProcessorEntity
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.const import ATTR_NAME as ATTR_NAME, ATTR_SW_VERSION as ATTR_SW_VERSION, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfPressure as UnitOfPressure, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
+from homeassistant.const import ATTR_MANUFACTURER as ATTR_MANUFACTURER, ATTR_NAME as ATTR_NAME, ATTR_SW_VERSION as ATTR_SW_VERSION, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfPressure as UnitOfPressure, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity import EntityDescription as EntityDescription
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Any
 
 @dataclass(frozen=True)
 class AranetSensorEntityDescription(SensorEntityDescription):
     name: str | 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) -> None: ...
+    scale: float | int = ...
+    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, scale) -> None: ...
 
 SENSOR_DESCRIPTIONS: Incomplete
 
 def _device_key_to_bluetooth_entity_key(device: BLEDevice, key: str) -> PassiveBluetoothEntityKey: ...
 def _sensor_device_info_to_hass(adv: Aranet4Advertisement) -> DeviceInfo: ...
-def sensor_update_to_bluetooth_data_update(adv: Aranet4Advertisement) -> PassiveBluetoothDataUpdate: ...
+def sensor_update_to_bluetooth_data_update(adv: Aranet4Advertisement) -> PassiveBluetoothDataUpdate[Any]: ...
 async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class Aranet4BluetoothSensorEntity(PassiveBluetoothProcessorEntity[PassiveBluetoothDataProcessor[float | int | None]], SensorEntity):
+class Aranet4BluetoothSensorEntity(PassiveBluetoothProcessorEntity[PassiveBluetoothDataProcessor[float | int | None, Aranet4Advertisement]], SensorEntity):
     @property
     def available(self) -> bool: ...
     @property

homeassistant-stubs/components/arcam_fmj/media_player.pyi

@@ -12,10 +12,8 @@ from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from typing import Any, ParamSpec, TypeVar
+from typing import Any
 
-_R = TypeVar('_R')
-_P = ParamSpec('_P')
 _LOGGER: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...

homeassistant-stubs/components/asuswrt/__init__.pyi

@@ -1,4 +1,3 @@
-from .const import DATA_ASUSWRT as DATA_ASUSWRT, DOMAIN as DOMAIN
 from .router import AsusWrtRouter as AsusWrtRouter
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -6,7 +5,8 @@ from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant
 
 PLATFORMS: Incomplete
+AsusWrtConfigEntry = ConfigEntry[AsusWrtRouter]
 
-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_setup_entry(hass: HomeAssistant, entry: AsusWrtConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: AsusWrtConfigEntry) -> bool: ...
+async def update_listener(hass: HomeAssistant, entry: AsusWrtConfigEntry) -> None: ...

homeassistant-stubs/components/asuswrt/bridge.pyi

@@ -11,7 +11,7 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession as asyn
 from homeassistant.helpers.device_registry import format_mac as format_mac
 from homeassistant.helpers.update_coordinator import UpdateFailed as UpdateFailed
 from pyasuswrt import AsusWrtHttp
-from typing import Any, NamedTuple, TypeVar
+from typing import Any, NamedTuple
 
 SENSORS_TYPE_BYTES: str
 SENSORS_TYPE_COUNT: str
@@ -25,11 +25,10 @@ class WrtDevice(NamedTuple):
     connected_to: Incomplete
 
 _LOGGER: Incomplete
-_AsusWrtBridgeT = TypeVar('_AsusWrtBridgeT', bound='AsusWrtBridge')
 _FuncType: Incomplete
-_ReturnFuncType = Callable[[_AsusWrtBridgeT], Coroutine[Any, Any, dict[str, Any]]]
+_ReturnFuncType = Callable[[_T], Coroutine[Any, Any, dict[str, Any]]]
 
-def handle_errors_and_zip(exceptions: type[Exception] | tuple[type[Exception], ...], keys: list[str] | None) -> Callable[[_FuncType], _ReturnFuncType]: ...
+def handle_errors_and_zip(exceptions: type[Exception] | tuple[type[Exception], ...], keys: list[str] | None) -> Callable[[_FuncType[_AsusWrtBridgeT]], _ReturnFuncType[_AsusWrtBridgeT]]: ...
 
 class AsusWrtBridge(ABC, metaclass=abc.ABCMeta):
     @staticmethod

homeassistant-stubs/components/asuswrt/const.pyi

@@ -6,7 +6,6 @@ CONF_INTERFACE: str
 CONF_REQUIRE_IP: str
 CONF_SSH_KEY: str
 CONF_TRACK_UNKNOWN: str
-DATA_ASUSWRT = DOMAIN
 DEFAULT_DNSMASQ: str
 DEFAULT_INTERFACE: str
 DEFAULT_TRACK_UNKNOWN: bool

homeassistant-stubs/components/asuswrt/device_tracker.pyi

@@ -1,8 +1,7 @@
-from .const import DATA_ASUSWRT as DATA_ASUSWRT, DOMAIN as DOMAIN
+from . import AsusWrtConfigEntry as AsusWrtConfigEntry
 from .router import AsusWrtDevInfo as AsusWrtDevInfo, AsusWrtRouter as AsusWrtRouter
 from _typeshed import Incomplete
 from homeassistant.components.device_tracker import ScannerEntity as ScannerEntity, SourceType as SourceType
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -10,7 +9,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEnti
 ATTR_LAST_TIME_REACHABLE: str
 DEFAULT_DEVICE_NAME: str
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AsusWrtConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 def add_entities(router: AsusWrtRouter, async_add_entities: AddEntitiesCallback, tracked: set[str]) -> None: ...
 
 class AsusWrtDevice(ScannerEntity):

homeassistant-stubs/components/asuswrt/diagnostics.pyi

@@ -1,8 +1,6 @@
-from .const import DATA_ASUSWRT as DATA_ASUSWRT, DOMAIN as DOMAIN
-from .router import AsusWrtRouter as AsusWrtRouter
+from . import AsusWrtConfigEntry as AsusWrtConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_CONNECTIONS as ATTR_CONNECTIONS, ATTR_IDENTIFIERS as ATTR_IDENTIFIERS, CONF_PASSWORD as CONF_PASSWORD, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
@@ -10,4 +8,4 @@ from typing import Any
 TO_REDACT: Incomplete
 TO_REDACT_DEV: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, dict[str, Any]]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: AsusWrtConfigEntry) -> dict[str, dict[str, Any]]: ...

homeassistant-stubs/components/asuswrt/router.pyi

@@ -12,6 +12,7 @@ from homeassistant.helpers.dispatcher import async_dispatcher_send as async_disp
 from homeassistant.helpers.event import async_track_time_interval as async_track_time_interval
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
 from homeassistant.util import slugify as slugify
+from types import MappingProxyType
 from typing import Any
 
 CONF_REQ_RELOAD: Incomplete
@@ -67,7 +68,7 @@ class AsusWrtRouter:
     async def _update_unpolled_sensors(self) -> None: ...
     async def close(self) -> None: ...
     def async_on_close(self, func: CALLBACK_TYPE) -> None: ...
-    def update_options(self, new_options: dict[str, Any]) -> bool: ...
+    def update_options(self, new_options: MappingProxyType[str, Any]) -> bool: ...
     @property
     def device_info(self) -> DeviceInfo: ...
     @property

homeassistant-stubs/components/asuswrt/sensor.pyi

@@ -1,9 +1,9 @@
-from .const import DATA_ASUSWRT as DATA_ASUSWRT, DOMAIN as DOMAIN, KEY_COORDINATOR as KEY_COORDINATOR, KEY_SENSORS as KEY_SENSORS, SENSORS_BYTES as SENSORS_BYTES, SENSORS_CONNECTED_DEVICE as SENSORS_CONNECTED_DEVICE, SENSORS_LOAD_AVG as SENSORS_LOAD_AVG, SENSORS_RATES as SENSORS_RATES, SENSORS_TEMPERATURES as SENSORS_TEMPERATURES
+from . import AsusWrtConfigEntry as AsusWrtConfigEntry
+from .const import KEY_COORDINATOR as KEY_COORDINATOR, KEY_SENSORS as KEY_SENSORS, SENSORS_BYTES as SENSORS_BYTES, SENSORS_CONNECTED_DEVICE as SENSORS_CONNECTED_DEVICE, SENSORS_LOAD_AVG as SENSORS_LOAD_AVG, SENSORS_RATES as SENSORS_RATES, SENSORS_TEMPERATURES as SENSORS_TEMPERATURES
 from .router import AsusWrtRouter as AsusWrtRouter
 from _typeshed import Incomplete
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory, UnitOfDataRate as UnitOfDataRate, UnitOfInformation as UnitOfInformation, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -18,7 +18,7 @@ class AsusWrtSensorEntityDescription(SensorEntityDescription):
 UNIT_DEVICES: str
 CONNECTION_SENSORS: tuple[AsusWrtSensorEntityDescription, ...]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AsusWrtConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AsusWrtSensor(CoordinatorEntity, SensorEntity):
     entity_description: AsusWrtSensorEntityDescription

homeassistant-stubs/components/auth/__init__.pyi

@@ -18,10 +18,10 @@ from multidict import MultiDictProxy
 from typing import Any
 
 DOMAIN: str
-STRICT_CONNECTION_URL: str
 StoreResultType = Callable[[str, Credentials], str]
 RetrieveResultType: Incomplete
 CONFIG_SCHEMA: Incomplete
+DELETE_CURRENT_TOKEN_DELAY: int
 
 def create_auth_code(hass: HomeAssistant, client_id: str, credential: Credentials) -> str: ...
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
@@ -51,12 +51,6 @@ class LinkUserView(HomeAssistantView):
     def __init__(self, retrieve_credentials: RetrieveResultType) -> None: ...
     async def post(self, request: web.Request, data: dict[str, Any]) -> web.Response: ...
 
-class StrictConnectionTempTokenView(HomeAssistantView):
-    url = STRICT_CONNECTION_URL
-    name: str
-    requires_auth: bool
-    async def get(self, request: web.Request) -> web.Response: ...
-
 def _create_auth_code_store() -> tuple[StoreResultType, RetrieveResultType]: ...
 async def websocket_current_user(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def websocket_create_long_lived_access_token(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
@@ -64,3 +58,4 @@ def websocket_refresh_tokens(hass: HomeAssistant, connection: websocket_api.Acti
 def websocket_delete_refresh_token(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def websocket_delete_all_refresh_tokens(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def websocket_sign_path(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+def websocket_refresh_token_set_expiry(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/axis/__init__.pyi

@@ -8,7 +8,8 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
 
 _LOGGER: Incomplete
+AxisConfigEntry = ConfigEntry[AxisHub]
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: AxisConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
 async def async_migrate_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/axis/binary_sensor.pyi

@@ -1,3 +1,4 @@
+from . import AxisConfigEntry as AxisConfigEntry
 from .entity import AxisEventDescription as AxisEventDescription, AxisEventEntity as AxisEventEntity
 from .hub import AxisHub as AxisHub
 from _typeshed import Incomplete
@@ -9,7 +10,6 @@ from axis.models.event import Event as Event
 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.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.event import async_call_later as async_call_later
@@ -30,7 +30,7 @@ def object_analytics_name_fn(hub: AxisHub, event: Event) -> str: ...
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: AxisConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AxisBinarySensor(AxisEventEntity, BinarySensorEntity):
     entity_description: AxisBinarySensorDescription

homeassistant-stubs/components/axis/camera.pyi

@@ -1,16 +1,16 @@
+from . import AxisConfigEntry as AxisConfigEntry
 from .const import DEFAULT_STREAM_PROFILE as DEFAULT_STREAM_PROFILE, DEFAULT_VIDEO_SOURCE as DEFAULT_VIDEO_SOURCE
 from .entity import AxisEntity as AxisEntity
 from .hub import AxisHub as AxisHub
 from _typeshed import Incomplete
 from homeassistant.components.camera import CameraEntityFeature as CameraEntityFeature
 from homeassistant.components.mjpeg import MjpegCamera as MjpegCamera, filter_urllib3_logging as filter_urllib3_logging
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import HTTP_DIGEST_AUTHENTICATION as HTTP_DIGEST_AUTHENTICATION
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: AxisConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AxisCamera(AxisEntity, MjpegCamera):
     _attr_supported_features: Incomplete

homeassistant-stubs/components/axis/config_flow.pyi

@@ -1,3 +1,4 @@
+from . import AxisConfigEntry as AxisConfigEntry
 from .const import CONF_STREAM_PROFILE as CONF_STREAM_PROFILE, CONF_VIDEO_SOURCE as CONF_VIDEO_SOURCE, DEFAULT_STREAM_PROFILE as DEFAULT_STREAM_PROFILE, DEFAULT_VIDEO_SOURCE as DEFAULT_VIDEO_SOURCE, DOMAIN as AXIS_DOMAIN
 from .errors import AuthenticationRequired as AuthenticationRequired, CannotConnect as CannotConnect
 from .hub import AxisHub as AxisHub, get_axis_api as get_axis_api
@@ -34,6 +35,7 @@ class AxisFlowHandler(ConfigFlow, domain=AXIS_DOMAIN):
     async def _process_discovered_device(self, discovery_info: dict[str, Any]) -> ConfigFlowResult: ...
 
 class AxisOptionsFlowHandler(OptionsFlowWithConfigEntry):
+    config_entry: AxisConfigEntry
     hub: AxisHub
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_configure_stream(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/axis/diagnostics.pyi

@@ -1,7 +1,6 @@
-from .hub import AxisHub as AxisHub
+from . import AxisConfigEntry as AxisConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_MAC as CONF_MAC, CONF_PASSWORD as CONF_PASSWORD, CONF_UNIQUE_ID as CONF_UNIQUE_ID, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
@@ -10,4 +9,4 @@ REDACT_CONFIG: Incomplete
 REDACT_BASIC_DEVICE_INFO: Incomplete
 REDACT_VAPIX_PARAMS: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: AxisConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/axis/hub/hub.pyi

@@ -1,10 +1,10 @@
 import axis
+from .. import AxisConfigEntry as AxisConfigEntry
 from ..const import ATTR_MANUFACTURER as ATTR_MANUFACTURER
 from .config import AxisConfig as AxisConfig
 from .entity_loader import AxisEntityLoader as AxisEntityLoader
 from .event_source import AxisEventSource as AxisEventSource
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC as CONNECTION_NETWORK_MAC, format_mac as format_mac
 from homeassistant.helpers.dispatcher import async_dispatcher_send as async_dispatcher_send
@@ -19,9 +19,7 @@ class AxisHub:
     product_type: Incomplete
     unique_id: Incomplete
     additional_diagnostics: Incomplete
-    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry, api: axis.AxisDevice) -> None: ...
-    @staticmethod
-    def get_hub(hass: HomeAssistant, config_entry: ConfigEntry) -> AxisHub: ...
+    def __init__(self, hass: HomeAssistant, config_entry: AxisConfigEntry, api: axis.AxisDevice) -> None: ...
     @property
     def available(self) -> bool: ...
     @property
@@ -29,7 +27,7 @@ class AxisHub:
     @property
     def signal_new_address(self) -> str: ...
     @staticmethod
-    async def async_new_address_callback(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
+    async def async_new_address_callback(hass: HomeAssistant, config_entry: AxisConfigEntry) -> None: ...
     async def async_update_device_registry(self) -> None: ...
     def setup(self) -> None: ...
     async def shutdown(self, event: Event) -> None: ...

homeassistant-stubs/components/axis/light.pyi

@@ -1,10 +1,10 @@
+from . import AxisConfigEntry as AxisConfigEntry
 from .entity import AxisEventDescription as AxisEventDescription, AxisEventEntity as AxisEventEntity, TOPIC_TO_EVENT_TYPE as TOPIC_TO_EVENT_TYPE
 from .hub import AxisHub as AxisHub
 from _typeshed import Incomplete
 from axis.models.event import Event as Event
 from dataclasses import dataclass
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ColorMode as ColorMode, LightEntity as LightEntity, LightEntityDescription as LightEntityDescription
-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 typing import Any
@@ -17,7 +17,7 @@ class AxisLightDescription(AxisEventDescription, LightEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: AxisConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AxisLight(AxisEventEntity, LightEntity):
     entity_description: AxisLightDescription

homeassistant-stubs/components/axis/switch.pyi

@@ -1,10 +1,10 @@
+from . import AxisConfigEntry as AxisConfigEntry
 from .entity import AxisEventDescription as AxisEventDescription, AxisEventEntity as AxisEventEntity
 from .hub import AxisHub as AxisHub
 from _typeshed import Incomplete
 from axis.models.event import Event as Event
 from dataclasses import dataclass
 from homeassistant.components.switch import SwitchDeviceClass as SwitchDeviceClass, 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, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -16,7 +16,7 @@ class AxisSwitchDescription(AxisEventDescription, SwitchEntityDescription):
 
 ENTITY_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: AxisConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AxisSwitch(AxisEventEntity, SwitchEntity):
     entity_description: AxisSwitchDescription

homeassistant-stubs/components/baf/__init__.pyi

@@ -1,11 +1,12 @@
-from .const import DOMAIN as DOMAIN, QUERY_INTERVAL as QUERY_INTERVAL, RUN_TIMEOUT as RUN_TIMEOUT
-from .models import BAFData as BAFData
+from .const import QUERY_INTERVAL as QUERY_INTERVAL, RUN_TIMEOUT as RUN_TIMEOUT
+from aiobafi6 import Device
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_IP_ADDRESS as CONF_IP_ADDRESS, Platform as Platform
-from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 
+BAFConfigEntry = ConfigEntry[Device]
 PLATFORMS: list[Platform]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: BAFConfigEntry) -> bool: ...

homeassistant-stubs/components/baf/binary_sensor.pyi

@@ -1,11 +1,9 @@
-from .const import DOMAIN as DOMAIN
+from . import BAFConfigEntry as BAFConfigEntry
 from .entity import BAFEntity as BAFEntity
-from .models import BAFData as BAFData
 from _typeshed import Incomplete
 from aiobafi6 import Device as Device
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
-from homeassistant import config_entries as config_entries
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -17,7 +15,7 @@ class BAFBinarySensorDescription(BinarySensorEntityDescription):
 
 OCCUPANCY_SENSORS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFBinarySensor(BAFEntity, BinarySensorEntity):
     entity_description: BAFBinarySensorDescription

homeassistant-stubs/components/baf/climate.pyi

@@ -1,15 +1,13 @@
-from .const import DOMAIN as DOMAIN
+from . import BAFConfigEntry as BAFConfigEntry
 from .entity import BAFEntity as BAFEntity
-from .models import BAFData as BAFData
 from _typeshed import Incomplete
-from homeassistant import config_entries as config_entries
 from homeassistant.components.climate import ClimateEntity as ClimateEntity, ClimateEntityFeature as ClimateEntityFeature, HVACAction as HVACAction, HVACMode as HVACMode
 from homeassistant.const import ATTR_TEMPERATURE as ATTR_TEMPERATURE, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
 
-async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFAutoComfort(BAFEntity, ClimateEntity):
     _attr_supported_features: Incomplete

homeassistant-stubs/components/baf/fan.pyi

@@ -1,15 +1,14 @@
-from .const import DOMAIN as DOMAIN, PRESET_MODE_AUTO as PRESET_MODE_AUTO, SPEED_COUNT as SPEED_COUNT, SPEED_RANGE as SPEED_RANGE
+from . import BAFConfigEntry as BAFConfigEntry
+from .const import PRESET_MODE_AUTO as PRESET_MODE_AUTO, SPEED_COUNT as SPEED_COUNT, SPEED_RANGE as SPEED_RANGE
 from .entity import BAFEntity as BAFEntity
-from .models import BAFData as BAFData
 from _typeshed import Incomplete
-from homeassistant import config_entries as config_entries
 from homeassistant.components.fan import DIRECTION_FORWARD as DIRECTION_FORWARD, DIRECTION_REVERSE as DIRECTION_REVERSE, FanEntity as FanEntity, FanEntityFeature as FanEntityFeature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.util.percentage import percentage_to_ranged_value as percentage_to_ranged_value, ranged_value_to_percentage as ranged_value_to_percentage
 from typing import Any
 
-async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFFan(BAFEntity, FanEntity):
     _attr_supported_features: Incomplete

homeassistant-stubs/components/baf/light.pyi

@@ -1,16 +1,14 @@
-from .const import DOMAIN as DOMAIN
+from . import BAFConfigEntry as BAFConfigEntry
 from .entity import BAFEntity as BAFEntity
-from .models import BAFData as BAFData
 from _typeshed import Incomplete
 from aiobafi6 import Device as Device
-from homeassistant import config_entries as config_entries
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP as ATTR_COLOR_TEMP, ColorMode as ColorMode, LightEntity as LightEntity
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.util.color import color_temperature_kelvin_to_mired as color_temperature_kelvin_to_mired, color_temperature_mired_to_kelvin as color_temperature_mired_to_kelvin
 from typing import Any
 
-async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFLight(BAFEntity, LightEntity):
     _attr_name: Incomplete

homeassistant-stubs/components/baf/models.pyi

@@ -1,13 +1,5 @@
-import asyncio
-from aiobafi6 import Device as Device
 from dataclasses import dataclass
 
-@dataclass
-class BAFData:
-    device: Device
-    run_future: asyncio.Future
-    def __init__(self, device, run_future) -> None: ...
-
 @dataclass
 class BAFDiscovery:
     ip_address: str

homeassistant-stubs/components/baf/number.pyi

@@ -1,11 +1,10 @@
-from .const import DOMAIN as DOMAIN, HALF_DAY_SECS as HALF_DAY_SECS, ONE_DAY_SECS as ONE_DAY_SECS, ONE_MIN_SECS as ONE_MIN_SECS, SPEED_RANGE as SPEED_RANGE
+from . import BAFConfigEntry as BAFConfigEntry
+from .const import HALF_DAY_SECS as HALF_DAY_SECS, ONE_DAY_SECS as ONE_DAY_SECS, ONE_MIN_SECS as ONE_MIN_SECS, SPEED_RANGE as SPEED_RANGE
 from .entity import BAFEntity as BAFEntity
-from .models import BAFData as BAFData
 from _typeshed import Incomplete
 from aiobafi6 import Device as Device
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
-from homeassistant import config_entries as config_entries
 from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription, NumberMode as NumberMode
 from homeassistant.const import EntityCategory as EntityCategory, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
@@ -20,7 +19,7 @@ AUTO_COMFORT_NUMBER_DESCRIPTIONS: Incomplete
 FAN_NUMBER_DESCRIPTIONS: Incomplete
 LIGHT_NUMBER_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFNumber(BAFEntity, NumberEntity):
     entity_description: BAFNumberDescription

homeassistant-stubs/components/baf/sensor.pyi

@@ -1,11 +1,9 @@
-from .const import DOMAIN as DOMAIN
+from . import BAFConfigEntry as BAFConfigEntry
 from .entity import BAFEntity as BAFEntity
-from .models import BAFData as BAFData
 from _typeshed import Incomplete
 from aiobafi6 import Device as Device
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
-from homeassistant import config_entries as config_entries
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
 from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, REVOLUTIONS_PER_MINUTE as REVOLUTIONS_PER_MINUTE, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
@@ -20,7 +18,7 @@ AUTO_COMFORT_SENSORS: Incomplete
 DEFINED_ONLY_SENSORS: Incomplete
 FAN_SENSORS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: BAFConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class BAFSensor(BAFEntity, SensorEntity):
     entity_description: BAFSensorDescription