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

homeassistant-stubs/components/http/__init__.pyi

@@ -1,8 +1,8 @@
 import asyncio
 import ssl
-from .auth import async_setup_auth as async_setup_auth, async_sign_path as async_sign_path
+from .auth import async_setup_auth as async_setup_auth
 from .ban import setup_bans as setup_bans
-from .const import DOMAIN as DOMAIN, KEY_HASS_REFRESH_TOKEN_ID as KEY_HASS_REFRESH_TOKEN_ID, KEY_HASS_USER as KEY_HASS_USER, StrictConnectionMode as StrictConnectionMode
+from .const import DOMAIN as DOMAIN, KEY_HASS_REFRESH_TOKEN_ID as KEY_HASS_REFRESH_TOKEN_ID, KEY_HASS_USER as KEY_HASS_USER
 from .cors import setup_cors as setup_cors
 from .decorators import require_admin as require_admin
 from .forwarded import async_setup_forwarded as async_setup_forwarded
@@ -21,10 +21,11 @@ from aiohttp.web_exceptions import HTTPRedirection as HTTPRedirection
 from aiohttp.web_protocol import RequestHandler as RequestHandler
 from homeassistant.components.network import async_get_source_ip as async_get_source_ip
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, SERVER_PORT as SERVER_PORT
-from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, ServiceResponse as ServiceResponse, callback as callback
-from homeassistant.exceptions import HomeAssistantError as HomeAssistantError, ServiceValidationError as ServiceValidationError, Unauthorized as Unauthorized, UnknownUser as UnknownUser
+from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import storage as storage
 from homeassistant.helpers.http import HomeAssistantView as HomeAssistantView, KEY_ALLOW_CONFIGRED_CORS as KEY_ALLOW_CONFIGRED_CORS, KEY_AUTHENTICATED as KEY_AUTHENTICATED, KEY_HASS as KEY_HASS, current_request as current_request
+from homeassistant.helpers.importlib import async_import_module as async_import_module
 from homeassistant.helpers.network import NoURLAvailableError as NoURLAvailableError, get_url as get_url
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
@@ -47,7 +48,6 @@ CONF_TRUSTED_PROXIES: Final[str]
 CONF_LOGIN_ATTEMPTS_THRESHOLD: Final[str]
 CONF_IP_BAN_ENABLED: Final[str]
 CONF_SSL_PROFILE: Final[str]
-CONF_STRICT_CONNECTION: Final[str]
 SSL_MODERN: Final[str]
 SSL_INTERMEDIATE: Final[str]
 _LOGGER: Final[Incomplete]
@@ -110,7 +110,7 @@ class HomeAssistantHTTP:
     site: Incomplete
     context: Incomplete
     def __init__(self, hass: HomeAssistant, ssl_certificate: str | None, ssl_peer_certificate: str | None, ssl_key: str | None, server_host: list[str] | None, server_port: int, trusted_proxies: list[IPv4Network | IPv6Network], ssl_profile: str) -> None: ...
-    async def async_initialize(self, *, cors_origins: list[str], use_x_forwarded_for: bool, login_threshold: int, is_ban_enabled: bool, use_x_frame_options: bool, strict_connection_non_cloud: StrictConnectionMode) -> None: ...
+    async def async_initialize(self, *, cors_origins: list[str], use_x_forwarded_for: bool, login_threshold: int, is_ban_enabled: bool, use_x_frame_options: bool) -> None: ...
     def register_view(self, view: HomeAssistantView | type[HomeAssistantView]) -> None: ...
     def register_redirect(self, url: str, redirect_to: StrOrURL, *, redirect_exc: type[HTTPRedirection] = ...) -> None: ...
     def register_static_path(self, url_path: str, path: str, cache_headers: bool = True) -> None: ...
@@ -120,4 +120,3 @@ class HomeAssistantHTTP:
     async def stop(self) -> None: ...
 
 async def start_http_server_and_save_config(hass: HomeAssistant, conf: dict, server: HomeAssistantHTTP) -> None: ...
-def _setup_services(hass: HomeAssistant, conf: ConfData) -> None: ...

homeassistant-stubs/components/http/auth.pyi

