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

homeassistant-stubs/auth/__init__.pyi

@@ -3,6 +3,7 @@ from .const import ACCESS_TOKEN_EXPIRATION as ACCESS_TOKEN_EXPIRATION, GROUP_ID_
 from .mfa_modules import MultiFactorAuthModule as MultiFactorAuthModule, auth_mfa_module_from_config as auth_mfa_module_from_config
 from .models import AuthFlowResult as AuthFlowResult
 from .providers import AuthProvider as AuthProvider, LoginFlow as LoginFlow, auth_provider_from_config as auth_provider_from_config
+from .providers.homeassistant import HassAuthProvider as HassAuthProvider
 from _typeshed import Incomplete
 from datetime import datetime, timedelta
 from homeassistant import data_entry_flow as data_entry_flow
@@ -58,6 +59,7 @@ class AuthManager:
     async def async_link_user(self, user: models.User, credentials: models.Credentials) -> None: ...
     async def async_remove_user(self, user: models.User) -> None: ...
     async def async_update_user(self, user: models.User, name: str | None = None, is_active: bool | None = None, group_ids: list[str] | None = None, local_only: bool | None = None) -> None: ...
+    def async_update_user_credentials_data(self, credentials: models.Credentials, data: dict[str, Any]) -> None: ...
     async def async_activate_user(self, user: models.User) -> None: ...
     async def async_deactivate_user(self, user: models.User) -> None: ...
     async def async_remove_credentials(self, credentials: models.Credentials) -> None: ...

homeassistant-stubs/auth/auth_store.pyi

@@ -44,6 +44,7 @@ class AuthStore:
     def async_get_refresh_tokens(self) -> list[models.RefreshToken]: ...
     def async_log_refresh_token_usage(self, refresh_token: models.RefreshToken, remote_ip: str | None = None) -> None: ...
     def async_set_expiry(self, refresh_token: models.RefreshToken, *, enable_expiry: bool) -> None: ...
+    def async_update_user_credentials_data(self, credentials: models.Credentials, data: dict[str, Any]) -> None: ...
     async def async_load(self) -> None: ...
     def _build_token_id_to_user_id(self) -> None: ...
     def _async_schedule_save(self, delay: float = ...) -> None: ...

homeassistant-stubs/auth/providers/homeassistant.pyi

@@ -18,7 +18,11 @@ CONFIG_SCHEMA: Incomplete
 def async_get_provider(hass: HomeAssistant) -> HassAuthProvider: ...
 
 class InvalidAuth(HomeAssistantError): ...
-class InvalidUser(HomeAssistantError): ...
+
+class InvalidUser(HomeAssistantError):
+    def __init__(self, *args: object, translation_key: str | None = None, translation_placeholders: dict[str, str] | None = None) -> None: ...
+
+class InvalidUsername(InvalidUser): ...
 
 class Data:
     hass: Incomplete
@@ -26,8 +30,9 @@ class Data:
     _data: Incomplete
     is_legacy: bool
     def __init__(self, hass: HomeAssistant) -> None: ...
-    def normalize_username(self, username: str) -> str: ...
+    def normalize_username(self, username: str, *, force_normalize: bool = False) -> str: ...
     async def async_load(self) -> None: ...
+    def _async_check_for_not_normalized_usernames(self, data: dict[str, list[dict[str, str]]]) -> None: ...
     @property
     def users(self) -> list[dict[str, str]]: ...
     def validate_login(self, username: str, password: str) -> None: ...
@@ -35,6 +40,8 @@ class Data:
     def add_auth(self, username: str, password: str) -> None: ...
     def async_remove_auth(self, username: str) -> None: ...
     def change_password(self, username: str, new_password: str) -> None: ...
+    def _validate_new_username(self, new_username: str) -> None: ...
+    def change_username(self, username: str, new_username: str) -> None: ...
     async def async_save(self) -> None: ...
 
 class HassAuthProvider(AuthProvider):
@@ -48,6 +55,7 @@ class HassAuthProvider(AuthProvider):
     async def async_add_auth(self, username: str, password: str) -> None: ...
     async def async_remove_auth(self, username: str) -> None: ...
     async def async_change_password(self, username: str, new_password: str) -> None: ...
+    async def async_change_username(self, credential: Credentials, new_username: str) -> None: ...
     async def async_get_or_create_credentials(self, flow_result: Mapping[str, str]) -> Credentials: ...
     async def async_user_meta_for_credentials(self, credentials: Credentials) -> UserMeta: ...
     async def async_will_remove_credentials(self, credentials: Credentials) -> None: ...

homeassistant-stubs/block_async_io.pyi

@@ -1,6 +1,8 @@
 from .helpers.frame import get_current_frame as get_current_frame
 from .util.loop import protect_loop as protect_loop
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
 from typing import Any
 
 _IN_TESTS: Incomplete
