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

homeassistant-stubs/components/prusalink/coordinator.pyi

@@ -0,0 +1,34 @@
+import abc
+from .const import DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from abc import ABC, abstractmethod
+from datetime import timedelta
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+from pyprusalink import JobInfo, LegacyPrinterStatus, PrinterStatus, PrusaLink as PrusaLink
+from typing import TypeVar
+
+_LOGGER: Incomplete
+T = TypeVar('T', PrinterStatus, LegacyPrinterStatus, JobInfo)
+
+class PrusaLinkUpdateCoordinator(DataUpdateCoordinator[T], ABC, metaclass=abc.ABCMeta):
+    config_entry: ConfigEntry
+    expect_change_until: float
+    api: Incomplete
+    def __init__(self, hass: HomeAssistant, api: PrusaLink) -> None: ...
+    update_interval: Incomplete
+    async def _async_update_data(self) -> T: ...
+    @abstractmethod
+    async def _fetch_data(self) -> T: ...
+    def expect_change(self) -> None: ...
+    def _get_update_interval(self, data: T) -> timedelta: ...
+
+class StatusCoordinator(PrusaLinkUpdateCoordinator[PrinterStatus]):
+    async def _fetch_data(self) -> PrinterStatus: ...
+
+class LegacyStatusCoordinator(PrusaLinkUpdateCoordinator[LegacyPrinterStatus]):
+    async def _fetch_data(self) -> LegacyPrinterStatus: ...
+
+class JobUpdateCoordinator(PrusaLinkUpdateCoordinator[JobInfo]):
+    async def _fetch_data(self) -> JobInfo: ...

homeassistant-stubs/components/prusalink/sensor.pyi

@@ -1,4 +1,6 @@
-from . import DOMAIN as DOMAIN, PrusaLinkEntity as PrusaLinkEntity, PrusaLinkUpdateCoordinator as PrusaLinkUpdateCoordinator
+from . import PrusaLinkEntity as PrusaLinkEntity
+from .const import DOMAIN as DOMAIN
+from .coordinator import PrusaLinkUpdateCoordinator as PrusaLinkUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass

homeassistant-stubs/components/pure_energie/__init__.pyi

@@ -1,25 +1,12 @@
-from .const import DOMAIN as DOMAIN, LOGGER as LOGGER, SCAN_INTERVAL as SCAN_INTERVAL
+from .const import DOMAIN as DOMAIN
+from .coordinator import PureEnergieDataUpdateCoordinator as PureEnergieDataUpdateCoordinator
 from _typeshed import Incomplete
-from gridnet import Device as Device, SmartBridge as SmartBridge
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_HOST as CONF_HOST, Platform as Platform
+from homeassistant.const import Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
-from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from typing import NamedTuple
 
 PLATFORMS: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-
-class PureEnergieData(NamedTuple):
-    device: Device
-    smartbridge: SmartBridge
-
-class PureEnergieDataUpdateCoordinator(DataUpdateCoordinator[PureEnergieData]):
-    config_entry: ConfigEntry
-    gridnet: Incomplete
-    def __init__(self, hass: HomeAssistant) -> None: ...
-    async def _async_update_data(self) -> PureEnergieData: ...

homeassistant-stubs/components/pure_energie/coordinator.pyi

@@ -0,0 +1,19 @@
+from .const import DOMAIN as DOMAIN, LOGGER as LOGGER, SCAN_INTERVAL as SCAN_INTERVAL
+from _typeshed import Incomplete
+from gridnet import Device as Device, SmartBridge as SmartBridge
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_HOST as CONF_HOST
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+from typing import NamedTuple
+
+class PureEnergieData(NamedTuple):
+    device: Device
+    smartbridge: SmartBridge
+
+class PureEnergieDataUpdateCoordinator(DataUpdateCoordinator[PureEnergieData]):
+    config_entry: ConfigEntry
+    gridnet: Incomplete
+    def __init__(self, hass: HomeAssistant) -> None: ...
+    async def _async_update_data(self) -> PureEnergieData: ...

homeassistant-stubs/components/pure_energie/diagnostics.pyi

@@ -1,5 +1,5 @@
-from . import PureEnergieDataUpdateCoordinator as PureEnergieDataUpdateCoordinator
 from .const import DOMAIN as DOMAIN
