homeassistant-stubs 2025.3.4__py3-none-any.whl → 2025.4.0b1__py3-none-any.whl

homeassistant-stubs/components/openai_conversation/const.pyi

@@ -1,17 +1,28 @@
-from _typeshed import Incomplete
+import logging
 
 DOMAIN: str
-LOGGER: Incomplete
-CONF_RECOMMENDED: str
-CONF_PROMPT: str
+LOGGER: logging.Logger
 CONF_CHAT_MODEL: str
-RECOMMENDED_CHAT_MODEL: str
+CONF_FILENAMES: str
 CONF_MAX_TOKENS: str
-RECOMMENDED_MAX_TOKENS: int
-CONF_TOP_P: str
-RECOMMENDED_TOP_P: float
-CONF_TEMPERATURE: str
-RECOMMENDED_TEMPERATURE: float
+CONF_PROMPT: str
 CONF_REASONING_EFFORT: str
+CONF_RECOMMENDED: str
+CONF_TEMPERATURE: str
+CONF_TOP_P: str
+CONF_WEB_SEARCH: str
+CONF_WEB_SEARCH_USER_LOCATION: str
+CONF_WEB_SEARCH_CONTEXT_SIZE: str
+CONF_WEB_SEARCH_CITY: str
+CONF_WEB_SEARCH_REGION: str
+CONF_WEB_SEARCH_COUNTRY: str
+CONF_WEB_SEARCH_TIMEZONE: str
+RECOMMENDED_CHAT_MODEL: str
+RECOMMENDED_MAX_TOKENS: int
 RECOMMENDED_REASONING_EFFORT: str
-UNSUPPORTED_MODELS: Incomplete
+RECOMMENDED_TEMPERATURE: float
+RECOMMENDED_TOP_P: float
+RECOMMENDED_WEB_SEARCH: bool
+RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE: str
+RECOMMENDED_WEB_SEARCH_USER_LOCATION: bool
+UNSUPPORTED_MODELS: list[str]

homeassistant-stubs/components/openai_conversation/conversation.pyi

@@ -1,5 +1,5 @@
 from . import OpenAIConfigEntry as OpenAIConfigEntry
-from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, DOMAIN as DOMAIN, LOGGER as LOGGER, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P
+from .const import CONF_CHAT_MODEL as CONF_CHAT_MODEL, CONF_MAX_TOKENS as CONF_MAX_TOKENS, CONF_PROMPT as CONF_PROMPT, CONF_REASONING_EFFORT as CONF_REASONING_EFFORT, CONF_TEMPERATURE as CONF_TEMPERATURE, CONF_TOP_P as CONF_TOP_P, CONF_WEB_SEARCH as CONF_WEB_SEARCH, CONF_WEB_SEARCH_CITY as CONF_WEB_SEARCH_CITY, CONF_WEB_SEARCH_CONTEXT_SIZE as CONF_WEB_SEARCH_CONTEXT_SIZE, CONF_WEB_SEARCH_COUNTRY as CONF_WEB_SEARCH_COUNTRY, CONF_WEB_SEARCH_REGION as CONF_WEB_SEARCH_REGION, CONF_WEB_SEARCH_TIMEZONE as CONF_WEB_SEARCH_TIMEZONE, CONF_WEB_SEARCH_USER_LOCATION as CONF_WEB_SEARCH_USER_LOCATION, DOMAIN as DOMAIN, LOGGER as LOGGER, RECOMMENDED_CHAT_MODEL as RECOMMENDED_CHAT_MODEL, RECOMMENDED_MAX_TOKENS as RECOMMENDED_MAX_TOKENS, RECOMMENDED_REASONING_EFFORT as RECOMMENDED_REASONING_EFFORT, RECOMMENDED_TEMPERATURE as RECOMMENDED_TEMPERATURE, RECOMMENDED_TOP_P as RECOMMENDED_TOP_P, RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE as RECOMMENDED_WEB_SEARCH_CONTEXT_SIZE
 from _typeshed import Incomplete
 from collections.abc import AsyncGenerator, Callable as Callable
 from homeassistant.components import assist_pipeline as assist_pipeline, conversation as conversation
@@ -7,18 +7,18 @@ from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_LLM_HASS_API as CONF_LLM_HASS_API, MATCH_ALL as MATCH_ALL
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
-from homeassistant.helpers import chat_session as chat_session, intent as intent, llm as llm
+from homeassistant.helpers import intent as intent, llm as llm
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
 from openai._streaming import AsyncStream as AsyncStream