@@ -9,4 +11,25 @@ ALLOWED_FILE_PREFIXES: Incomplete
 def _check_import_call_allowed(mapped_args: dict[str, Any]) -> bool: ...
 def _check_file_allowed(mapped_args: dict[str, Any]) -> bool: ...
 def _check_sleep_call_allowed(mapped_args: dict[str, Any]) -> bool: ...
+
+@dataclass(slots=True, frozen=True)
+class BlockingCall:
+    original_func: Callable
+    object: object
+    function: str
+    check_allowed: Callable[[dict[str, Any]], bool] | None
+    strict: bool
+    strict_core: bool
+    skip_for_tests: bool
+    def __init__(self, original_func, object, function, check_allowed, strict, strict_core, skip_for_tests) -> None: ...
+
+_BLOCKING_CALLS: tuple[BlockingCall, ...]
+
+@dataclass(slots=True)
+class BlockedCalls:
+    calls: set[BlockingCall]
+    def __init__(self, calls) -> None: ...
+
+_BLOCKED_CALLS: Incomplete
+
 def enable() -> None: ...

homeassistant-stubs/bootstrap.pyi

@@ -1,5 +1,4 @@
 import logging
-import logging.handlers
 from . import block_async_io as block_async_io, config_entries as config_entries, core as core, loader as loader, requirements as requirements
 from .components import http as http
 from .const import BASE_PLATFORMS as BASE_PLATFORMS, FORMAT_DATETIME as FORMAT_DATETIME, REQUIRED_NEXT_PYTHON_HA_RELEASE as REQUIRED_NEXT_PYTHON_HA_RELEASE, REQUIRED_NEXT_PYTHON_VER as REQUIRED_NEXT_PYTHON_VER, SIGNAL_BOOTSTRAP_INTEGRATIONS as SIGNAL_BOOTSTRAP_INTEGRATIONS
@@ -17,6 +16,7 @@ from .util.logging import async_activate_log_queue_handler as async_activate_log
 from .util.package import async_get_user_site as async_get_user_site, is_virtual_env as is_virtual_env
 from _typeshed import Incomplete
 from anyio._backends import _asyncio as _asyncio
+from logging.handlers import RotatingFileHandler, TimedRotatingFileHandler
 from typing import Any
 
 _LOGGER: Incomplete
@@ -48,9 +48,10 @@ def open_hass_ui(hass: core.HomeAssistant) -> None: ...
 def _init_blocking_io_modules_in_executor() -> None: ...
 async def async_load_base_functionality(hass: core.HomeAssistant) -> None: ...
 async def async_from_config_dict(config: ConfigType, hass: core.HomeAssistant) -> core.HomeAssistant | None: ...
-def async_enable_logging(hass: core.HomeAssistant, verbose: bool = False, log_rotate_days: int | None = None, log_file: str | None = None, log_no_color: bool = False) -> None: ...
+async def async_enable_logging(hass: core.HomeAssistant, verbose: bool = False, log_rotate_days: int | None = None, log_file: str | None = None, log_no_color: bool = False) -> None: ...
+def _create_log_file(err_log_path: str, log_rotate_days: int | None) -> RotatingFileHandler | TimedRotatingFileHandler: ...
 
-class _RotatingFileHandlerWithoutShouldRollOver(logging.handlers.RotatingFileHandler):
+class _RotatingFileHandlerWithoutShouldRollOver(RotatingFileHandler):
     def shouldRollover(self, record: logging.LogRecord) -> bool: ...
 
 async def async_mount_local_lib_path(config_dir: str) -> str: ...

homeassistant-stubs/components/accuweather/sensor.pyi

@@ -19,12 +19,7 @@ class AccuWeatherSensorDescription(SensorEntityDescription):
     attr_fn: Callable[[dict[str, Any]], 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, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn, attr_fn) -> None: ...
 