+from .coordinator import PureEnergieDataUpdateCoordinator as PureEnergieDataUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.components.diagnostics import async_redact_data as async_redact_data
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/pure_energie/sensor.pyi

@@ -1,5 +1,5 @@
-from . import PureEnergieData as PureEnergieData, PureEnergieDataUpdateCoordinator as PureEnergieDataUpdateCoordinator
 from .const import DOMAIN as DOMAIN
+from .coordinator import PureEnergieData as PureEnergieData, PureEnergieDataUpdateCoordinator as PureEnergieDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass

homeassistant-stubs/components/rainmachine/__init__.pyi

@@ -1,7 +1,7 @@
 from .config_flow import get_client_controller as get_client_controller
 from .const import CONF_ALLOW_INACTIVE_ZONES_TO_RUN as CONF_ALLOW_INACTIVE_ZONES_TO_RUN, CONF_DEFAULT_ZONE_RUN_TIME as CONF_DEFAULT_ZONE_RUN_TIME, CONF_DURATION as CONF_DURATION, CONF_USE_APP_RUN_TIMES as CONF_USE_APP_RUN_TIMES, DATA_API_VERSIONS as DATA_API_VERSIONS, DATA_MACHINE_FIRMWARE_UPDATE_STATUS as DATA_MACHINE_FIRMWARE_UPDATE_STATUS, DATA_PROGRAMS as DATA_PROGRAMS, DATA_PROVISION_SETTINGS as DATA_PROVISION_SETTINGS, DATA_RESTRICTIONS_CURRENT as DATA_RESTRICTIONS_CURRENT, DATA_RESTRICTIONS_UNIVERSAL as DATA_RESTRICTIONS_UNIVERSAL, DATA_ZONES as DATA_ZONES, DEFAULT_ZONE_RUN as DEFAULT_ZONE_RUN, DOMAIN as DOMAIN, LOGGER as LOGGER
+from .coordinator import RainMachineDataUpdateCoordinator as RainMachineDataUpdateCoordinator
 from .model import RainMachineEntityDescription as RainMachineEntityDescription
-from .util import RainMachineDataUpdateCoordinator as RainMachineDataUpdateCoordinator
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from dataclasses import dataclass

homeassistant-stubs/components/rainmachine/coordinator.pyi

@@ -0,0 +1,22 @@
+from .const import LOGGER as LOGGER
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable, Coroutine
+from datetime import timedelta
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+from typing import Any
+
+SIGNAL_REBOOT_COMPLETED: str
+SIGNAL_REBOOT_REQUESTED: str
+
+class RainMachineDataUpdateCoordinator(DataUpdateCoordinator[dict]):
+    config_entry: ConfigEntry
+    _rebooting: bool
+    _signal_handler_unsubs: Incomplete
+    signal_reboot_completed: Incomplete
+    signal_reboot_requested: Incomplete
+    def __init__(self, hass: HomeAssistant, *, entry: ConfigEntry, name: str, api_category: str, update_interval: timedelta, update_method: Callable[[], Coroutine[Any, Any, dict]]) -> None: ...
+    last_update_success: bool
+    def async_initialize(self) -> None: ...

homeassistant-stubs/components/rainmachine/switch.pyi

@@ -3,7 +3,7 @@ from .const import CONF_ALLOW_INACTIVE_ZONES_TO_RUN as CONF_ALLOW_INACTIVE_ZONES
 from .model import RainMachineEntityDescription as RainMachineEntityDescription
 from .util import RUN_STATE_MAP as RUN_STATE_MAP, key_exists as key_exists
 from _typeshed import Incomplete
-from collections.abc import Awaitable, Callable as Callable, Coroutine
+from collections.abc import Awaitable as Awaitable, Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.switch import SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -12,13 +12,13 @@ from homeassistant.core import HomeAssistant as HomeAssistant, callback as callb
 from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers import entity_platform as entity_platform
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from typing import Any, Concatenate, ParamSpec, TypeVar
+from typing import Any, Concatenate
 
+ATTR_ACTIVITY_TYPE: str
 ATTR_AREA: str
 ATTR_CS_ON: str
 ATTR_CURRENT_CYCLE: str
 ATTR_CYCLES: str
-ATTR_ZONE_RUN_TIME: str
 ATTR_DELAY: str
 ATTR_DELAY_ON: str
 ATTR_FIELD_CAPACITY: str