@@ -1,5 +1,4 @@
-from .const import DOMAIN as DOMAIN, KEY_AUTHENTICATED as KEY_AUTHENTICATED, KEY_HASS_REFRESH_TOKEN_ID as KEY_HASS_REFRESH_TOKEN_ID, KEY_HASS_USER as KEY_HASS_USER, StrictConnectionMode as StrictConnectionMode
-from .session import HomeAssistantCookieStorage as HomeAssistantCookieStorage
+from .const import KEY_AUTHENTICATED as KEY_AUTHENTICATED, KEY_HASS_REFRESH_TOKEN_ID as KEY_HASS_REFRESH_TOKEN_ID, KEY_HASS_USER as KEY_HASS_USER
 from _typeshed import Incomplete
 from aiohttp.web import Application as Application, Request as Request, StreamResponse as StreamResponse
 from collections.abc import Callable as Callable
@@ -9,7 +8,6 @@ from homeassistant.auth.const import GROUP_ID_READ_ONLY as GROUP_ID_READ_ONLY
 from homeassistant.auth.models import User as User
 from homeassistant.components import websocket_api as websocket_api
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers import singleton as singleton
 from homeassistant.helpers.http import current_request as current_request
 from homeassistant.helpers.json import json_bytes as json_bytes
 from homeassistant.helpers.network import is_cloud_connection as is_cloud_connection
@@ -25,13 +23,7 @@ SAFE_QUERY_PARAMS: Final[Incomplete]
 STORAGE_VERSION: int
 STORAGE_KEY: str
 CONTENT_USER_NAME: str
-STRICT_CONNECTION_EXCLUDED_PATH: str
-STRICT_CONNECTION_GUARD_PAGE_NAME: str
-STRICT_CONNECTION_GUARD_PAGE: Incomplete
 
 def async_sign_path(hass: HomeAssistant, path: str, expiration: timedelta, *, refresh_token_id: str | None = None, use_content_user: bool = False) -> str: ...
 def async_user_not_allowed_do_auth(hass: HomeAssistant, user: User, request: Request | None = None) -> str | None: ...
-async def async_setup_auth(hass: HomeAssistant, app: Application, strict_connection_mode_non_cloud: StrictConnectionMode) -> None: ...
-async def _async_perform_strict_connection_action_on_non_local(hass: HomeAssistant, request: Request, guard_page: bool) -> StreamResponse | None: ...
-async def _async_perform_strict_connection_action(hass: HomeAssistant, request: Request, guard_page: bool) -> StreamResponse | None: ...
-async def _read_strict_connection_guard_page(hass: HomeAssistant) -> str: ...
+async def async_setup_auth(hass: HomeAssistant, app: Application) -> None: ...

homeassistant-stubs/components/http/ban.pyi

@@ -9,10 +9,8 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.util import yaml as yaml
 from ipaddress import IPv4Address, IPv6Address
-from typing import Any, Concatenate, Final, ParamSpec, TypeVar
+from typing import Any, Concatenate, Final
 
-_HassViewT = TypeVar('_HassViewT', bound=HomeAssistantView)
-_P = ParamSpec('_P')
 _LOGGER: Final[Incomplete]
 KEY_BAN_MANAGER: Incomplete
 KEY_FAILED_LOGIN_ATTEMPTS: Incomplete

homeassistant-stubs/components/http/const.pyi

@@ -1,12 +1,6 @@
-from enum import StrEnum
 from homeassistant.helpers.http import KEY_AUTHENTICATED as KEY_AUTHENTICATED, KEY_HASS as KEY_HASS
 from typing import Final
 
 DOMAIN: Final[str]
 KEY_HASS_USER: Final[str]
 KEY_HASS_REFRESH_TOKEN_ID: Final[str]
-
-class StrictConnectionMode(StrEnum):
-    DISABLED: str
-    GUARD_PAGE: str
-    DROP_CONNECTION: str

homeassistant-stubs/components/http/data_validator.pyi

@@ -2,11 +2,9 @@ import voluptuous as vol
 from .view import HomeAssistantView as HomeAssistantView
 from _typeshed import Incomplete
 from aiohttp import web as web
-from collections.abc import Awaitable, Callable as Callable, Coroutine
-from typing import Any, Concatenate, ParamSpec, TypeVar
+from collections.abc import Awaitable as Awaitable, Callable as Callable, Coroutine
+from typing import Any, Concatenate
 
-_HassViewT = TypeVar('_HassViewT', bound=HomeAssistantView)
-_P = ParamSpec('_P')
 _LOGGER: Incomplete
 
 class RequestDataValidator:

homeassistant-stubs/components/http/decorators.pyi

@@ -1,14 +1,11 @@
 from .view import HomeAssistantView as HomeAssistantView
 from _typeshed import Incomplete