-@dataclass(frozen=True, kw_only=True)
-class AccuWeatherForecastSensorDescription(AccuWeatherSensorDescription):
-    day: 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, value_fn, attr_fn, day) -> None: ...
-
-FORECAST_SENSOR_TYPES: tuple[AccuWeatherForecastSensorDescription, ...]
+FORECAST_SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...]
 SENSOR_TYPES: tuple[AccuWeatherSensorDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, entry: AccuWeatherConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
@@ -48,12 +43,13 @@ class AccuWeatherSensor(CoordinatorEntity[AccuWeatherObservationDataUpdateCoordi
 class AccuWeatherForecastSensor(CoordinatorEntity[AccuWeatherDailyForecastDataUpdateCoordinator], SensorEntity):
     _attr_attribution = ATTRIBUTION
     _attr_has_entity_name: bool
-    entity_description: AccuWeatherForecastSensorDescription
-    forecast_day: Incomplete
+    entity_description: AccuWeatherSensorDescription
     _sensor_data: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, coordinator: AccuWeatherDailyForecastDataUpdateCoordinator, description: AccuWeatherForecastSensorDescription) -> None: ...
+    _attr_translation_placeholders: Incomplete
+    forecast_day: Incomplete
+    def __init__(self, coordinator: AccuWeatherDailyForecastDataUpdateCoordinator, description: AccuWeatherSensorDescription, forecast_day: int) -> None: ...
     @property
     def native_value(self) -> str | int | float | None: ...
     @property

homeassistant-stubs/components/accuweather/weather.pyi

@@ -2,7 +2,7 @@ from . import AccuWeatherConfigEntry as AccuWeatherConfigEntry, AccuWeatherData
 from .const import API_METRIC as API_METRIC, ATTRIBUTION as ATTRIBUTION, ATTR_DIRECTION as ATTR_DIRECTION, ATTR_SPEED as ATTR_SPEED, ATTR_VALUE as ATTR_VALUE, CONDITION_MAP as CONDITION_MAP
 from .coordinator import AccuWeatherDailyForecastDataUpdateCoordinator as AccuWeatherDailyForecastDataUpdateCoordinator, AccuWeatherObservationDataUpdateCoordinator as AccuWeatherObservationDataUpdateCoordinator
 from _typeshed import Incomplete
-from homeassistant.components.weather import ATTR_FORECAST_CLOUD_COVERAGE as ATTR_FORECAST_CLOUD_COVERAGE, ATTR_FORECAST_CONDITION as ATTR_FORECAST_CONDITION, ATTR_FORECAST_NATIVE_APPARENT_TEMP as ATTR_FORECAST_NATIVE_APPARENT_TEMP, ATTR_FORECAST_NATIVE_PRECIPITATION as ATTR_FORECAST_NATIVE_PRECIPITATION, ATTR_FORECAST_NATIVE_TEMP as ATTR_FORECAST_NATIVE_TEMP, ATTR_FORECAST_NATIVE_TEMP_LOW as ATTR_FORECAST_NATIVE_TEMP_LOW, ATTR_FORECAST_NATIVE_WIND_GUST_SPEED as ATTR_FORECAST_NATIVE_WIND_GUST_SPEED, ATTR_FORECAST_NATIVE_WIND_SPEED as ATTR_FORECAST_NATIVE_WIND_SPEED, ATTR_FORECAST_PRECIPITATION_PROBABILITY as ATTR_FORECAST_PRECIPITATION_PROBABILITY, ATTR_FORECAST_TIME as ATTR_FORECAST_TIME, ATTR_FORECAST_UV_INDEX as ATTR_FORECAST_UV_INDEX, ATTR_FORECAST_WIND_BEARING as ATTR_FORECAST_WIND_BEARING, CoordinatorWeatherEntity as CoordinatorWeatherEntity, Forecast as Forecast, WeatherEntityFeature as WeatherEntityFeature
+from homeassistant.components.weather import ATTR_FORECAST_CLOUD_COVERAGE as ATTR_FORECAST_CLOUD_COVERAGE, ATTR_FORECAST_CONDITION as ATTR_FORECAST_CONDITION, ATTR_FORECAST_HUMIDITY as ATTR_FORECAST_HUMIDITY, ATTR_FORECAST_NATIVE_APPARENT_TEMP as ATTR_FORECAST_NATIVE_APPARENT_TEMP, ATTR_FORECAST_NATIVE_PRECIPITATION as ATTR_FORECAST_NATIVE_PRECIPITATION, ATTR_FORECAST_NATIVE_TEMP as ATTR_FORECAST_NATIVE_TEMP, ATTR_FORECAST_NATIVE_TEMP_LOW as ATTR_FORECAST_NATIVE_TEMP_LOW, ATTR_FORECAST_NATIVE_WIND_GUST_SPEED as ATTR_FORECAST_NATIVE_WIND_GUST_SPEED, ATTR_FORECAST_NATIVE_WIND_SPEED as ATTR_FORECAST_NATIVE_WIND_SPEED, ATTR_FORECAST_PRECIPITATION_PROBABILITY as ATTR_FORECAST_PRECIPITATION_PROBABILITY, ATTR_FORECAST_TIME as ATTR_FORECAST_TIME, ATTR_FORECAST_UV_INDEX as ATTR_FORECAST_UV_INDEX, ATTR_FORECAST_WIND_BEARING as ATTR_FORECAST_WIND_BEARING, CoordinatorWeatherEntity as CoordinatorWeatherEntity, Forecast as Forecast, WeatherEntityFeature as WeatherEntityFeature
 from homeassistant.const import UnitOfLength as UnitOfLength, UnitOfPrecipitationDepth as UnitOfPrecipitationDepth, UnitOfPressure as UnitOfPressure, UnitOfSpeed as UnitOfSpeed, UnitOfTemperature as UnitOfTemperature
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback

homeassistant-stubs/components/acer_projector/switch.pyi

@@ -1,6 +1,6 @@
 from .const import CMD_DICT as CMD_DICT, CONF_WRITE_TIMEOUT as CONF_WRITE_TIMEOUT, DEFAULT_NAME as DEFAULT_NAME, DEFAULT_TIMEOUT as DEFAULT_TIMEOUT, DEFAULT_WRITE_TIMEOUT as DEFAULT_WRITE_TIMEOUT, ECO_MODE as ECO_MODE, ICON as ICON, INPUT_SOURCE as INPUT_SOURCE, LAMP as LAMP, LAMP_HOURS as LAMP_HOURS
 from _typeshed import Incomplete
-from homeassistant.components.switch import PLATFORM_SCHEMA as PLATFORM_SCHEMA, SwitchEntity as SwitchEntity
+from homeassistant.components.switch import SwitchEntity as SwitchEntity
 from homeassistant.const import CONF_FILENAME as CONF_FILENAME, CONF_NAME as CONF_NAME, CONF_TIMEOUT as CONF_TIMEOUT, STATE_OFF as STATE_OFF, STATE_ON as STATE_ON, STATE_UNKNOWN as STATE_UNKNOWN
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
@@ -8,6 +8,7 @@ from homeassistant.helpers.typing import ConfigType as ConfigType, DiscoveryInfo
 from typing import Any
 
 _LOGGER: Incomplete
+PLATFORM_SCHEMA: Incomplete
 
 def setup_platform(hass: HomeAssistant, config: ConfigType, add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None) -> None: ...
 

homeassistant-stubs/components/air_quality/__init__.pyi

@@ -3,13 +3,16 @@ from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.config_validation import PLATFORM_SCHEMA as PLATFORM_SCHEMA, PLATFORM_SCHEMA_BASE as PLATFORM_SCHEMA_BASE
 from homeassistant.helpers.entity import Entity as Entity
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
 from homeassistant.helpers.typing import ConfigType as ConfigType, StateType as StateType
 from typing import Final
 
 _LOGGER: Final[Incomplete]
+ENTITY_ID_FORMAT: Final[Incomplete]
+PLATFORM_SCHEMA: Incomplete
+PLATFORM_SCHEMA_BASE: Incomplete
+SCAN_INTERVAL: Final[Incomplete]
 ATTR_AQI: Final[str]
 ATTR_CO2: Final[str]
 ATTR_CO: Final[str]
@@ -21,8 +24,6 @@ ATTR_PM_0_1: Final[str]
 ATTR_PM_10: Final[str]
 ATTR_PM_2_5: Final[str]
 ATTR_SO2: Final[str]
-ENTITY_ID_FORMAT: Final[Incomplete]
-SCAN_INTERVAL: Final[Incomplete]
 PROP_TO_ATTR: Final[dict[str, str]]
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...

homeassistant-stubs/components/airgradient/__init__.pyi

@@ -1,5 +1,6 @@
 from .const import DOMAIN as DOMAIN
 from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator, AirGradientMeasurementCoordinator as AirGradientMeasurementCoordinator
+from dataclasses import dataclass
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_HOST as CONF_HOST, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
@@ -7,5 +8,12 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession as asyn
 
 PLATFORMS: list[Platform]
 
+@dataclass
+class AirGradientData:
+    measurement: AirGradientMeasurementCoordinator
+    config: AirGradientConfigCoordinator
+    def __init__(self, measurement, config) -> None: ...
+AirGradientConfigEntry = ConfigEntry[AirGradientData]
+
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/airgradient/button.pyi

@@ -0,0 +1,28 @@
+from . import AirGradientConfigEntry as AirGradientConfigEntry, DOMAIN as DOMAIN
+from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator
+from .entity import AirGradientEntity as AirGradientEntity
+from _typeshed import Incomplete
+from airgradient import AirGradientClient as AirGradientClient
+from collections.abc import Awaitable, Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
+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
+
+@dataclass(frozen=True, kw_only=True)
+class AirGradientButtonEntityDescription(ButtonEntityDescription):
+    press_fn: Callable[[AirGradientClient], Awaitable[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, press_fn) -> None: ...
+
+CO2_CALIBRATION: Incomplete
+LED_BAR_TEST: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: AirGradientConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AirGradientButton(AirGradientEntity, ButtonEntity):
+    entity_description: AirGradientButtonEntityDescription
+    coordinator: AirGradientConfigCoordinator
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: AirGradientConfigCoordinator, description: AirGradientButtonEntityDescription) -> None: ...
+    async def async_press(self) -> None: ...

homeassistant-stubs/components/airgradient/const.pyi

@@ -2,3 +2,5 @@ from _typeshed import Incomplete
 
 DOMAIN: str
 LOGGER: Incomplete
+PM_STANDARD: Incomplete
+PM_STANDARD_REVERSE: Incomplete

homeassistant-stubs/components/airgradient/coordinator.pyi

@@ -1,14 +1,14 @@
+from . import AirGradientConfigEntry as AirGradientConfigEntry
 from .const import LOGGER as LOGGER
 from _typeshed import Incomplete
 from airgradient import AirGradientClient as AirGradientClient, Config, Measures
 from datetime import timedelta
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
 
 class AirGradientCoordinator(DataUpdateCoordinator[_DataT]):
     _update_interval: timedelta
-    config_entry: ConfigEntry
+    config_entry: AirGradientConfigEntry
     client: Incomplete
     serial_number: Incomplete
     def __init__(self, hass: HomeAssistant, client: AirGradientClient) -> None: ...

homeassistant-stubs/components/airgradient/number.pyi

@@ -0,0 +1,32 @@
+from . import AirGradientConfigEntry as AirGradientConfigEntry
+from .const import DOMAIN as DOMAIN
+from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator
+from .entity import AirGradientEntity as AirGradientEntity
+from _typeshed import Incomplete
+from airgradient import AirGradientClient as AirGradientClient, Config as Config
+from collections.abc import Awaitable, Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+@dataclass(frozen=True, kw_only=True)
+class AirGradientNumberEntityDescription(NumberEntityDescription):
+    value_fn: Callable[[Config], int]
+    set_value_fn: Callable[[AirGradientClient, int], Awaitable[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, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, value_fn, set_value_fn) -> None: ...
+
+DISPLAY_BRIGHTNESS: Incomplete
+LED_BAR_BRIGHTNESS: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: AirGradientConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AirGradientNumber(AirGradientEntity, NumberEntity):
+    entity_description: AirGradientNumberEntityDescription
+    coordinator: AirGradientConfigCoordinator
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: AirGradientConfigCoordinator, description: AirGradientNumberEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> int | None: ...
+    async def async_set_native_value(self, value: float) -> None: ...

homeassistant-stubs/components/airgradient/select.pyi

@@ -1,28 +1,33 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator, AirGradientMeasurementCoordinator as AirGradientMeasurementCoordinator
+from . import AirGradientConfigEntry as AirGradientConfigEntry
+from .const import DOMAIN as DOMAIN, PM_STANDARD as PM_STANDARD, PM_STANDARD_REVERSE as PM_STANDARD_REVERSE
+from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator
 from .entity import AirGradientEntity as AirGradientEntity
 from _typeshed import Incomplete
 from airgradient import AirGradientClient as AirGradientClient, Config as Config
 from collections.abc import Awaitable, Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.select import SelectEntity as SelectEntity, SelectEntityDescription as SelectEntityDescription
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import EntityCategory as EntityCategory
-from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
 @dataclass(frozen=True, kw_only=True)
 class AirGradientSelectEntityDescription(SelectEntityDescription):
     value_fn: Callable[[Config], str | None]
     set_value_fn: Callable[[AirGradientClient, str], Awaitable[None]]
-    requires_display: 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, options, value_fn, set_value_fn, requires_display) -> 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, options, value_fn, set_value_fn) -> None: ...
 
 CONFIG_CONTROL_ENTITY: Incomplete
-PROTECTED_SELECT_TYPES: tuple[AirGradientSelectEntityDescription, ...]
+DISPLAY_SELECT_TYPES: tuple[AirGradientSelectEntityDescription, ...]
+LED_BAR_ENTITIES: tuple[AirGradientSelectEntityDescription, ...]
+LEARNING_TIME_OFFSET_OPTIONS: Incomplete
+ABC_DAYS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+def _get_value(value: int, values: list[str]) -> str | None: ...
+
+CONTROL_ENTITIES: tuple[AirGradientSelectEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: AirGradientConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AirGradientSelect(AirGradientEntity, SelectEntity):
     entity_description: AirGradientSelectEntityDescription
@@ -32,6 +37,3 @@ class AirGradientSelect(AirGradientEntity, SelectEntity):
     @property
     def current_option(self) -> str | None: ...
     async def async_select_option(self, option: str) -> None: ...
-
-class AirGradientProtectedSelect(AirGradientSelect):
-    async def async_select_option(self, option: str) -> None: ...

homeassistant-stubs/components/airgradient/sensor.pyi

@@ -1,30 +1,48 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import AirGradientMeasurementCoordinator as AirGradientMeasurementCoordinator
+from . import AirGradientConfigEntry as AirGradientConfigEntry
+from .const import PM_STANDARD as PM_STANDARD, PM_STANDARD_REVERSE as PM_STANDARD_REVERSE
+from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator, AirGradientMeasurementCoordinator as AirGradientMeasurementCoordinator
 from .entity import AirGradientEntity as AirGradientEntity
 from _typeshed import Incomplete
+from airgradient import Config as Config
 from airgradient.models import Measures as Measures
 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 CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfTemperature as UnitOfTemperature
+from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER as CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_MILLION as CONCENTRATION_PARTS_PER_MILLION, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
 
 @dataclass(frozen=True, kw_only=True)
-class AirGradientSensorEntityDescription(SensorEntityDescription):
+class AirGradientMeasurementSensorEntityDescription(SensorEntityDescription):
     value_fn: Callable[[Measures], StateType]
     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: ...
 
-SENSOR_TYPES: tuple[AirGradientSensorEntityDescription, ...]
+@dataclass(frozen=True, kw_only=True)
+class AirGradientConfigSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[Config], StateType]
+    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: ...
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+MEASUREMENT_SENSOR_TYPES: tuple[AirGradientMeasurementSensorEntityDescription, ...]
+CONFIG_SENSOR_TYPES: tuple[AirGradientConfigSensorEntityDescription, ...]
+CONFIG_LED_BAR_SENSOR_TYPES: tuple[AirGradientConfigSensorEntityDescription, ...]
+CONFIG_DISPLAY_SENSOR_TYPES: tuple[AirGradientConfigSensorEntityDescription, ...]
 
-class AirGradientSensor(AirGradientEntity, SensorEntity):
-    entity_description: AirGradientSensorEntityDescription
+async def async_setup_entry(hass: HomeAssistant, entry: AirGradientConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AirGradientMeasurementSensor(AirGradientEntity, SensorEntity):
+    entity_description: AirGradientMeasurementSensorEntityDescription
     coordinator: AirGradientMeasurementCoordinator
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: AirGradientMeasurementCoordinator, description: AirGradientSensorEntityDescription) -> None: ...
+    def __init__(self, coordinator: AirGradientMeasurementCoordinator, description: AirGradientMeasurementSensorEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> StateType: ...
+
+class AirGradientConfigSensor(AirGradientEntity, SensorEntity):
+    entity_description: AirGradientConfigSensorEntityDescription
+    coordinator: AirGradientConfigCoordinator
+    _attr_unique_id: Incomplete
+    _attr_entity_registry_enabled_default: Incomplete
+    def __init__(self, coordinator: AirGradientConfigCoordinator, description: AirGradientConfigSensorEntityDescription) -> None: ...
     @property
     def native_value(self) -> StateType: ...

homeassistant-stubs/components/airgradient/switch.pyi

@@ -0,0 +1,33 @@
+from . import AirGradientConfigEntry as AirGradientConfigEntry
+from .const import DOMAIN as DOMAIN
+from .coordinator import AirGradientConfigCoordinator as AirGradientConfigCoordinator
+from .entity import AirGradientEntity as AirGradientEntity
+from _typeshed import Incomplete
+from airgradient import AirGradientClient as AirGradientClient, Config as Config
+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.const import EntityCategory as EntityCategory
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Any
+
+@dataclass(frozen=True, kw_only=True)
+class AirGradientSwitchEntityDescription(SwitchEntityDescription):
+    value_fn: Callable[[Config], bool]
+    set_value_fn: Callable[[AirGradientClient, bool], Awaitable[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, value_fn, set_value_fn) -> None: ...
+
+POST_DATA_TO_AIRGRADIENT: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: AirGradientConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AirGradientSwitch(AirGradientEntity, SwitchEntity):
+    entity_description: AirGradientSwitchEntityDescription
+    coordinator: AirGradientConfigCoordinator
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: AirGradientConfigCoordinator, description: AirGradientSwitchEntityDescription) -> 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/airnow/const.pyi

@@ -12,6 +12,7 @@ ATTR_API_POLLUTANT: str
 ATTR_API_REPORT_DATE: str
 ATTR_API_REPORT_HOUR: str
 ATTR_API_REPORT_TZ: str
+ATTR_API_REPORT_TZINFO: str
 ATTR_API_STATE: str
 ATTR_API_STATION: str
 ATTR_API_STATION_LATITUDE: str

homeassistant-stubs/components/airnow/coordinator.pyi

@@ -1,4 +1,4 @@
-from .const import ATTR_API_AQI as ATTR_API_AQI, ATTR_API_AQI_DESCRIPTION as ATTR_API_AQI_DESCRIPTION, ATTR_API_AQI_LEVEL as ATTR_API_AQI_LEVEL, ATTR_API_AQI_PARAM as ATTR_API_AQI_PARAM, ATTR_API_CATEGORY as ATTR_API_CATEGORY, ATTR_API_CAT_DESCRIPTION as ATTR_API_CAT_DESCRIPTION, ATTR_API_CAT_LEVEL as ATTR_API_CAT_LEVEL, ATTR_API_PM25 as ATTR_API_PM25, ATTR_API_POLLUTANT as ATTR_API_POLLUTANT, ATTR_API_REPORT_DATE as ATTR_API_REPORT_DATE, ATTR_API_REPORT_HOUR as ATTR_API_REPORT_HOUR, ATTR_API_REPORT_TZ as ATTR_API_REPORT_TZ, ATTR_API_STATE as ATTR_API_STATE, ATTR_API_STATION as ATTR_API_STATION, ATTR_API_STATION_LATITUDE as ATTR_API_STATION_LATITUDE, ATTR_API_STATION_LONGITUDE as ATTR_API_STATION_LONGITUDE, DOMAIN as DOMAIN
+from .const import ATTR_API_AQI as ATTR_API_AQI, ATTR_API_AQI_DESCRIPTION as ATTR_API_AQI_DESCRIPTION, ATTR_API_AQI_LEVEL as ATTR_API_AQI_LEVEL, ATTR_API_AQI_PARAM as ATTR_API_AQI_PARAM, ATTR_API_CATEGORY as ATTR_API_CATEGORY, ATTR_API_CAT_DESCRIPTION as ATTR_API_CAT_DESCRIPTION, ATTR_API_CAT_LEVEL as ATTR_API_CAT_LEVEL, ATTR_API_PM25 as ATTR_API_PM25, ATTR_API_POLLUTANT as ATTR_API_POLLUTANT, ATTR_API_REPORT_DATE as ATTR_API_REPORT_DATE, ATTR_API_REPORT_HOUR as ATTR_API_REPORT_HOUR, ATTR_API_REPORT_TZ as ATTR_API_REPORT_TZ, ATTR_API_REPORT_TZINFO as ATTR_API_REPORT_TZINFO, ATTR_API_STATE as ATTR_API_STATE, ATTR_API_STATION as ATTR_API_STATION, ATTR_API_STATION_LATITUDE as ATTR_API_STATION_LATITUDE, ATTR_API_STATION_LONGITUDE as ATTR_API_STATION_LONGITUDE, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from aiohttp import ClientSession as ClientSession
 from datetime import timedelta

homeassistant-stubs/components/airnow/sensor.pyi

@@ -1,5 +1,5 @@
 from . import AirNowConfigEntry as AirNowConfigEntry, AirNowDataUpdateCoordinator as AirNowDataUpdateCoordinator
-from .const import ATTR_API_AQI as ATTR_API_AQI, ATTR_API_AQI_DESCRIPTION as ATTR_API_AQI_DESCRIPTION, ATTR_API_AQI_LEVEL as ATTR_API_AQI_LEVEL, ATTR_API_O3 as ATTR_API_O3, ATTR_API_PM10 as ATTR_API_PM10, ATTR_API_PM25 as ATTR_API_PM25, ATTR_API_REPORT_DATE as ATTR_API_REPORT_DATE, ATTR_API_REPORT_HOUR as ATTR_API_REPORT_HOUR, ATTR_API_REPORT_TZ as ATTR_API_REPORT_TZ, ATTR_API_STATION as ATTR_API_STATION, ATTR_API_STATION_LATITUDE as ATTR_API_STATION_LATITUDE, ATTR_API_STATION_LONGITUDE as ATTR_API_STATION_LONGITUDE, DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN
+from .const import ATTR_API_AQI as ATTR_API_AQI, ATTR_API_AQI_DESCRIPTION as ATTR_API_AQI_DESCRIPTION, ATTR_API_AQI_LEVEL as ATTR_API_AQI_LEVEL, ATTR_API_O3 as ATTR_API_O3, ATTR_API_PM10 as ATTR_API_PM10, ATTR_API_PM25 as ATTR_API_PM25, ATTR_API_REPORT_DATE as ATTR_API_REPORT_DATE, ATTR_API_REPORT_HOUR as ATTR_API_REPORT_HOUR, ATTR_API_REPORT_TZINFO as ATTR_API_REPORT_TZINFO, ATTR_API_STATION as ATTR_API_STATION, ATTR_API_STATION_LATITUDE as ATTR_API_STATION_LATITUDE, ATTR_API_STATION_LONGITUDE as ATTR_API_STATION_LONGITUDE, DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass
@@ -10,7 +10,6 @@ from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntry
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from homeassistant.util.dt import get_time_zone as get_time_zone
 from typing import Any
 
 ATTRIBUTION: str

homeassistant-stubs/components/airzone/__init__.pyi

@@ -1,4 +1,3 @@
-from .const import DOMAIN as DOMAIN
 from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -8,7 +7,8 @@ from homeassistant.helpers import aiohttp_client as aiohttp_client
 
 PLATFORMS: list[Platform]
 _LOGGER: Incomplete
+AirzoneConfigEntry = ConfigEntry[AirzoneUpdateCoordinator]
 
-async def _async_migrate_unique_ids(hass: HomeAssistant, entry: ConfigEntry, coordinator: AirzoneUpdateCoordinator) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def _async_migrate_unique_ids(hass: HomeAssistant, entry: AirzoneConfigEntry, coordinator: AirzoneUpdateCoordinator) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AirzoneConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: AirzoneConfigEntry) -> bool: ...

homeassistant-stubs/components/airzone/binary_sensor.pyi

@@ -1,4 +1,4 @@
-from .const import DOMAIN as DOMAIN
+from . import AirzoneConfigEntry as AirzoneConfigEntry
 from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
 from .entity import AirzoneEntity as AirzoneEntity, AirzoneSystemEntity as AirzoneSystemEntity, AirzoneZoneEntity as AirzoneZoneEntity
 from _typeshed import Incomplete
@@ -18,7 +18,7 @@ class AirzoneBinarySensorEntityDescription(BinarySensorEntityDescription):
 SYSTEM_BINARY_SENSOR_TYPES: Final[tuple[AirzoneBinarySensorEntityDescription, ...]]
 ZONE_BINARY_SENSOR_TYPES: Final[tuple[AirzoneBinarySensorEntityDescription, ...]]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AirzoneConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AirzoneBinarySensor(AirzoneEntity, BinarySensorEntity):
     entity_description: AirzoneBinarySensorEntityDescription

homeassistant-stubs/components/airzone/climate.pyi

@@ -1,4 +1,5 @@
-from .const import API_TEMPERATURE_STEP as API_TEMPERATURE_STEP, DOMAIN as DOMAIN, TEMP_UNIT_LIB_TO_HASS as TEMP_UNIT_LIB_TO_HASS
+from . import AirzoneConfigEntry as AirzoneConfigEntry
+from .const import API_TEMPERATURE_STEP as API_TEMPERATURE_STEP, TEMP_UNIT_LIB_TO_HASS as TEMP_UNIT_LIB_TO_HASS
 from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
 from .entity import AirzoneZoneEntity as AirzoneZoneEntity
 from _typeshed import Incomplete
@@ -17,7 +18,7 @@ HVAC_ACTION_LIB_TO_HASS: Final[dict[OperationAction, HVACAction]]
 HVAC_MODE_LIB_TO_HASS: Final[dict[OperationMode, HVACMode]]
 HVAC_MODE_HASS_TO_LIB: Final[dict[HVACMode, OperationMode]]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AirzoneConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AirzoneClimate(AirzoneZoneEntity, ClimateEntity):
     _attr_name: Incomplete

homeassistant-stubs/components/airzone/diagnostics.pyi

@@ -1,8 +1,6 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
+from . import AirzoneConfigEntry as AirzoneConfigEntry
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics.util import async_redact_data as async_redact_data
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_UNIQUE_ID as CONF_UNIQUE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
@@ -11,4 +9,4 @@ TO_REDACT_API: Incomplete
 TO_REDACT_CONFIG: Incomplete
 TO_REDACT_COORD: 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: AirzoneConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/airzone/entity.pyi

@@ -1,3 +1,4 @@
+from . import AirzoneConfigEntry as AirzoneConfigEntry
 from .const import DOMAIN as DOMAIN, MANUFACTURER as MANUFACTURER
 from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
 from _typeshed import Incomplete
@@ -17,7 +18,7 @@ class AirzoneSystemEntity(AirzoneEntity):
     system_id: Incomplete
     _attr_device_info: Incomplete
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: AirzoneUpdateCoordinator, entry: ConfigEntry, system_data: dict[str, Any]) -> None: ...
+    def __init__(self, coordinator: AirzoneUpdateCoordinator, entry: AirzoneConfigEntry, system_data: dict[str, Any]) -> None: ...
     @property
     def available(self) -> bool: ...
     def get_airzone_value(self, key: str) -> Any: ...