@@ -34,14 +34,13 @@ ATTR_STATUS: str
 ATTR_SUN_EXPOSURE: str
 ATTR_VEGETATION_TYPE: str
 ATTR_ZONES: str
+ATTR_ZONE_RUN_TIME: str
 DAYS: Incomplete
 SOIL_TYPE_MAP: Incomplete
 SLOPE_TYPE_MAP: Incomplete
 SPRINKLER_TYPE_MAP: Incomplete
 SUN_EXPOSURE_MAP: Incomplete
 VEGETATION_MAP: Incomplete
-_T = TypeVar('_T', bound='RainMachineBaseSwitch')
-_P = ParamSpec('_P')
 
 def raise_on_request_error(func: Callable[Concatenate[_T, _P], Awaitable[None]]) -> Callable[Concatenate[_T, _P], Coroutine[Any, Any, None]]: ...
 
@@ -51,8 +50,9 @@ class RainMachineSwitchDescription(SwitchEntityDescription, RainMachineEntityDes
 
 @dataclass(frozen=True, kw_only=True)
 class RainMachineActivitySwitchDescription(RainMachineSwitchDescription):
+    kind: str
     uid: 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, api_category, uid) -> 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, api_category, kind, uid) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
 class RainMachineRestrictionSwitchDescription(RainMachineSwitchDescription):
@@ -79,6 +79,7 @@ class RainMachineBaseSwitch(RainMachineEntity, SwitchEntity):
 class RainMachineActivitySwitch(RainMachineBaseSwitch):
     _attr_icon: str
     entity_description: RainMachineActivitySwitchDescription
+    def __init__(self, entry: ConfigEntry, data: RainMachineData, description: RainMachineSwitchDescription) -> None: ...
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     async def async_turn_off_when_active(self, **kwargs: Any) -> None: ...
     _attr_is_on: bool
@@ -89,6 +90,7 @@ class RainMachineEnabledSwitch(RainMachineBaseSwitch):
     _attr_entity_category: Incomplete
     _attr_icon: str
     entity_description: RainMachineActivitySwitchDescription
+    def __init__(self, entry: ConfigEntry, data: RainMachineData, description: RainMachineSwitchDescription) -> None: ...
     _attr_is_on: Incomplete
     def update_from_latest_data(self) -> None: ...
 

homeassistant-stubs/components/rainmachine/util.pyi

@@ -1,18 +1,12 @@
 from .const import LOGGER as LOGGER
 from _typeshed import Incomplete
-from collections.abc import Awaitable, Callable as Callable, Iterable
+from collections.abc import Iterable
 from dataclasses import dataclass
-from datetime import timedelta
 from enum import StrEnum
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
 from typing import Any
 
-SIGNAL_REBOOT_COMPLETED: str
-SIGNAL_REBOOT_REQUESTED: str
-
 class RunStates(StrEnum):
     NOT_RUNNING: str
     QUEUED: str
@@ -31,13 +25,3 @@ class EntityDomainReplacementStrategy:
 
 def async_finish_entity_domain_replacements(hass: HomeAssistant, entry: ConfigEntry, entity_replacement_strategies: Iterable[EntityDomainReplacementStrategy]) -> None: ...
 def key_exists(data: dict[str, Any], search_key: str) -> bool: ...
-
-class RainMachineDataUpdateCoordinator(DataUpdateCoordinator[dict]):
-    config_entry: ConfigEntry
-    _rebooting: bool
-    _signal_handler_unsubs: Incomplete
-    signal_reboot_completed: Incomplete
-    signal_reboot_requested: Incomplete
-    def __init__(self, hass: HomeAssistant, *, entry: ConfigEntry, name: str, api_category: str, update_interval: timedelta, update_method: Callable[..., Awaitable]) -> None: ...
-    last_update_success: bool
-    def async_initialize(self) -> None: ...

homeassistant-stubs/components/recorder/core.pyi

@@ -20,6 +20,7 @@ from .util import async_create_backup_failure_issue as async_create_backup_failu
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable
 from datetime import datetime