-from openai.types.chat import ChatCompletionChunk as ChatCompletionChunk, ChatCompletionMessageParam, ChatCompletionToolParam
+from openai.types.responses import FunctionToolParam, ResponseInputParam as ResponseInputParam, ResponseStreamEvent as ResponseStreamEvent, ToolParam as ToolParam
 from typing import Any, Literal
 
 MAX_TOOL_ITERATIONS: int
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: OpenAIConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
-def _format_tool(tool: llm.Tool, custom_serializer: Callable[[Any], Any] | None) -> ChatCompletionToolParam: ...
-def _convert_content_to_param(content: conversation.Content) -> ChatCompletionMessageParam: ...
-async def _transform_stream(result: AsyncStream[ChatCompletionChunk]) -> AsyncGenerator[conversation.AssistantContentDeltaDict]: ...
+def _format_tool(tool: llm.Tool, custom_serializer: Callable[[Any], Any] | None) -> FunctionToolParam: ...
+def _convert_content_to_param(content: conversation.Content) -> ResponseInputParam: ...
+async def _transform_stream(chat_log: conversation.ChatLog, result: AsyncStream[ResponseStreamEvent]) -> AsyncGenerator[conversation.AssistantContentDeltaDict]: ...
 
 class OpenAIConversationEntity(conversation.ConversationEntity, conversation.AbstractConversationAgent):
     _attr_has_entity_name: bool