homeassistant-stubs/components/airzone/select.pyi

@@ -1,4 +1,4 @@
-from .const import DOMAIN as DOMAIN
+from . import AirzoneConfigEntry as AirzoneConfigEntry
 from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
 from .entity import AirzoneEntity as AirzoneEntity, AirzoneZoneEntity as AirzoneZoneEntity
 from _typeshed import Incomplete
@@ -20,7 +20,7 @@ GRILLE_ANGLE_DICT: Final[dict[str, int]]
 SLEEP_DICT: Final[dict[str, int]]
 ZONE_SELECT_TYPES: Final[tuple[AirzoneSelectDescription, ...]]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AirzoneConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AirzoneBaseSelect(AirzoneEntity, SelectEntity):
     entity_description: AirzoneSelectDescription

homeassistant-stubs/components/airzone/sensor.pyi

@@ -1,4 +1,5 @@
-from .const import DOMAIN as DOMAIN, TEMP_UNIT_LIB_TO_HASS as TEMP_UNIT_LIB_TO_HASS
+from . import AirzoneConfigEntry as AirzoneConfigEntry
+from .const import TEMP_UNIT_LIB_TO_HASS as TEMP_UNIT_LIB_TO_HASS
 from .coordinator import AirzoneUpdateCoordinator as AirzoneUpdateCoordinator
 from .entity import AirzoneEntity as AirzoneEntity, AirzoneHotWaterEntity as AirzoneHotWaterEntity, AirzoneWebServerEntity as AirzoneWebServerEntity, AirzoneZoneEntity as AirzoneZoneEntity
 from _typeshed import Incomplete
@@ -13,7 +14,7 @@ HOT_WATER_SENSOR_TYPES: Final[tuple[SensorEntityDescription, ...]]
 WEBSERVER_SENSOR_TYPES: Final[tuple[SensorEntityDescription, ...]]
 ZONE_SENSOR_TYPES: Final[tuple[SensorEntityDescription, ...]]
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: AirzoneConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class AirzoneSensor(AirzoneEntity, SensorEntity):
     def _handle_coordinator_update(self) -> None: ...