+from functools import cached_property as cached_property
 from homeassistant.components import persistent_notification as persistent_notification
 from homeassistant.const import ATTR_ENTITY_ID as ATTR_ENTITY_ID, EVENT_HOMEASSISTANT_CLOSE as EVENT_HOMEASSISTANT_CLOSE, EVENT_HOMEASSISTANT_FINAL_WRITE as EVENT_HOMEASSISTANT_FINAL_WRITE, EVENT_STATE_CHANGED as EVENT_STATE_CHANGED, MATCH_ALL as MATCH_ALL
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, Event as Event, EventStateChangedData as EventStateChangedData, HomeAssistant as HomeAssistant, callback as callback
@@ -31,10 +32,9 @@ from homeassistant.util.event_type import EventType as EventType
 from sqlalchemy.engine import Engine as Engine
 from sqlalchemy.engine.interfaces import DBAPIConnection as DBAPIConnection
 from sqlalchemy.orm.session import Session as Session
-from typing import Any, TypeVar
+from typing import Any
 
 _LOGGER: Incomplete
-T = TypeVar('T')
 DEFAULT_URL: str
 EXPIRE_AFTER_COMMITS: int
 SHUTDOWN_TASK: Incomplete
@@ -52,6 +52,7 @@ class Recorder(threading.Thread):
     stop_requested: bool
     hass: Incomplete
     thread_id: Incomplete
+    recorder_and_worker_thread_ids: Incomplete
     auto_purge: Incomplete
     auto_repack: Incomplete
     keep_days: Incomplete
@@ -103,7 +104,7 @@ class Recorder(threading.Thread):
     def __init__(self, hass: HomeAssistant, auto_purge: bool, auto_repack: bool, keep_days: int, commit_interval: int, uri: str, db_max_retries: int, db_retry_wait: int, entity_filter: Callable[[str], bool], exclude_event_types: set[EventType[Any] | str]) -> None: ...
     @property
     def backlog(self) -> int: ...
-    @property
+    @cached_property
     def dialect_name(self) -> SupportedDialect | None: ...
     @property
     def _using_file_sqlite(self) -> bool: ...
@@ -117,7 +118,7 @@ class Recorder(threading.Thread):
     def async_initialize(self) -> None: ...
     def _async_keep_alive(self, now: datetime) -> None: ...
     def _async_commit(self, now: datetime) -> None: ...
-    def async_add_executor_job(self, target: Callable[..., T], *args: Any) -> asyncio.Future[T]: ...
+    def async_add_executor_job(self, target: Callable[..., _T], *args: Any) -> asyncio.Future[_T]: ...
     def _stop_executor(self) -> None: ...
     def _async_check_queue(self, *_: Any) -> None: ...
     def _available_memory(self) -> int: ...

homeassistant-stubs/components/recorder/executor.pyi

@@ -3,9 +3,10 @@ from collections.abc import Callable as Callable
 from homeassistant.util.executor import InterruptibleThreadPoolExecutor as InterruptibleThreadPoolExecutor
 from typing import Any
 
-def _worker_with_shutdown_hook(shutdown_hook: Callable[[], None], *args: Any, **kwargs: Any) -> None: ...
+def _worker_with_shutdown_hook(shutdown_hook: Callable[[], None], recorder_and_worker_thread_ids: set[int], *args: Any, **kwargs: Any) -> None: ...
 
 class DBInterruptibleThreadPoolExecutor(InterruptibleThreadPoolExecutor):
     _shutdown_hook: Incomplete
-    def __init__(self, *args: Any, **kwargs: Any) -> None: ...
+    recorder_and_worker_thread_ids: Incomplete
+    def __init__(self, recorder_and_worker_thread_ids: set[int], *args: Any, **kwargs: Any) -> None: ...
     def _adjust_thread_count(self) -> None: ...

homeassistant-stubs/components/recorder/pool.pyi

@@ -1,8 +1,7 @@
 import threading
-from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX
 from _typeshed import Incomplete
 from homeassistant.helpers.frame import report as report
-from homeassistant.util.loop import check_loop as check_loop
+from homeassistant.util.loop import raise_for_blocking_call as raise_for_blocking_call
 from sqlalchemy.pool import ConnectionPoolEntry as ConnectionPoolEntry, NullPool, SingletonThreadPool, StaticPool
 from typing import Any
 
@@ -13,9 +12,9 @@ POOL_SIZE: int
 ADVISE_MSG: str
 
 class RecorderPool(SingletonThreadPool, NullPool):