-from aiohttp.web import Response, StreamResponse
 from collections.abc import Callable
 from homeassistant.auth.models import User as User
 from homeassistant.exceptions import Unauthorized as Unauthorized
-from typing import ParamSpec, TypeVar, overload
+from typing import overload
 
-_HomeAssistantViewT = TypeVar('_HomeAssistantViewT', bound=HomeAssistantView)
-_ResponseT = TypeVar('_ResponseT', bound=Response | StreamResponse)
-_P = ParamSpec('_P')
+_ResponseType: Incomplete
 _FuncType: Incomplete
 
 @overload

homeassistant-stubs/components/http/web_runner.pyi

@@ -8,7 +8,7 @@ class HomeAssistantTCPSite(web.BaseSite):
     _port: Incomplete
     _reuse_address: Incomplete
     _reuse_port: Incomplete
-    def __init__(self, runner: web.BaseRunner, host: None | str | list[str], port: int, *, ssl_context: SSLContext | None = None, backlog: int = 128, reuse_address: bool | None = None, reuse_port: bool | None = None) -> None: ...
+    def __init__(self, runner: web.BaseRunner, host: str | list[str] | None, port: int, *, ssl_context: SSLContext | None = None, backlog: int = 128, reuse_address: bool | None = None, reuse_port: bool | None = None) -> None: ...
     @property
     def name(self) -> str: ...
     _server: Incomplete

homeassistant-stubs/components/humidifier/intent.pyi

@@ -11,10 +11,14 @@ async def async_setup_intents(hass: HomeAssistant) -> None: ...
 
 class HumidityHandler(intent.IntentHandler):
     intent_type = INTENT_HUMIDITY
+    description: str
     slot_schema: Incomplete
+    platforms: Incomplete
     async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse: ...
 
 class SetModeHandler(intent.IntentHandler):
     intent_type = INTENT_MODE
+    description: str
     slot_schema: Incomplete
+    platforms: Incomplete
     async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse: ...

homeassistant-stubs/components/{poolsense → husqvarna_automower}/__init__.pyi

@@ -1,13 +1,15 @@
+from . import api as api
 from .const import DOMAIN as DOMAIN
-from .coordinator import PoolSenseDataUpdateCoordinator as PoolSenseDataUpdateCoordinator
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_EMAIL as CONF_EMAIL, CONF_PASSWORD as CONF_PASSWORD, Platform as Platform
+from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers import aiohttp_client as aiohttp_client
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers import aiohttp_client as aiohttp_client, config_entry_oauth2_flow as config_entry_oauth2_flow
 
-PLATFORMS: Incomplete
 _LOGGER: Incomplete
+PLATFORMS: list[Platform]
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...

homeassistant-stubs/components/husqvarna_automower/api.pyi

@@ -0,0 +1,11 @@
+from _typeshed import Incomplete
+from aioautomower.auth import AbstractAuth
+from aiohttp import ClientSession as ClientSession
+from homeassistant.helpers import config_entry_oauth2_flow as config_entry_oauth2_flow
+
+_LOGGER: Incomplete
+
+class AsyncConfigEntryAuth(AbstractAuth):
+    _oauth_session: Incomplete
+    def __init__(self, websession: ClientSession, oauth_session: config_entry_oauth2_flow.OAuth2Session) -> None: ...
+    async def async_get_access_token(self) -> str: ...

homeassistant-stubs/components/husqvarna_automower/application_credentials.pyi

@@ -0,0 +1,5 @@
+from .const import OAUTH2_AUTHORIZE as OAUTH2_AUTHORIZE, OAUTH2_TOKEN as OAUTH2_TOKEN
+from homeassistant.components.application_credentials import AuthorizationServer as AuthorizationServer
+from homeassistant.core import HomeAssistant as HomeAssistant
+
+async def async_get_authorization_server(hass: HomeAssistant) -> AuthorizationServer: ...

homeassistant-stubs/components/husqvarna_automower/binary_sensor.pyi