@@ -32,6 +32,5 @@ class OpenAIConversationEntity(conversation.ConversationEntity, conversation.Abs
     def supported_languages(self) -> list[str] | Literal['*']: ...
     async def async_added_to_hass(self) -> None: ...
     async def async_will_remove_from_hass(self) -> None: ...
-    async def async_process(self, user_input: conversation.ConversationInput) -> conversation.ConversationResult: ...
     async def _async_handle_message(self, user_input: conversation.ConversationInput, chat_log: conversation.ChatLog) -> conversation.ConversationResult: ...
     async def _async_entry_update_listener(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...

homeassistant-stubs/components/overkiz/water_heater/__init__.pyi

@@ -1,6 +1,7 @@
 from .. import OverkizDataConfigEntry as OverkizDataConfigEntry
 from ..entity import OverkizEntity as OverkizEntity
 from .atlantic_domestic_hot_water_production_mlb_component import AtlanticDomesticHotWaterProductionMBLComponent as AtlanticDomesticHotWaterProductionMBLComponent
+from .atlantic_domestic_hot_water_production_v2_io_component import AtlanticDomesticHotWaterProductionV2IOComponent as AtlanticDomesticHotWaterProductionV2IOComponent
 from .atlantic_pass_apc_dhw import AtlanticPassAPCDHW as AtlanticPassAPCDHW
 from .domestic_hot_water_production import DomesticHotWaterProduction as DomesticHotWaterProduction
 from .hitachi_dhw import HitachiDHW as HitachiDHW

homeassistant-stubs/components/overkiz/water_heater/atlantic_domestic_hot_water_production_v2_io_component.pyi

@@ -0,0 +1,41 @@
+from ..entity import OverkizEntity as OverkizEntity
+from _typeshed import Incomplete
+from homeassistant.components.water_heater import STATE_ECO as STATE_ECO, STATE_ELECTRIC as STATE_ELECTRIC, STATE_HEAT_PUMP as STATE_HEAT_PUMP, STATE_PERFORMANCE as STATE_PERFORMANCE, WaterHeaterEntity as WaterHeaterEntity, WaterHeaterEntityFeature as WaterHeaterEntityFeature
+from homeassistant.const import ATTR_TEMPERATURE as ATTR_TEMPERATURE, UnitOfTemperature as UnitOfTemperature
+from typing import Any
+
+DEFAULT_MIN_TEMP: float
+DEFAULT_MAX_TEMP: float
+MAX_BOOST_MODE_DURATION: int
+DHWP_AWAY_MODES: Incomplete
+
+class AtlanticDomesticHotWaterProductionV2IOComponent(OverkizEntity, WaterHeaterEntity):
+    _attr_temperature_unit: Incomplete
+    _attr_supported_features: Incomplete
+    _attr_operation_list: Incomplete
+    @property
+    def min_temp(self) -> float: ...
+    @property
+    def max_temp(self) -> float: ...
+    @property
+    def current_temperature(self) -> float: ...
+    @property
+    def target_temperature(self) -> float: ...
+    async def async_set_temperature(self, **kwargs: Any) -> None: ...
+    @property
+    def is_state_eco(self) -> bool: ...
+    @property
+    def is_state_performance(self) -> bool: ...
+    @property
+    def is_state_heat_pump(self) -> bool: ...
+    @property
+    def is_away_mode_on(self) -> bool: ...
+    @property
+    def current_operation(self) -> str | None: ...
+    @property
+    def is_boost_mode_on(self) -> bool: ...
+    async def async_set_operation_mode(self, operation_mode: str) -> None: ...
+    async def async_turn_away_mode_on(self, refresh_afterwards: bool = True) -> None: ...
+    async def async_turn_away_mode_off(self, refresh_afterwards: bool = True) -> None: ...
+    async def async_turn_boost_mode_on(self, refresh_afterwards: bool = True) -> None: ...
+    async def async_turn_boost_mode_off(self, refresh_afterwards: bool = True) -> None: ...

homeassistant-stubs/components/purpleair/__init__.pyi

@@ -1,12 +1,7 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator as PurpleAirDataUpdateCoordinator
-from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import Platform as Platform
+from .const import PLATFORMS as PLATFORMS
+from .coordinator import PurpleAirConfigEntry as PurpleAirConfigEntry, PurpleAirDataUpdateCoordinator as PurpleAirDataUpdateCoordinator
 from homeassistant.core import HomeAssistant as HomeAssistant
 
-PLATFORMS: Incomplete
-
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_handle_entry_update(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> bool: ...
+async def async_reload_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> None: ...
+async def async_unload_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> bool: ...

homeassistant-stubs/components/purpleair/const.pyi

@@ -1,6 +1,8 @@
 from _typeshed import Incomplete
+from homeassistant.const import Platform as Platform
+from typing import Final
 
-DOMAIN: str
-LOGGER: Incomplete
-CONF_READ_KEY: str
-CONF_SENSOR_INDICES: str
+LOGGER: Final[Incomplete]
+PLATFORMS: Final[Incomplete]
+DOMAIN: Final[str]
+CONF_SENSOR_INDICES: Final[str]

homeassistant-stubs/components/purpleair/coordinator.pyi

@@ -10,11 +10,12 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as Da
 
 SENSOR_FIELDS_TO_RETRIEVE: Incomplete
 UPDATE_INTERVAL: Incomplete
+type PurpleAirConfigEntry = ConfigEntry[PurpleAirDataUpdateCoordinator]
 
 class PurpleAirDataUpdateCoordinator(DataUpdateCoordinator[GetSensorsResponse]):
-    config_entry: ConfigEntry
+    config_entry: PurpleAirConfigEntry
     _api: Incomplete
-    def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+    def __init__(self, hass: HomeAssistant, entry: PurpleAirConfigEntry) -> None: ...
     async def _async_update_data(self) -> GetSensorsResponse: ...
     @callback
     def async_get_map_url(self, sensor_index: int) -> str: ...

homeassistant-stubs/components/purpleair/diagnostics.pyi

@@ -1,8 +1,6 @@
-from .const import DOMAIN as DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator as PurpleAirDataUpdateCoordinator
+from .coordinator import PurpleAirConfigEntry as PurpleAirConfigEntry
 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_API_KEY as CONF_API_KEY, CONF_LATITUDE as CONF_LATITUDE, CONF_LONGITUDE as CONF_LONGITUDE, CONF_UNIQUE_ID as CONF_UNIQUE_ID
 from homeassistant.core import HomeAssistant as HomeAssistant
 from typing import Any
@@ -10,4 +8,4 @@ from typing import Any
 CONF_TITLE: str
 TO_REDACT: Incomplete
 
-async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry) -> dict[str, Any]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: PurpleAirConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/purpleair/entity.pyi

@@ -1,9 +1,8 @@
 from .const import DOMAIN as DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator as PurpleAirDataUpdateCoordinator
+from .coordinator import PurpleAirConfigEntry as PurpleAirConfigEntry, PurpleAirDataUpdateCoordinator as PurpleAirDataUpdateCoordinator
 from _typeshed import Incomplete
 from aiopurpleair.models.sensors import SensorModel as SensorModel
 from collections.abc import Mapping
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_LATITUDE as ATTR_LATITUDE, ATTR_LONGITUDE as ATTR_LONGITUDE, CONF_SHOW_ON_MAP as CONF_SHOW_ON_MAP
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
@@ -14,7 +13,7 @@ class PurpleAirEntity(CoordinatorEntity[PurpleAirDataUpdateCoordinator]):
     _sensor_index: Incomplete
     _attr_device_info: Incomplete
     _entry: Incomplete
-    def __init__(self, coordinator: PurpleAirDataUpdateCoordinator, entry: ConfigEntry, sensor_index: int) -> None: ...
+    def __init__(self, entry: PurpleAirConfigEntry, sensor_index: int) -> None: ...
     @property
     def extra_state_attributes(self) -> Mapping[str, Any]: ...
     @property

homeassistant-stubs/components/purpleair/sensor.pyi

@@ -1,12 +1,11 @@
-from .const import CONF_SENSOR_INDICES as CONF_SENSOR_INDICES, DOMAIN as DOMAIN
-from .coordinator import PurpleAirDataUpdateCoordinator as PurpleAirDataUpdateCoordinator
+from .const import CONF_SENSOR_INDICES as CONF_SENSOR_INDICES
+from .coordinator import PurpleAirConfigEntry as PurpleAirConfigEntry
 from .entity import PurpleAirEntity as PurpleAirEntity
 from _typeshed import Incomplete
 from aiopurpleair.models.sensors import SensorModel as SensorModel
 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, EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfPressure as UnitOfPressure, UnitOfTemperature as UnitOfTemperature, UnitOfTime as UnitOfTime, UnitOfVolume as UnitOfVolume
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
@@ -19,11 +18,11 @@ class PurpleAirSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: PurpleAirConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
 
 class PurpleAirSensorEntity(PurpleAirEntity, SensorEntity):
     entity_description: PurpleAirSensorEntityDescription
     _attr_unique_id: Incomplete
-    def __init__(self, coordinator: PurpleAirDataUpdateCoordinator, entry: ConfigEntry, sensor_index: int, description: PurpleAirSensorEntityDescription) -> None: ...
+    def __init__(self, entry: PurpleAirConfigEntry, sensor_index: int, description: PurpleAirSensorEntityDescription) -> None: ...
     @property
     def native_value(self) -> float | str | None: ...

homeassistant-stubs/components/pyload/__init__.pyi

@@ -0,0 +1,12 @@
+from .coordinator import PyLoadConfigEntry as PyLoadConfigEntry, PyLoadCoordinator as PyLoadCoordinator
+from _typeshed import Incomplete
+from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_SSL as CONF_SSL, CONF_URL as CONF_URL, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL, Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession
+
+_LOGGER: Incomplete
+PLATFORMS: list[Platform]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool: ...
+async def async_migrate_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool: ...

homeassistant-stubs/components/pyload/button.pyi

@@ -0,0 +1,32 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import PyLoadConfigEntry as PyLoadConfigEntry
+from .entity import BasePyLoadEntity as BasePyLoadEntity
+from collections.abc import Awaitable, Callable as Callable
+from dataclasses import dataclass
+from enum import StrEnum
+from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from pyloadapi import PyLoadAPI as PyLoadAPI
+from typing import Any
+
+PARALLEL_UPDATES: int
+
+@dataclass(kw_only=True, frozen=True)
+class PyLoadButtonEntityDescription(ButtonEntityDescription):
+    press_fn: Callable[[PyLoadAPI], Awaitable[Any]]
+
+class PyLoadButtonEntity(StrEnum):
+    ABORT_DOWNLOADS = 'abort_downloads'
+    RESTART_FAILED = 'restart_failed'
+    DELETE_FINISHED = 'delete_finished'
+    RESTART = 'restart'
+
+SENSOR_DESCRIPTIONS: tuple[PyLoadButtonEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class PyLoadBinarySensor(BasePyLoadEntity, ButtonEntity):
+    entity_description: PyLoadButtonEntityDescription
+    async def async_press(self) -> None: ...

homeassistant-stubs/components/pyload/config_flow.pyi

@@ -0,0 +1,23 @@
+from .const import DEFAULT_NAME as DEFAULT_NAME, DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from collections.abc import Mapping
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.const import CONF_NAME as CONF_NAME, CONF_PASSWORD as CONF_PASSWORD, CONF_URL as CONF_URL, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession
+from homeassistant.helpers.selector import TextSelector as TextSelector, TextSelectorConfig as TextSelectorConfig, TextSelectorType as TextSelectorType
+from typing import Any
+
+_LOGGER: Incomplete
+STEP_USER_DATA_SCHEMA: Incomplete
+REAUTH_SCHEMA: Incomplete
+
+async def validate_input(hass: HomeAssistant, user_input: dict[str, Any]) -> None: ...
+
+class PyLoadConfigFlow(ConfigFlow, domain=DOMAIN):
+    VERSION: int
+    MINOR_VERSION: int
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    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_reconfigure(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/pyload/const.pyi

@@ -0,0 +1,6 @@
+DOMAIN: str
+DEFAULT_NAME: str
+DEFAULT_PORT: int
+MANUFACTURER: str
+SERVICE_NAME: str
+UNIT_DOWNLOADS: str

homeassistant-stubs/components/pyload/coordinator.pyi

@@ -0,0 +1,33 @@
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_USERNAME as CONF_USERNAME
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed, ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+from pyloadapi import PyLoadAPI as PyLoadAPI
+
+_LOGGER: Incomplete
+SCAN_INTERVAL: Incomplete
+
+@dataclass(kw_only=True)
+class PyLoadData:
+    pause: bool
+    active: int
+    queue: int
+    total: int
+    speed: float
+    download: bool
+    reconnect: bool
+    captcha: bool | None = ...
+    free_space: int
+type PyLoadConfigEntry = ConfigEntry[PyLoadCoordinator]
+
+class PyLoadCoordinator(DataUpdateCoordinator[PyLoadData]):
+    config_entry: PyLoadConfigEntry
+    pyload: Incomplete
+    version: str | None
+    def __init__(self, hass: HomeAssistant, config_entry: PyLoadConfigEntry, pyload: PyLoadAPI) -> None: ...
+    async def _async_update_data(self) -> PyLoadData: ...
+    async def _async_setup(self) -> None: ...

homeassistant-stubs/components/pyload/diagnostics.pyi

@@ -0,0 +1,10 @@
+from .coordinator import PyLoadConfigEntry as PyLoadConfigEntry, PyLoadData as PyLoadData
+from _typeshed import Incomplete
+from homeassistant.components.diagnostics import REDACTED as REDACTED, async_redact_data as async_redact_data
+from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_URL as CONF_URL, CONF_USERNAME as CONF_USERNAME
+from homeassistant.core import HomeAssistant as HomeAssistant
+from typing import Any
+
+TO_REDACT: Incomplete
+
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry: PyLoadConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/pyload/entity.pyi

@@ -0,0 +1,13 @@
+from .const import DOMAIN as DOMAIN, MANUFACTURER as MANUFACTURER, SERVICE_NAME as SERVICE_NAME
+from .coordinator import PyLoadCoordinator as PyLoadCoordinator
+from _typeshed import Incomplete
+from homeassistant.components.button import EntityDescription as EntityDescription
+from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+
+class BasePyLoadEntity(CoordinatorEntity[PyLoadCoordinator]):
+    _attr_has_entity_name: bool
+    _attr_unique_id: Incomplete
+    entity_description: Incomplete
+    _attr_device_info: Incomplete
+    def __init__(self, coordinator: PyLoadCoordinator, entity_description: EntityDescription) -> None: ...

homeassistant-stubs/components/pyload/sensor.pyi

@@ -0,0 +1,33 @@
+from .const import UNIT_DOWNLOADS as UNIT_DOWNLOADS
+from .coordinator import PyLoadConfigEntry as PyLoadConfigEntry, PyLoadData as PyLoadData
+from .entity import BasePyLoadEntity as BasePyLoadEntity
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from enum import StrEnum
+from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
+from homeassistant.const import UnitOfDataRate as UnitOfDataRate, UnitOfInformation as UnitOfInformation
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from homeassistant.helpers.typing import StateType as StateType
+
+PARALLEL_UPDATES: int
+
+class PyLoadSensorEntity(StrEnum):
+    ACTIVE = 'active'
+    FREE_SPACE = 'free_space'
+    QUEUE = 'queue'
+    SPEED = 'speed'
+    TOTAL = 'total'
+
+@dataclass(kw_only=True, frozen=True)
+class PyLoadSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[PyLoadData], StateType]
+
+SENSOR_DESCRIPTIONS: tuple[PyLoadSensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class PyLoadSensor(BasePyLoadEntity, SensorEntity):
+    entity_description: PyLoadSensorEntityDescription
+    @property
+    def native_value(self) -> StateType: ...

homeassistant-stubs/components/pyload/switch.pyi

@@ -0,0 +1,37 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import PyLoadConfigEntry as PyLoadConfigEntry, PyLoadData as PyLoadData
+from .entity import BasePyLoadEntity as BasePyLoadEntity
+from collections.abc import Awaitable, Callable as Callable
+from dataclasses import dataclass
+from enum import StrEnum
+from homeassistant.components.switch import SwitchDeviceClass as SwitchDeviceClass, SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from pyloadapi import PyLoadAPI as PyLoadAPI
+from typing import Any
+
+PARALLEL_UPDATES: int
+
+class PyLoadSwitch(StrEnum):
+    PAUSE_RESUME_QUEUE = 'download'
+    RECONNECT = 'reconnect'
+
+@dataclass(kw_only=True, frozen=True)
+class PyLoadSwitchEntityDescription(SwitchEntityDescription):
+    turn_on_fn: Callable[[PyLoadAPI], Awaitable[Any]]
+    turn_off_fn: Callable[[PyLoadAPI], Awaitable[Any]]
+    toggle_fn: Callable[[PyLoadAPI], Awaitable[Any]]
+    value_fn: Callable[[PyLoadData], bool]
+
+SENSOR_DESCRIPTIONS: tuple[PyLoadSwitchEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class PyLoadSwitchEntity(BasePyLoadEntity, SwitchEntity):
+    entity_description: PyLoadSwitchEntityDescription
+    @property
+    def is_on(self) -> bool | None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_toggle(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/qbus/climate.pyi

@@ -0,0 +1,41 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import QbusConfigEntry as QbusConfigEntry
+from .entity import QbusEntity as QbusEntity, add_new_outputs as add_new_outputs
+from _typeshed import Incomplete
+from homeassistant.components.climate import ClimateEntity as ClimateEntity, ClimateEntityFeature as ClimateEntityFeature, HVACAction as HVACAction, HVACMode as HVACMode
+from homeassistant.components.mqtt import ReceiveMessage as ReceiveMessage
+from homeassistant.const import ATTR_TEMPERATURE as ATTR_TEMPERATURE, UnitOfTemperature as UnitOfTemperature
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
+from homeassistant.helpers.debounce import Debouncer as Debouncer
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback as AddConfigEntryEntitiesCallback
+from qbusmqttapi.discovery import QbusMqttOutput as QbusMqttOutput
+from typing import Any
+
+PARALLEL_UPDATES: int
+STATE_REQUEST_DELAY: int
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: QbusConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback) -> None: ...
+
+class QbusClimate(QbusEntity, ClimateEntity):
+    _attr_hvac_modes: Incomplete
+    _attr_supported_features: Incomplete
+    _attr_temperature_unit: Incomplete
+    _attr_hvac_action: Incomplete
+    _attr_hvac_mode: Incomplete
+    _attr_min_temp: float
+    _attr_max_temp: float
+    _attr_target_temperature_step: float
+    _attr_preset_modes: list[str]
+    _attr_preset_mode: str
+    _request_state_debouncer: Debouncer | None
+    def __init__(self, mqtt_output: QbusMqttOutput) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    async def async_set_preset_mode(self, preset_mode: str) -> None: ...
+    async def async_set_temperature(self, **kwargs: Any) -> None: ...
+    _attr_current_temperature: Incomplete
+    _attr_target_temperature: Incomplete
+    async def _state_received(self, msg: ReceiveMessage) -> None: ...
+    def _set_hvac_action(self) -> None: ...
+    async def _async_request_state(self) -> None: ...

homeassistant-stubs/components/qbus/light.pyi

@@ -20,6 +20,6 @@ class QbusLight(QbusEntity, LightEntity):
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     async def _state_received(self, msg: ReceiveMessage) -> None: ...
-    _attr_is_on: bool
-    _attr_brightness: int
-    def _set_state(self, *, percentage: int | None = None, on: bool | None = None) -> None: ...
+    _attr_is_on: Incomplete
+    _attr_brightness: Incomplete
+    def _set_state(self, percentage: int = 0) -> None: ...

homeassistant-stubs/components/recorder/const.pyi

@@ -25,6 +25,7 @@ CONTEXT_ID_AS_BINARY_SCHEMA_VERSION: int
 EVENT_TYPE_IDS_SCHEMA_VERSION: int
 STATES_META_SCHEMA_VERSION: int
 LAST_REPORTED_SCHEMA_VERSION: int
+CIRCULAR_MEAN_SCHEMA_VERSION: int
 LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION: int
 LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION: int
 INTEGRATION_PLATFORM_COMPILE_STATISTICS: str

homeassistant-stubs/components/recorder/core.pyi

@@ -6,7 +6,7 @@ from . import migration as migration, statistics as statistics
 from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, DEFAULT_MAX_BIND_VARS as DEFAULT_MAX_BIND_VARS, DOMAIN as DOMAIN, KEEPALIVE_TIME as KEEPALIVE_TIME, LAST_REPORTED_SCHEMA_VERSION as LAST_REPORTED_SCHEMA_VERSION, MARIADB_PYMYSQL_URL_PREFIX as MARIADB_PYMYSQL_URL_PREFIX, MARIADB_URL_PREFIX as MARIADB_URL_PREFIX, MAX_QUEUE_BACKLOG_MIN_VALUE as MAX_QUEUE_BACKLOG_MIN_VALUE, MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG as MIN_AVAILABLE_MEMORY_FOR_QUEUE_BACKLOG, MYSQLDB_PYMYSQL_URL_PREFIX as MYSQLDB_PYMYSQL_URL_PREFIX, MYSQLDB_URL_PREFIX as MYSQLDB_URL_PREFIX, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, SupportedDialect as SupportedDialect
 from .db_schema import Base as Base, EventData as EventData, EventTypes as EventTypes, Events as Events, SCHEMA_VERSION as SCHEMA_VERSION, StateAttributes as StateAttributes, States as States, StatesMeta as StatesMeta, Statistics as Statistics, StatisticsShortTerm as StatisticsShortTerm
 from .executor import DBInterruptibleThreadPoolExecutor as DBInterruptibleThreadPoolExecutor
-from .models import DatabaseEngine as DatabaseEngine, StatisticData as StatisticData, StatisticMetaData as StatisticMetaData, UnsupportedDialect as UnsupportedDialect
+from .models import DatabaseEngine as DatabaseEngine, StatisticData as StatisticData, StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData, UnsupportedDialect as UnsupportedDialect
 from .pool import MutexPool as MutexPool, POOL_SIZE as POOL_SIZE, RecorderPool as RecorderPool
 from .table_managers.event_data import EventDataManager as EventDataManager
 from .table_managers.event_types import EventTypeManager as EventTypeManager
@@ -36,7 +36,6 @@ from sqlalchemy.orm.session import Session as Session
 from typing import Any
 
 _LOGGER: Incomplete
-DEFAULT_URL: str
 EXPIRE_AFTER_COMMITS: int
 SHUTDOWN_TASK: Incomplete
 COMMIT_TASK: Incomplete

homeassistant-stubs/components/recorder/db_schema.pyi

@@ -1,5 +1,5 @@
 from .const import ALL_DOMAIN_EXCLUDE_ATTRS as ALL_DOMAIN_EXCLUDE_ATTRS, SupportedDialect as SupportedDialect
-from .models import StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMetaData as StatisticMetaData, bytes_to_ulid_or_none as bytes_to_ulid_or_none, bytes_to_uuid_hex_or_none as bytes_to_uuid_hex_or_none, datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp, ulid_to_bytes_or_none as ulid_to_bytes_or_none, uuid_hex_to_bytes_or_none as uuid_hex_to_bytes_or_none
+from .models import StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMeanType as StatisticMeanType, StatisticMetaData as StatisticMetaData, bytes_to_ulid_or_none as bytes_to_ulid_or_none, bytes_to_uuid_hex_or_none as bytes_to_uuid_hex_or_none, datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp, ulid_to_bytes_or_none as ulid_to_bytes_or_none, uuid_hex_to_bytes_or_none as uuid_hex_to_bytes_or_none
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from datetime import datetime, timedelta
@@ -209,6 +209,7 @@ class StatisticsBase:
     start: Mapped[datetime | None]
     start_ts: Mapped[float | None]
     mean: Mapped[float | None]
+    mean_weight: Mapped[float | None]
     min: Mapped[float | None]
     max: Mapped[float | None]
     last_reset: Mapped[datetime | None]
@@ -247,6 +248,7 @@ class _StatisticsMeta:
     has_mean: Mapped[bool | None]
     has_sum: Mapped[bool | None]
     name: Mapped[str | None]
+    mean_type: Mapped[StatisticMeanType]
     @staticmethod
     def from_meta(meta: StatisticMetaData) -> StatisticsMeta: ...
 

homeassistant-stubs/components/recorder/migration.pyi

@@ -3,7 +3,7 @@ from . import Recorder as Recorder
 from .auto_repairs.statistics.duplicates import delete_statistics_duplicates as delete_statistics_duplicates, delete_statistics_meta_duplicates as delete_statistics_meta_duplicates
 from .const import CONTEXT_ID_AS_BINARY_SCHEMA_VERSION as CONTEXT_ID_AS_BINARY_SCHEMA_VERSION, EVENT_TYPE_IDS_SCHEMA_VERSION as EVENT_TYPE_IDS_SCHEMA_VERSION, LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION as LEGACY_STATES_EVENT_FOREIGN_KEYS_FIXED_SCHEMA_VERSION, LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION as LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION, STATES_META_SCHEMA_VERSION as STATES_META_SCHEMA_VERSION, SupportedDialect as SupportedDialect
 from .db_schema import BIG_INTEGER_SQL as BIG_INTEGER_SQL, Base as Base, CONTEXT_ID_BIN_MAX_LENGTH as CONTEXT_ID_BIN_MAX_LENGTH, DOUBLE_PRECISION_TYPE_SQL as DOUBLE_PRECISION_TYPE_SQL, EventTypes as EventTypes, Events as Events, LEGACY_STATES_ENTITY_ID_LAST_UPDATED_TS_INDEX as LEGACY_STATES_ENTITY_ID_LAST_UPDATED_TS_INDEX, LEGACY_STATES_EVENT_ID_INDEX as LEGACY_STATES_EVENT_ID_INDEX, LegacyBase as LegacyBase, MYSQL_COLLATE as MYSQL_COLLATE, MYSQL_DEFAULT_CHARSET as MYSQL_DEFAULT_CHARSET, MigrationChanges as MigrationChanges, SCHEMA_VERSION as SCHEMA_VERSION, STATISTICS_TABLES as STATISTICS_TABLES, SchemaChanges as SchemaChanges, States as States, StatesMeta as StatesMeta, Statistics as Statistics, StatisticsMeta as StatisticsMeta, StatisticsRuns as StatisticsRuns, StatisticsShortTerm as StatisticsShortTerm, TABLE_STATES as TABLE_STATES
-from .models import process_timestamp as process_timestamp
+from .models import StatisticMeanType as StatisticMeanType, process_timestamp as process_timestamp
 from .models.time import datetime_to_timestamp_or_none as datetime_to_timestamp_or_none
 from .queries import batch_cleanup_entity_ids as batch_cleanup_entity_ids, delete_duplicate_short_term_statistics_row as delete_duplicate_short_term_statistics_row, delete_duplicate_statistics_row as delete_duplicate_statistics_row, find_entity_ids_to_migrate as find_entity_ids_to_migrate, find_event_type_to_migrate as find_event_type_to_migrate, find_events_context_ids_to_migrate as find_events_context_ids_to_migrate, find_states_context_ids_to_migrate as find_states_context_ids_to_migrate, find_unmigrated_short_term_statistics_rows as find_unmigrated_short_term_statistics_rows, find_unmigrated_statistics_rows as find_unmigrated_statistics_rows, get_migration_changes as get_migration_changes, has_entity_ids_to_migrate as has_entity_ids_to_migrate, has_event_type_to_migrate as has_event_type_to_migrate, has_events_context_ids_to_migrate as has_events_context_ids_to_migrate, has_states_context_ids_to_migrate as has_states_context_ids_to_migrate, has_used_states_entity_ids as has_used_states_entity_ids, has_used_states_event_ids as has_used_states_event_ids, migrate_single_short_term_statistics_row_to_timestamp as migrate_single_short_term_statistics_row_to_timestamp, migrate_single_statistics_row_to_timestamp as migrate_single_statistics_row_to_timestamp
 from .statistics import cleanup_statistics_timestamp_migration as cleanup_statistics_timestamp_migration, get_start_time as get_start_time
@@ -16,12 +16,13 @@ from dataclasses import dataclass
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.util.enum import try_parse_enum as try_parse_enum
 from homeassistant.util.ulid import ulid_at_time as ulid_at_time, ulid_to_bytes as ulid_to_bytes
+from sqlalchemy import ForeignKeyConstraint
 from sqlalchemy.engine import CursorResult as CursorResult, Engine as Engine
 from sqlalchemy.orm import DeclarativeBase as DeclarativeBase
 from sqlalchemy.orm.session import Session as Session
 from sqlalchemy.schema import AddConstraint
 from sqlalchemy.sql.lambdas import StatementLambdaElement as StatementLambdaElement
-from typing import Any, final
+from typing import Any, TypedDict, final
 
 LIVE_MIGRATION_MIN_SCHEMA_VERSION: int
 MIGRATION_NOTE_OFFLINE: str
@@ -36,6 +37,8 @@ class _ColumnTypesForDialect:
     big_int_type: str
     timestamp_type: str
     context_bin_type: str
+    small_int_type: str
+    double_type: str
 
 _MYSQL_COLUMN_TYPES: Incomplete
 _POSTGRESQL_COLUMN_TYPES: Incomplete
@@ -75,6 +78,11 @@ def _execute_or_collect_error(session_maker: Callable[[], Session], query: str,
 def _drop_index(session_maker: Callable[[], Session], table_name: str, index_name: str, quiet: bool | None = None) -> None: ...
 def _add_columns(session_maker: Callable[[], Session], table_name: str, columns_def: list[str]) -> None: ...
 def _modify_columns(session_maker: Callable[[], Session], engine: Engine, table_name: str, columns_def: list[str]) -> None: ...
+
+class _FKAlterDict(TypedDict):
+    old_fk: ForeignKeyConstraint
+    columns: list[str]
+
 def _update_states_table_with_foreign_key_options(session_maker: Callable[[], Session], engine: Engine) -> None: ...
 def _drop_foreign_key_constraints(session_maker: Callable[[], Session], engine: Engine, table: str, column: str) -> None: ...
 def _restore_foreign_key_constraints(session_maker: Callable[[], Session], engine: Engine, foreign_columns: list[tuple[str, str, str | None, str | None]]) -> None: ...
@@ -245,6 +253,12 @@ class _SchemaVersion47Migrator(_SchemaVersionMigrator, target_version=47):
 class _SchemaVersion48Migrator(_SchemaVersionMigrator, target_version=48):
     def _apply_update(self) -> None: ...
 
+class _SchemaVersion49Migrator(_SchemaVersionMigrator, target_version=49):
+    def _apply_update(self) -> None: ...
+
+class _SchemaVersion50Migrator(_SchemaVersionMigrator, target_version=50):
+    def _apply_update(self) -> None: ...
+
 def _migrate_statistics_columns_to_timestamp_removing_duplicates(hass: HomeAssistant, instance: Recorder, session_maker: Callable[[], Session], engine: Engine) -> None: ...
 def _correct_table_character_set_and_collation(table: str, session_maker: Callable[[], Session]) -> None: ...
 def _wipe_old_string_time_columns(instance: Recorder, engine: Engine, session: Session) -> None: ...