-    def __init__(self, *args: Any, **kw: Any) -> None: ...
-    @property
-    def recorder_or_dbworker(self) -> bool: ...
+    recorder_and_worker_thread_ids: Incomplete
+    def __init__(self, creator: Any, recorder_and_worker_thread_ids: set[int] | None = None, **kw: Any) -> None: ...
+    def recreate(self) -> RecorderPool: ...
     def _do_return_conn(self, record: ConnectionPoolEntry) -> None: ...
     def shutdown(self) -> None: ...
     def dispose(self) -> None: ...

homeassistant-stubs/components/recorder/purge.pyi

@@ -3,10 +3,11 @@ from .db_schema import Events as Events, States as States, StatesMeta as StatesM
 from .models import DatabaseEngine as DatabaseEngine
 from .queries import attributes_ids_exist_in_states as attributes_ids_exist_in_states, attributes_ids_exist_in_states_with_fast_in_distinct as attributes_ids_exist_in_states_with_fast_in_distinct, data_ids_exist_in_events as data_ids_exist_in_events, data_ids_exist_in_events_with_fast_in_distinct as data_ids_exist_in_events_with_fast_in_distinct, delete_event_data_rows as delete_event_data_rows, delete_event_rows as delete_event_rows, delete_event_types_rows as delete_event_types_rows, delete_recorder_runs_rows as delete_recorder_runs_rows, delete_states_attributes_rows as delete_states_attributes_rows, delete_states_meta_rows as delete_states_meta_rows, delete_states_rows as delete_states_rows, delete_statistics_runs_rows as delete_statistics_runs_rows, delete_statistics_short_term_rows as delete_statistics_short_term_rows, disconnect_states_rows as disconnect_states_rows, find_entity_ids_to_purge as find_entity_ids_to_purge, find_event_types_to_purge as find_event_types_to_purge, find_events_to_purge as find_events_to_purge, find_latest_statistics_runs_run_id as find_latest_statistics_runs_run_id, find_legacy_detached_states_and_attributes_to_purge as find_legacy_detached_states_and_attributes_to_purge, find_legacy_event_state_and_attributes_and_data_ids_to_purge as find_legacy_event_state_and_attributes_and_data_ids_to_purge, find_legacy_row as find_legacy_row, find_short_term_statistics_to_purge as find_short_term_statistics_to_purge, find_states_to_purge as find_states_to_purge, find_statistics_runs_to_purge as find_statistics_runs_to_purge
 from .repack import repack_database as repack_database
-from .util import chunked_or_all as chunked_or_all, retryable_database_job as retryable_database_job, session_scope as session_scope
+from .util import retryable_database_job as retryable_database_job, session_scope as session_scope
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
 from datetime import datetime
+from homeassistant.util.collection import chunked_or_all as chunked_or_all
 from sqlalchemy.orm.session import Session as Session
 
 _LOGGER: Incomplete

homeassistant-stubs/components/recorder/table_managers/__init__.pyi

@@ -2,11 +2,9 @@ from ..core import Recorder as Recorder
 from _typeshed import Incomplete
 from homeassistant.util.event_type import EventType as EventType
 from lru import LRU
-from typing import Any, Generic, TypeVar
+from typing import Any
 
-_DataT = TypeVar('_DataT')
-
-class BaseTableManager(Generic[_DataT]):
+class BaseTableManager:
     _id_map: LRU[EventType[Any] | str, int]
     active: bool
     recorder: Incomplete

homeassistant-stubs/components/recorder/table_managers/event_data.pyi

@@ -2,10 +2,11 @@ from . import BaseLRUTableManager as BaseLRUTableManager
 from ..core import Recorder as Recorder
 from ..db_schema import EventData as EventData
 from ..queries import get_shared_event_datas as get_shared_event_datas
-from ..util import chunked as chunked, execute_stmt_lambda_element as execute_stmt_lambda_element
+from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element
 from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.core import Event as Event
+from homeassistant.util.collection import chunked as chunked
 from homeassistant.util.json import JSON_ENCODE_EXCEPTIONS as JSON_ENCODE_EXCEPTIONS
 from sqlalchemy.orm.session import Session as Session
 

homeassistant-stubs/components/recorder/table_managers/event_types.pyi