@@ -0,0 +1,29 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerBaseEntity as AutomowerBaseEntity
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes as MowerAttributes
+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
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+_LOGGER: Incomplete
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerBinarySensorEntityDescription(BinarySensorEntityDescription):
+    value_fn: Callable[[MowerAttributes], 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: ...
+
+BINARY_SENSOR_TYPES: tuple[AutomowerBinarySensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerBinarySensorEntity(AutomowerBaseEntity, BinarySensorEntity):
+    entity_description: AutomowerBinarySensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerBinarySensorEntityDescription) -> None: ...
+    @property
+    def is_on(self) -> bool: ...

homeassistant-stubs/components/husqvarna_automower/config_flow.pyi

@@ -0,0 +1,23 @@
+import logging
+from .const import DOMAIN as DOMAIN, NAME as NAME
+from _typeshed import Incomplete
+from collections.abc import Mapping
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlowResult as ConfigFlowResult
+from homeassistant.const import CONF_ACCESS_TOKEN as CONF_ACCESS_TOKEN, CONF_TOKEN as CONF_TOKEN
+from homeassistant.helpers import config_entry_oauth2_flow as config_entry_oauth2_flow
+from typing import Any
+
+_LOGGER: Incomplete
+CONF_USER_ID: str
+HUSQVARNA_DEV_PORTAL_URL: str
+
+class HusqvarnaConfigFlowHandler(config_entry_oauth2_flow.AbstractOAuth2FlowHandler, domain=DOMAIN):
+    VERSION: int
+    DOMAIN = DOMAIN
+    reauth_entry: ConfigEntry | None
+    async def async_oauth_create_entry(self, data: dict[str, Any]) -> ConfigFlowResult: ...
+    @property
+    def logger(self) -> logging.Logger: ...
+    async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
+    async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_missing_scope(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/husqvarna_automower/const.pyi

@@ -0,0 +1,5 @@
+DOMAIN: str
+EXECUTION_TIME_DELAY: int
+NAME: str
+OAUTH2_AUTHORIZE: str
+OAUTH2_TOKEN: str

homeassistant-stubs/components/husqvarna_automower/coordinator.pyi

@@ -0,0 +1,20 @@
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes
+from aioautomower.session import AutomowerSession as AutomowerSession
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+
+_LOGGER: Incomplete
+MAX_WS_RECONNECT_TIME: int
+SCAN_INTERVAL: Incomplete
+
+class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttributes]]):
+    api: Incomplete
+    ws_connected: bool
+    def __init__(self, hass: HomeAssistant, api: AutomowerSession, entry: ConfigEntry) -> None: ...
+    async def _async_update_data(self) -> dict[str, MowerAttributes]: ...
+    def callback(self, ws_data: dict[str, MowerAttributes]) -> None: ...
+    async def client_listen(self, hass: HomeAssistant, entry: ConfigEntry, automower_client: AutomowerSession, reconnect_time: int = 2) -> None: ...

homeassistant-stubs/components/husqvarna_automower/device_tracker.pyi

@@ -0,0 +1,21 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerBaseEntity as AutomowerBaseEntity
+from _typeshed import Incomplete
+from homeassistant.components.device_tracker import SourceType as SourceType, TrackerEntity as TrackerEntity
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerDeviceTrackerEntity(AutomowerBaseEntity, TrackerEntity):
+    _attr_name: Incomplete
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def source_type(self) -> SourceType: ...
+    @property
+    def latitude(self) -> float: ...
+    @property
+    def longitude(self) -> float: ...

homeassistant-stubs/components/husqvarna_automower/diagnostics.pyi