@@ -3,10 +3,11 @@ from ..core import Recorder as Recorder
 from ..db_schema import EventTypes as EventTypes
 from ..queries import find_event_type_ids as find_event_type_ids
 from ..tasks import RefreshEventTypesTask as RefreshEventTypesTask
-from ..util import chunked as chunked, execute_stmt_lambda_element as execute_stmt_lambda_element
+from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element
 from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.core import Event as Event
+from homeassistant.util.collection import chunked as chunked
 from homeassistant.util.event_type import EventType as EventType
 from sqlalchemy.orm.session import Session as Session
 from typing import Any

homeassistant-stubs/components/recorder/table_managers/state_attributes.pyi

@@ -2,10 +2,11 @@ from . import BaseLRUTableManager as BaseLRUTableManager
 from ..core import Recorder as Recorder
 from ..db_schema import StateAttributes as StateAttributes
 from ..queries import get_shared_attributes as get_shared_attributes
-from ..util import chunked as chunked, execute_stmt_lambda_element as execute_stmt_lambda_element
+from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element
 from _typeshed import Incomplete
 from collections.abc import Iterable
 from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData
+from homeassistant.util.collection import chunked as chunked
 from homeassistant.util.json import JSON_ENCODE_EXCEPTIONS as JSON_ENCODE_EXCEPTIONS
 from sqlalchemy.orm.session import Session as Session
 

homeassistant-stubs/components/recorder/table_managers/states_meta.pyi

@@ -2,9 +2,10 @@ from . import BaseLRUTableManager as BaseLRUTableManager
 from ..core import Recorder as Recorder
 from ..db_schema import StatesMeta as StatesMeta
 from ..queries import find_all_states_metadata_ids as find_all_states_metadata_ids, find_states_metadata_ids as find_states_metadata_ids
-from ..util import chunked as chunked, execute_stmt_lambda_element as execute_stmt_lambda_element
+from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element
 from collections.abc import Iterable
 from homeassistant.core import Event as Event, EventStateChangedData as EventStateChangedData
+from homeassistant.util.collection import chunked as chunked
 from sqlalchemy.orm.session import Session as Session
 
 CACHE_SIZE: int

homeassistant-stubs/components/recorder/util.pyi

@@ -4,7 +4,7 @@ from .db_schema import RecorderRuns as RecorderRuns, TABLES_TO_CHECK as TABLES_T
 from .models import DatabaseEngine as DatabaseEngine, DatabaseOptimizer as DatabaseOptimizer, StatisticPeriod as StatisticPeriod, UnsupportedDialect as UnsupportedDialect, process_timestamp as process_timestamp
 from _typeshed import Incomplete
 from awesomeversion import AwesomeVersion
-from collections.abc import Callable, Collection, Generator, Iterable, Sequence
+from collections.abc import Callable, Generator, Sequence
 from datetime import date, datetime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from sqlalchemy.engine import Result as Result, Row as Row
@@ -14,10 +14,8 @@ from sqlalchemy.orm.query import Query as Query
 from sqlalchemy.orm.session import Session as Session
 from sqlalchemy.sql.lambdas import StatementLambdaElement as StatementLambdaElement
 from sqlite3.dbapi2 import Cursor as SQLiteCursor
-from typing import Any, NoReturn, ParamSpec, TypeVar
+from typing import Any, NoReturn
 
-_RecorderT = TypeVar('_RecorderT', bound='Recorder')
-_P = ParamSpec('_P')
 _LOGGER: Incomplete
 RETRIES: int
 QUERY_RETRY_WAIT: float
@@ -73,11 +71,8 @@ def _is_retryable_error(instance: Recorder, err: OperationalError) -> bool: ...
 
 _FuncType: Incomplete
 
-def retryable_database_job(description: str) -> Callable[[_FuncType[_RecorderT, _P]], _FuncType[_RecorderT, _P]]: ...
-
-_WrappedFuncType: Incomplete
-
-def database_job_retry_wrapper(description: str, attempts: int = 5) -> Callable[[_WrappedFuncType[_RecorderT, _P]], _WrappedFuncType[_RecorderT, _P]]: ...
+def retryable_database_job(description: str) -> Callable[[_FuncType[_RecorderT, _P, bool]], _FuncType[_RecorderT, _P, bool]]: ...
+def database_job_retry_wrapper(description: str, attempts: int = 5) -> Callable[[_FuncType[_RecorderT, _P, None]], _FuncType[_RecorderT, _P, None]]: ...
 def periodic_db_cleanups(instance: Recorder) -> None: ...
 def write_lock_db_sqlite(instance: Recorder) -> Generator[None, None, None]: ...
 def async_migration_in_progress(hass: HomeAssistant) -> bool: ...
@@ -89,8 +84,5 @@ def get_instance(hass: HomeAssistant) -> Recorder: ...
 PERIOD_SCHEMA: Incomplete
 
 def resolve_period(period_def: StatisticPeriod) -> tuple[datetime | None, datetime | None]: ...
-def take(take_num: int, iterable: Iterable) -> list[Any]: ...
-def chunked(iterable: Iterable, chunked_num: int) -> Iterable[Any]: ...
-def chunked_or_all(iterable: Collection[Any], chunked_num: int) -> Iterable[Any]: ...
 def get_index_by_name(session: Session, table_name: str, index_name: str) -> str | None: ...
 def filter_unique_constraint_integrity_error(instance: Recorder, row_type: str) -> Callable[[Exception], bool]: ...

homeassistant-stubs/components/recorder/websocket_api.pyi

@@ -1,6 +1,7 @@
+from .core import Recorder as Recorder
 from .models import StatisticPeriod as StatisticPeriod
 from .statistics import STATISTIC_UNIT_TO_UNIT_CONVERTER as STATISTIC_UNIT_TO_UNIT_CONVERTER, async_add_external_statistics as async_add_external_statistics, async_change_statistics_unit as async_change_statistics_unit, async_import_statistics as async_import_statistics, async_list_statistic_ids as async_list_statistic_ids, list_statistic_ids as list_statistic_ids, statistic_during_period as statistic_during_period, statistics_during_period as statistics_during_period, validate_statistics as validate_statistics
-from .util import PERIOD_SCHEMA as PERIOD_SCHEMA, get_instance as get_instance, resolve_period as resolve_period
+from .util import PERIOD_SCHEMA as PERIOD_SCHEMA, get_instance as get_instance, resolve_period as resolve_period, session_scope as session_scope
 from _typeshed import Incomplete
 from datetime import datetime as dt
 from homeassistant.components import websocket_api as websocket_api