@@ -0,0 +1,17 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+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_ACCESS_TOKEN as CONF_ACCESS_TOKEN
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
+from typing import Any
+
+CONF_REFRESH_TOKEN: str
+POSITIONS: str
+TO_REDACT: Incomplete
+_LOGGER: Incomplete
+
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_device_diagnostics(hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/husqvarna_automower/entity.pyi

@@ -0,0 +1,20 @@
+from . import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes as MowerAttributes
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+
+_LOGGER: Incomplete
+
+class AutomowerBaseEntity(CoordinatorEntity[AutomowerDataUpdateCoordinator]):
+    _attr_has_entity_name: bool
+    mower_id: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def mower_attributes(self) -> MowerAttributes: ...
+
+class AutomowerControlEntity(AutomowerBaseEntity):
+    @property
+    def available(self) -> bool: ...

homeassistant-stubs/components/husqvarna_automower/lawn_mower.pyi

@@ -0,0 +1,28 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from homeassistant.components.lawn_mower import LawnMowerActivity as LawnMowerActivity, LawnMowerEntity as LawnMowerEntity, LawnMowerEntityFeature as LawnMowerEntityFeature
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+SUPPORT_STATE_SERVICES: Incomplete
+DOCKED_ACTIVITIES: Incomplete
+MOWING_ACTIVITIES: Incomplete
+PAUSED_STATES: Incomplete
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerLawnMowerEntity(AutomowerControlEntity, LawnMowerEntity):
+    _attr_name: Incomplete
+    _attr_supported_features = SUPPORT_STATE_SERVICES
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def activity(self) -> LawnMowerActivity: ...
+    async def async_start_mowing(self) -> None: ...
+    async def async_pause(self) -> None: ...
+    async def async_dock(self) -> None: ...

homeassistant-stubs/components/husqvarna_automower/number.pyi

@@ -0,0 +1,67 @@
+from .const import DOMAIN as DOMAIN, EXECUTION_TIME_DELAY as EXECUTION_TIME_DELAY
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes as MowerAttributes, WorkArea as WorkArea
+from aioautomower.session import AutomowerSession as AutomowerSession
+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.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, Platform as Platform
+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 typing import Any
+
+_LOGGER: Incomplete
+
+def _async_get_cutting_height(data: MowerAttributes) -> int: ...
+def _work_area_translation_key(work_area_id: int) -> str: ...
+async def async_set_work_area_cutting_height(coordinator: AutomowerDataUpdateCoordinator, mower_id: str, cheight: float, work_area_id: int) -> None: ...
+async def async_set_cutting_height(session: AutomowerSession, mower_id: str, cheight: float) -> None: ...
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerNumberEntityDescription(NumberEntityDescription):
+    exists_fn: Callable[[MowerAttributes], bool] = ...
+    value_fn: Callable[[MowerAttributes], int]
+    set_value_fn: Callable[[AutomowerSession, str, float], Awaitable[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, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, exists_fn, value_fn, set_value_fn) -> None: ...
+
+NUMBER_TYPES: tuple[AutomowerNumberEntityDescription, ...]
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerWorkAreaNumberEntityDescription(NumberEntityDescription):
+    value_fn: Callable[[WorkArea], int]
+    translation_key_fn: Callable[[int], str]
+    set_value_fn: Callable[[AutomowerDataUpdateCoordinator, str, float, int], Awaitable[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, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, value_fn, translation_key_fn, set_value_fn) -> None: ...
+
+WORK_AREA_NUMBER_TYPES: tuple[AutomowerWorkAreaNumberEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerNumberEntity(AutomowerControlEntity, NumberEntity):
+    entity_description: AutomowerNumberEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerNumberEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> float: ...
+    async def async_set_native_value(self, value: float) -> None: ...
+
+class AutomowerWorkAreaNumberEntity(AutomowerControlEntity, NumberEntity):
+    entity_description: AutomowerWorkAreaNumberEntityDescription
+    coordinator: Incomplete
+    work_area_id: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_translation_placeholders: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerWorkAreaNumberEntityDescription, work_area_id: int) -> None: ...
+    @property
+    def work_area(self) -> WorkArea: ...
+    @property
+    def translation_key(self) -> str: ...
+    @property
+    def native_value(self) -> float: ...
+    async def async_set_native_value(self, value: float) -> None: ...
+
+def async_remove_entities(hass: HomeAssistant, coordinator: AutomowerDataUpdateCoordinator, config_entry: ConfigEntry, mower_id: str) -> None: ...

homeassistant-stubs/components/husqvarna_automower/select.pyi

@@ -0,0 +1,25 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from homeassistant.components.select import SelectEntity as SelectEntity
+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 HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+_LOGGER: Incomplete
+HEADLIGHT_MODES: list
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerSelectEntity(AutomowerControlEntity, SelectEntity):
+    _attr_options = HEADLIGHT_MODES
+    _attr_entity_category: Incomplete
+    _attr_translation_key: str
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def current_option(self) -> str: ...
+    async def async_select_option(self, option: str) -> None: ...

homeassistant-stubs/components/husqvarna_automower/sensor.pyi

@@ -0,0 +1,35 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerBaseEntity as AutomowerBaseEntity
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes as MowerAttributes
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from datetime import datetime
+from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfLength as UnitOfLength, UnitOfTime as UnitOfTime
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import StateType as StateType
+
+_LOGGER: Incomplete
+ERROR_KEY_LIST: Incomplete
+RESTRICTED_REASONS: list
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerSensorEntityDescription(SensorEntityDescription):
+    exists_fn: Callable[[MowerAttributes], bool] = ...
+    value_fn: Callable[[MowerAttributes], StateType | datetime]
+    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, exists_fn, value_fn) -> None: ...
+
+SENSOR_TYPES: tuple[AutomowerSensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerSensorEntity(AutomowerBaseEntity, SensorEntity):
+    entity_description: AutomowerSensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerSensorEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> StateType | datetime: ...

homeassistant-stubs/components/husqvarna_automower/switch.pyi

@@ -0,0 +1,49 @@
+from .const import DOMAIN as DOMAIN, EXECUTION_TIME_DELAY as EXECUTION_TIME_DELAY
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from aioautomower.model import StayOutZones as StayOutZones, Zone as Zone
+from homeassistant.components.switch import SwitchEntity as SwitchEntity
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import Platform as Platform
+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 typing import Any
+
+_LOGGER: Incomplete
+ERROR_ACTIVITIES: Incomplete
+ERROR_STATES: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerScheduleSwitchEntity(AutomowerControlEntity, SwitchEntity):
+    _attr_translation_key: str
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def is_on(self) -> bool: ...
+    @property
+    def available(self) -> bool: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+
+class AutomowerStayOutZoneSwitchEntity(AutomowerControlEntity, SwitchEntity):
+    _attr_translation_key: str
+    coordinator: Incomplete
+    stay_out_zone_uid: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_translation_placeholders: Incomplete
+    def __init__(self, coordinator: AutomowerDataUpdateCoordinator, mower_id: str, stay_out_zone_uid: str) -> None: ...
+    @property
+    def stay_out_zones(self) -> StayOutZones: ...
+    @property
+    def stay_out_zone(self) -> Zone: ...
+    @property
+    def is_on(self) -> bool: ...
+    @property
+    def available(self) -> bool: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+
+def async_remove_entities(hass: HomeAssistant, coordinator: AutomowerDataUpdateCoordinator, config_entry: ConfigEntry, mower_id: str) -> None: ...

homeassistant-stubs/components/hydrawise/binary_sensor.pyi

@@ -2,18 +2,25 @@ from .const import DOMAIN as DOMAIN
 from .coordinator import HydrawiseDataUpdateCoordinator as HydrawiseDataUpdateCoordinator
 from .entity import HydrawiseEntity as HydrawiseEntity
 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.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from pydrawise.schema import Zone as Zone
 
-BINARY_SENSOR_STATUS: Incomplete
-BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...]
-BINARY_SENSOR_KEYS: list[str]
+@dataclass(frozen=True, kw_only=True)
+class HydrawiseBinarySensorEntityDescription(BinarySensorEntityDescription):
+    value_fn: Callable[[HydrawiseBinarySensor], bool | 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) -> None: ...
+
+CONTROLLER_BINARY_SENSORS: tuple[HydrawiseBinarySensorEntityDescription, ...]
+RAIN_SENSOR_BINARY_SENSOR: tuple[HydrawiseBinarySensorEntityDescription, ...]
+ZONE_BINARY_SENSORS: tuple[HydrawiseBinarySensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class HydrawiseBinarySensor(HydrawiseEntity, BinarySensorEntity):
+    entity_description: HydrawiseBinarySensorEntityDescription
     _attr_is_on: Incomplete
     def _update_attrs(self) -> None: ...

homeassistant-stubs/components/hydrawise/coordinator.pyi

@@ -3,17 +3,20 @@ from dataclasses import dataclass
 from datetime import timedelta
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from pydrawise import HydrawiseBase as HydrawiseBase
-from pydrawise.schema import Controller as Controller, User as User, Zone as Zone
+from homeassistant.util.dt import now as now
+from pydrawise import Hydrawise as Hydrawise
+from pydrawise.schema import Controller as Controller, ControllerWaterUseSummary, Sensor as Sensor, User as User, Zone as Zone
 
 @dataclass
 class HydrawiseData:
     user: User
     controllers: dict[int, Controller]
     zones: dict[int, Zone]
-    def __init__(self, user, controllers, zones) -> None: ...
+    sensors: dict[int, Sensor]
+    daily_water_use: dict[int, ControllerWaterUseSummary]
+    def __init__(self, user, controllers, zones, sensors, daily_water_use) -> None: ...
 
 class HydrawiseDataUpdateCoordinator(DataUpdateCoordinator[HydrawiseData]):
-    api: HydrawiseBase
-    def __init__(self, hass: HomeAssistant, api: HydrawiseBase, scan_interval: timedelta) -> None: ...
+    api: Hydrawise
+    def __init__(self, hass: HomeAssistant, api: Hydrawise, scan_interval: timedelta) -> None: ...
     async def _async_update_data(self) -> HydrawiseData: ...