@@ -30,3 +31,5 @@ def ws_change_statistics_unit(hass: HomeAssistant, connection: websocket_api.Act
 async def ws_adjust_sum_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def ws_import_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def ws_info(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+def _get_recorded_entities(hass: HomeAssistant, msg_id: int, instance: Recorder) -> bytes: ...
+async def ws_get_recorded_entities(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...

homeassistant-stubs/components/renault/__init__.pyi

@@ -1,10 +1,18 @@
 from .const import CONF_LOCALE as CONF_LOCALE, DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
 from .renault_hub import RenaultHub as RenaultHub
-from .services import SERVICE_AC_START as SERVICE_AC_START, setup_services as setup_services, unload_services as unload_services
+from .services import setup_services as setup_services
+from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, 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 import device_registry as dr
+from homeassistant.helpers.typing import ConfigType as ConfigType
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+CONFIG_SCHEMA: Incomplete
+RenaultConfigEntry = ConfigEntry[RenaultHub]
+
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: RenaultConfigEntry, device_entry: dr.DeviceEntry) -> bool: ...

homeassistant-stubs/components/renault/binary_sensor.pyi

@@ -1,9 +1,7 @@
-from .const import DOMAIN as DOMAIN
+from . import RenaultConfigEntry as RenaultConfigEntry
 from .entity import RenaultDataEntity as RenaultDataEntity, RenaultDataEntityDescription as RenaultDataEntityDescription
-from .renault_hub import RenaultHub as RenaultHub
 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 homeassistant.helpers.typing import StateType as StateType
@@ -15,7 +13,7 @@ class RenaultBinarySensorEntityDescription(BinarySensorEntityDescription, Renaul
     on_value: StateType
     def __init__(self, coordinator, *, 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, on_key, on_value) -> None: ...
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RenaultBinarySensor(RenaultDataEntity[KamereonVehicleBatteryStatusData], BinarySensorEntity):
     entity_description: RenaultBinarySensorEntityDescription

homeassistant-stubs/components/renault/button.pyi

@@ -1,10 +1,8 @@
-from .const import DOMAIN as DOMAIN
+from . import RenaultConfigEntry as RenaultConfigEntry
 from .entity import RenaultEntity as RenaultEntity
-from .renault_hub import RenaultHub as RenaultHub
 from collections.abc import Callable as Callable, Coroutine
 from dataclasses import dataclass
 from homeassistant.components.button import ButtonEntity as ButtonEntity, ButtonEntityDescription as ButtonEntityDescription
-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 typing import Any
@@ -15,7 +13,7 @@ class RenaultButtonEntityDescription(ButtonEntityDescription):
     requires_electricity: 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, async_press, requires_electricity) -> None: ...
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RenaultButtonEntity(RenaultEntity, ButtonEntity):
     entity_description: RenaultButtonEntityDescription

homeassistant-stubs/components/renault/device_tracker.pyi

@@ -1,13 +1,11 @@
-from .const import DOMAIN as DOMAIN
+from . import RenaultConfigEntry as RenaultConfigEntry
 from .entity import RenaultDataEntity as RenaultDataEntity, RenaultDataEntityDescription as RenaultDataEntityDescription
-from .renault_hub import RenaultHub as RenaultHub
 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
 from renault_api.kamereon.models import KamereonVehicleLocationData
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RenaultDeviceTracker(RenaultDataEntity[KamereonVehicleLocationData], TrackerEntity):
     @property

homeassistant-stubs/components/renault/diagnostics.pyi

@@ -1,9 +1,8 @@
-from . import RenaultHub as RenaultHub
-from .const import CONF_KAMEREON_ACCOUNT_ID as CONF_KAMEREON_ACCOUNT_ID, DOMAIN as DOMAIN
+from . import RenaultConfigEntry as RenaultConfigEntry
+from .const import CONF_KAMEREON_ACCOUNT_ID as CONF_KAMEREON_ACCOUNT_ID
 from .renault_vehicle import RenaultVehicleProxy as RenaultVehicleProxy
 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_PASSWORD as CONF_PASSWORD, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
@@ -11,6 +10,6 @@ from typing import Any
 
 TO_REDACT: 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]: ...
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: RenaultConfigEntry) -> dict[str, Any]: ...
+async def async_get_device_diagnostics(hass: HomeAssistant, entry: RenaultConfigEntry, device: DeviceEntry) -> dict[str, Any]: ...
 def _get_vehicle_diagnostics(vehicle: RenaultVehicleProxy) -> dict[str, Any]: ...

homeassistant-stubs/components/renault/renault_vehicle.pyi

@@ -9,13 +9,11 @@ from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from renault_api.kamereon import models as models
 from renault_api.renault_vehicle import RenaultVehicle as RenaultVehicle
-from typing import Any, Concatenate, ParamSpec, TypeVar
+from typing import Any, Concatenate
 
 LOGGER: Incomplete
-_T = TypeVar('_T')
-_P = ParamSpec('_P')
 
-def with_error_wrapping(func: Callable[Concatenate[RenaultVehicleProxy, _P], Awaitable[_T]]) -> Callable[Concatenate[RenaultVehicleProxy, _P], Coroutine[Any, Any, _T]]: ...
+def with_error_wrapping(func: Callable[Concatenate[RenaultVehicleProxy, _P], Awaitable[_R]]) -> Callable[Concatenate[RenaultVehicleProxy, _P], Coroutine[Any, Any, _R]]: ...
 
 @dataclass
 class RenaultCoordinatorDescription:

homeassistant-stubs/components/renault/select.pyi

@@ -1,9 +1,7 @@
-from .const import DOMAIN as DOMAIN
+from . import RenaultConfigEntry as RenaultConfigEntry
 from .entity import RenaultDataEntity as RenaultDataEntity, RenaultDataEntityDescription as RenaultDataEntityDescription
-from .renault_hub import RenaultHub as RenaultHub
 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.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import StateType as StateType
@@ -14,7 +12,7 @@ class RenaultSelectEntityDescription(SelectEntityDescription, RenaultDataEntityD
     data_key: str
     def __init__(self, coordinator, *, 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, data_key) -> None: ...
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RenaultSelectEntity(RenaultDataEntity[KamereonVehicleBatteryStatusData], SelectEntity):
     entity_description: RenaultSelectEntityDescription