homeassistant-stubs 2024.9.2__py3-none-any.whl → 2024.10.0__py3-none-any.whl

homeassistant-stubs/components/recorder/migration.pyi

@@ -6,9 +6,9 @@ from .db_schema import BIG_INTEGER_SQL as BIG_INTEGER_SQL, Base as Base, CONTEXT
 from .models import 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, 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_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 get_start_time as get_start_time
-from .tasks import CommitTask as CommitTask, EntityIDPostMigrationTask as EntityIDPostMigrationTask, PostSchemaMigrationTask as PostSchemaMigrationTask, RecorderTask as RecorderTask, StatisticsTimestampMigrationCleanupTask as StatisticsTimestampMigrationCleanupTask
-from .util import database_job_retry_wrapper as database_job_retry_wrapper, execute_stmt_lambda_element as execute_stmt_lambda_element, get_index_by_name as get_index_by_name, retryable_database_job as retryable_database_job, session_scope as session_scope
+from .statistics import cleanup_statistics_timestamp_migration as cleanup_statistics_timestamp_migration, get_start_time as get_start_time
+from .tasks import RecorderTask as RecorderTask
+from .util import database_job_retry_wrapper as database_job_retry_wrapper, execute_stmt_lambda_element as execute_stmt_lambda_element, get_index_by_name as get_index_by_name, retryable_database_job as retryable_database_job, retryable_database_job_method as retryable_database_job_method, session_scope as session_scope
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
 from collections.abc import Callable as Callable, Iterable
@@ -24,6 +24,8 @@ from typing import Any
 
 LIVE_MIGRATION_MIN_SCHEMA_VERSION: int
 MIGRATION_NOTE_OFFLINE: str
+MIGRATION_NOTE_MINUTES: str
+MIGRATION_NOTE_WHILE: str
 _EMPTY_ENTITY_ID: str
 _EMPTY_EVENT_TYPE: str
 _LOGGER: Incomplete
@@ -233,8 +235,6 @@ class _SchemaVersion47Migrator(_SchemaVersionMigrator, target_version=47):
 
 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 post_schema_migration(instance: Recorder, old_version: int, new_version: int) -> None: ...
-def _wipe_old_string_statistics_columns(instance: Recorder) -> None: ...
 def _wipe_old_string_time_columns(instance: Recorder, engine: Engine, session: Session) -> None: ...
 def _migrate_columns_to_timestamp(instance: Recorder, session_maker: Callable[[], Session], engine: Engine) -> None: ...
 def _migrate_statistics_columns_to_timestamp_one_by_one(instance: Recorder, session_maker: Callable[[], Session]) -> None: ...
@@ -242,7 +242,6 @@ def _migrate_statistics_columns_to_timestamp(instance: Recorder, session_maker:
 def _context_id_to_bytes(context_id: str | None) -> bytes | None: ...
 def _generate_ulid_bytes_at_time(timestamp: float | None) -> bytes: ...
 def post_migrate_entity_ids(instance: Recorder) -> bool: ...
-def cleanup_legacy_states_event_ids(instance: Recorder) -> bool: ...
 def _initialize_database(session: Session) -> bool: ...
 def initialize_database(session_maker: Callable[[], Session]) -> bool: ...
 
@@ -259,12 +258,13 @@ class CommitBeforeMigrationTask(MigrationTask):
     def __init__(self, migrator) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)
-class NeedsMigrateResult:
+class DataMigrationStatus:
     needs_migrate: bool
     migration_done: bool
     def __init__(self, *, needs_migrate, migration_done) -> None: ...
 
 class BaseRunTimeMigration(ABC, metaclass=abc.ABCMeta):
+    index_to_drop: tuple[str, str] | None
     required_schema_version: int
     migration_version: int
     migration_id: str
@@ -273,60 +273,63 @@ class BaseRunTimeMigration(ABC, metaclass=abc.ABCMeta):
     migration_changes: Incomplete
     def __init__(self, schema_version: int, migration_changes: dict[str, int]) -> None: ...
     def do_migrate(self, instance: Recorder, session: Session) -> None: ...
-    @staticmethod
+    def migrate_data(self, instance: Recorder) -> bool: ...
     @abstractmethod
-    def migrate_data(instance: Recorder) -> bool: ...
-    def migration_done(self, instance: Recorder, session: Session | None) -> None: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
+    def migration_done(self, instance: Recorder, session: Session) -> None: ...
     @abstractmethod
-    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> DataMigrationStatus: ...
     def needs_migrate(self, instance: Recorder, session: Session) -> bool: ...
 
 class BaseRunTimeMigrationWithQuery(BaseRunTimeMigration, metaclass=abc.ABCMeta):
     @abstractmethod
     def needs_migrate_query(self) -> StatementLambdaElement: ...
-    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> DataMigrationStatus: ...
 
 class StatesContextIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = CONTEXT_ID_AS_BINARY_SCHEMA_VERSION
     migration_id: str
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
+    migration_version: int
+    index_to_drop: Incomplete
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EventsContextIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = CONTEXT_ID_AS_BINARY_SCHEMA_VERSION
     migration_id: str
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
+    migration_version: int
+    index_to_drop: Incomplete
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EventTypeIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = EVENT_TYPE_IDS_SCHEMA_VERSION
     migration_id: str
     task = CommitBeforeMigrationTask
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
-    def migration_done(self, instance: Recorder, session: Session | None) -> None: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
+    def migration_done(self, instance: Recorder, session: Session) -> None: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EntityIDMigration(BaseRunTimeMigrationWithQuery):
     required_schema_version = STATES_META_SCHEMA_VERSION
     migration_id: str
     task = CommitBeforeMigrationTask
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
-    def migration_done(self, instance: Recorder, _session: Session | None) -> None: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
+    def migration_done(self, instance: Recorder, session: Session) -> None: ...
     def needs_migrate_query(self) -> StatementLambdaElement: ...
 
 class EventIDPostMigration(BaseRunTimeMigration):
     migration_id: str
     task = MigrationTask
     migration_version: int
-    @staticmethod
-    def migrate_data(instance: Recorder) -> bool: ...
+    def migrate_data_impl(self, instance: Recorder) -> DataMigrationStatus: ...
     @staticmethod
     def _legacy_event_id_foreign_key_exists(instance: Recorder) -> bool: ...
-    def needs_migrate_impl(self, instance: Recorder, session: Session) -> NeedsMigrateResult: ...
+    def needs_migrate_impl(self, instance: Recorder, session: Session) -> DataMigrationStatus: ...
+
+@dataclass(slots=True)
+class EntityIDPostMigrationTask(RecorderTask):
+    def run(self, instance: Recorder) -> None: ...
 
 def _mark_migration_done(session: Session, migration: type[BaseRunTimeMigration]) -> None: ...
 def rebuild_sqlite_table(session_maker: Callable[[], Session], engine: Engine, table: type[Base]) -> bool: ...

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

@@ -3,6 +3,6 @@ from .database import DatabaseEngine as DatabaseEngine, DatabaseOptimizer as Dat
 from .event import extract_event_type_ids as extract_event_type_ids
 from .state import LazyState as LazyState, extract_metadata_ids as extract_metadata_ids, row_to_compressed_state as row_to_compressed_state
 from .statistics import CalendarStatisticPeriod as CalendarStatisticPeriod, FixedStatisticPeriod as FixedStatisticPeriod, RollingWindowStatisticPeriod as RollingWindowStatisticPeriod, StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMetaData as StatisticMetaData, StatisticPeriod as StatisticPeriod, StatisticResult as StatisticResult
-from .time import datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_datetime_to_timestamp as process_datetime_to_timestamp, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat, timestamp_to_datetime_or_none as timestamp_to_datetime_or_none
+from .time import datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat, timestamp_to_datetime_or_none as timestamp_to_datetime_or_none
 
-__all__ = ['CalendarStatisticPeriod', 'DatabaseEngine', 'DatabaseOptimizer', 'FixedStatisticPeriod', 'LazyState', 'RollingWindowStatisticPeriod', 'StatisticData', 'StatisticDataTimestamp', 'StatisticMetaData', 'StatisticPeriod', 'StatisticResult', 'UnsupportedDialect', 'bytes_to_ulid_or_none', 'bytes_to_uuid_hex_or_none', 'datetime_to_timestamp_or_none', 'extract_event_type_ids', 'extract_metadata_ids', 'process_datetime_to_timestamp', 'process_timestamp', 'process_timestamp_to_utc_isoformat', 'row_to_compressed_state', 'timestamp_to_datetime_or_none', 'ulid_to_bytes_or_none', 'uuid_hex_to_bytes_or_none']
+__all__ = ['CalendarStatisticPeriod', 'DatabaseEngine', 'DatabaseOptimizer', 'FixedStatisticPeriod', 'LazyState', 'RollingWindowStatisticPeriod', 'StatisticData', 'StatisticDataTimestamp', 'StatisticMetaData', 'StatisticPeriod', 'StatisticResult', 'UnsupportedDialect', 'bytes_to_ulid_or_none', 'bytes_to_uuid_hex_or_none', 'datetime_to_timestamp_or_none', 'extract_event_type_ids', 'extract_metadata_ids', 'process_timestamp', 'process_timestamp_to_utc_isoformat', 'row_to_compressed_state', 'timestamp_to_datetime_or_none', 'ulid_to_bytes_or_none', 'uuid_hex_to_bytes_or_none']

homeassistant-stubs/components/recorder/models/legacy.pyi

@@ -1,5 +1,5 @@
 from .state_attributes import decode_attributes_from_source as decode_attributes_from_source
-from .time import process_datetime_to_timestamp as process_datetime_to_timestamp, process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat
+from .time import process_timestamp as process_timestamp
 from _typeshed import Incomplete
 from datetime import datetime
 from homeassistant.const import COMPRESSED_STATE_ATTRIBUTES as COMPRESSED_STATE_ATTRIBUTES, COMPRESSED_STATE_LAST_CHANGED as COMPRESSED_STATE_LAST_CHANGED, COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
@@ -7,42 +7,6 @@ from homeassistant.core import Context as Context, State as State
 from sqlalchemy.engine.row import Row as Row
 from typing import Any
 
-class LegacyLazyStatePreSchema31(State):
-    __slots__: Incomplete
-    _row: Incomplete
-    entity_id: Incomplete
-    state: Incomplete
-    _attributes: Incomplete
-    _last_changed: Incomplete
-    _last_reported: Incomplete
-    _last_updated: Incomplete
-    _context: Incomplete
-    attr_cache: Incomplete
-    def __init__(self, row: Row, attr_cache: dict[str, dict[str, Any]], start_time: datetime | None) -> None: ...
-    @property
-    def attributes(self) -> dict[str, Any]: ...
-    @attributes.setter
-    def attributes(self, value: dict[str, Any]) -> None: ...
-    @property
-    def context(self) -> Context: ...
-    @context.setter
-    def context(self, value: Context) -> None: ...
-    @property
-    def last_changed(self) -> datetime: ...
-    @last_changed.setter
-    def last_changed(self, value: datetime) -> None: ...
-    @property
-    def last_reported(self) -> datetime: ...
-    @last_reported.setter
-    def last_reported(self, value: datetime) -> None: ...
-    @property
-    def last_updated(self) -> datetime: ...
-    @last_updated.setter
-    def last_updated(self, value: datetime) -> None: ...
-    def as_dict(self) -> dict[str, Any]: ...
-
-def legacy_row_to_compressed_state_pre_schema_31(row: Row, attr_cache: dict[str, dict[str, Any]], start_time: datetime | None) -> dict[str, Any]: ...
-
 class LegacyLazyState(State):
     __slots__: Incomplete
     _row: Incomplete

homeassistant-stubs/components/recorder/models/time.pyi

@@ -14,6 +14,5 @@ def process_timestamp(ts: datetime) -> datetime: ...
 def process_timestamp_to_utc_isoformat(ts: None) -> None: ...
 @overload
 def process_timestamp_to_utc_isoformat(ts: datetime) -> str: ...
-def process_datetime_to_timestamp(ts: datetime) -> float: ...
 def datetime_to_timestamp_or_none(dt: datetime | None) -> float | None: ...
 def timestamp_to_datetime_or_none(ts: float | None) -> datetime | None: ...

homeassistant-stubs/components/recorder/statistics.pyi

@@ -1,6 +1,6 @@
 import dataclasses
 from . import Recorder as Recorder
-from .const import DOMAIN as DOMAIN, EVENT_RECORDER_5MIN_STATISTICS_GENERATED as EVENT_RECORDER_5MIN_STATISTICS_GENERATED, EVENT_RECORDER_HOURLY_STATISTICS_GENERATED as EVENT_RECORDER_HOURLY_STATISTICS_GENERATED, INTEGRATION_PLATFORM_COMPILE_STATISTICS as INTEGRATION_PLATFORM_COMPILE_STATISTICS, INTEGRATION_PLATFORM_LIST_STATISTIC_IDS as INTEGRATION_PLATFORM_LIST_STATISTIC_IDS, INTEGRATION_PLATFORM_VALIDATE_STATISTICS as INTEGRATION_PLATFORM_VALIDATE_STATISTICS, SupportedDialect as SupportedDialect
+from .const import DOMAIN as DOMAIN, EVENT_RECORDER_5MIN_STATISTICS_GENERATED as EVENT_RECORDER_5MIN_STATISTICS_GENERATED, EVENT_RECORDER_HOURLY_STATISTICS_GENERATED as EVENT_RECORDER_HOURLY_STATISTICS_GENERATED, INTEGRATION_PLATFORM_COMPILE_STATISTICS as INTEGRATION_PLATFORM_COMPILE_STATISTICS, INTEGRATION_PLATFORM_LIST_STATISTIC_IDS as INTEGRATION_PLATFORM_LIST_STATISTIC_IDS, INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES as INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES, INTEGRATION_PLATFORM_VALIDATE_STATISTICS as INTEGRATION_PLATFORM_VALIDATE_STATISTICS, SupportedDialect as SupportedDialect
 from .db_schema import STATISTICS_TABLES as STATISTICS_TABLES, Statistics as Statistics, StatisticsBase as StatisticsBase, StatisticsRuns as StatisticsRuns, StatisticsShortTerm as StatisticsShortTerm
 from .models import StatisticData as StatisticData, StatisticDataTimestamp as StatisticDataTimestamp, StatisticMetaData as StatisticMetaData, StatisticResult as StatisticResult, datetime_to_timestamp_or_none as datetime_to_timestamp_or_none, process_timestamp as process_timestamp
 from .util import execute as execute, execute_stmt_lambda_element as execute_stmt_lambda_element, filter_unique_constraint_integrity_error as filter_unique_constraint_integrity_error, get_instance as get_instance, retryable_database_job as retryable_database_job, session_scope as session_scope
@@ -138,6 +138,7 @@ def _build_stats(db_rows: list[Row], table_duration_seconds: float, start_ts_idx
 def _build_converted_stats(db_rows: list[Row], table_duration_seconds: float, start_ts_idx: int, row_mapping: tuple[tuple[str, int], ...], convert: Callable[[float | None], float | None] | Callable[[float], float]) -> list[StatisticsRow]: ...
 def _sorted_statistics_to_dict(hass: HomeAssistant, stats: Sequence[Row[Any]], statistic_ids: set[str] | None, _metadata: dict[str, tuple[int, StatisticMetaData]], convert_units: bool, table: type[StatisticsBase], units: dict[str, str] | None, types: set[Literal['last_reset', 'max', 'mean', 'min', 'state', 'sum']]) -> dict[str, list[StatisticsRow]]: ...
 def validate_statistics(hass: HomeAssistant) -> dict[str, list[ValidationIssue]]: ...
+def update_statistics_issues(hass: HomeAssistant) -> None: ...
 def _statistics_exists(session: Session, table: type[StatisticsBase], metadata_id: int, start: datetime) -> int | None: ...
 def _async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...
 def async_import_statistics(hass: HomeAssistant, metadata: StatisticMetaData, statistics: Iterable[StatisticData]) -> None: ...

homeassistant-stubs/components/recorder/tasks.pyi

@@ -33,17 +33,19 @@ class ChangeStatisticsUnitTask(RecorderTask):
 
 @dataclass(slots=True)
 class ClearStatisticsTask(RecorderTask):
+    on_done: Callable[[], None] | None
     statistic_ids: list[str]
     def run(self, instance: Recorder) -> None: ...
-    def __init__(self, statistic_ids) -> None: ...
+    def __init__(self, on_done, statistic_ids) -> None: ...
 
 @dataclass(slots=True)
 class UpdateStatisticsMetadataTask(RecorderTask):
+    on_done: Callable[[], None] | None
     statistic_id: str
     new_statistic_id: str | None | UndefinedType
     new_unit_of_measurement: str | None | UndefinedType
     def run(self, instance: Recorder) -> None: ...
-    def __init__(self, statistic_id, new_statistic_id, new_unit_of_measurement) -> None: ...
+    def __init__(self, on_done, statistic_id, new_statistic_id, new_unit_of_measurement) -> None: ...
 
 @dataclass(slots=True)
 class UpdateStatesMetadataTask(RecorderTask):
@@ -141,26 +143,11 @@ class SynchronizeTask(RecorderTask):
     def run(self, instance: Recorder) -> None: ...
     def __init__(self, event) -> None: ...
 
-@dataclass(slots=True)
-class PostSchemaMigrationTask(RecorderTask):
-    old_version: int
-    new_version: int
-    def run(self, instance: Recorder) -> None: ...
-    def __init__(self, old_version, new_version) -> None: ...
-
-@dataclass(slots=True)
-class StatisticsTimestampMigrationCleanupTask(RecorderTask):
-    def run(self, instance: Recorder) -> None: ...
-
 @dataclass(slots=True)
 class AdjustLRUSizeTask(RecorderTask):
     commit_before = ...
     def run(self, instance: Recorder) -> None: ...
 
-@dataclass(slots=True)
-class EntityIDPostMigrationTask(RecorderTask):
-    def run(self, instance: Recorder) -> None: ...
-
 @dataclass(slots=True)
 class RefreshEventTypesTask(RecorderTask):
     event_types: list[EventType[Any] | str]

homeassistant-stubs/components/recorder/util.pyi

@@ -74,9 +74,13 @@ def end_incomplete_runs(session: Session, start_time: datetime) -> None: ...
 def _is_retryable_error(instance: Recorder, err: OperationalError) -> bool: ...
 
 _FuncType: Incomplete
+_MethType: Incomplete
+_FuncOrMethType: Incomplete
 
-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 retryable_database_job(description: str) -> Callable[[_FuncType[_P, bool]], _FuncType[_P, bool]]: ...
+def retryable_database_job_method(description: str) -> Callable[[_MethType[_Self, _P, bool]], _MethType[_Self, _P, bool]]: ...
+def _wrap_func_or_meth(job: _FuncOrMethType[_P, bool], description: str, method: bool) -> _FuncOrMethType[_P, bool]: ...
+def database_job_retry_wrapper(description: str, attempts: int = 5) -> Callable[[_FuncType[_P, None]], _FuncType[_P, None]]: ...
 def periodic_db_cleanups(instance: Recorder) -> None: ...
 def write_lock_db_sqlite(instance: Recorder) -> Generator[None]: ...
 def async_migration_in_progress(hass: HomeAssistant) -> bool: ...

homeassistant-stubs/components/recorder/websocket_api.pyi

@@ -1,5 +1,5 @@
 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 .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, update_statistics_issues as update_statistics_issues, validate_statistics as validate_statistics
 from .util import PERIOD_SCHEMA as PERIOD_SCHEMA, get_instance as get_instance, resolve_period as resolve_period
 from _typeshed import Incomplete
 from datetime import datetime as dt
@@ -11,6 +11,8 @@ from homeassistant.helpers.json import json_bytes as json_bytes
 from homeassistant.util.unit_conversion import ConductivityConverter as ConductivityConverter, DataRateConverter as DataRateConverter, DistanceConverter as DistanceConverter, DurationConverter as DurationConverter, ElectricCurrentConverter as ElectricCurrentConverter, ElectricPotentialConverter as ElectricPotentialConverter, EnergyConverter as EnergyConverter, InformationConverter as InformationConverter, MassConverter as MassConverter, PowerConverter as PowerConverter, PressureConverter as PressureConverter, SpeedConverter as SpeedConverter, TemperatureConverter as TemperatureConverter, UnitlessRatioConverter as UnitlessRatioConverter, VolumeConverter as VolumeConverter, VolumeFlowRateConverter as VolumeFlowRateConverter
 from typing import Any, Literal
 
+CLEAR_STATISTICS_TIME_OUT: int
+UPDATE_STATISTICS_METADATA_TIME_OUT: int
 UNIT_SCHEMA: Incomplete
 
 def async_setup(hass: HomeAssistant) -> None: ...
@@ -23,9 +25,10 @@ def _ws_get_list_statistic_ids(hass: HomeAssistant, msg_id: int, statistic_type:
 async def ws_handle_list_statistic_ids(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict) -> None: ...
 async def ws_list_statistic_ids(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def ws_validate_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
-def ws_clear_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def ws_update_statistics_issues(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def ws_clear_statistics(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 async def ws_get_statistics_metadata(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
-def ws_update_statistics_metadata(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
+async def ws_update_statistics_metadata(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 def ws_change_statistics_unit(hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]) -> None: ...
 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: ...

homeassistant-stubs/components/remote/__init__.pyi

@@ -10,10 +10,12 @@ from homeassistant.helpers.entity import ToggleEntity as ToggleEntity, ToggleEnt
 from homeassistant.helpers.entity_component import EntityComponent as EntityComponent
 from homeassistant.helpers.typing import ConfigType as ConfigType
 from homeassistant.loader import bind_hass as bind_hass
+from homeassistant.util.hass_dict import HassKey as HassKey
 from typing import Any
 
 _LOGGER: Incomplete
 DOMAIN: str
+DATA_COMPONENT: HassKey[EntityComponent[RemoteEntity]]
 ENTITY_ID_FORMAT: Incomplete
 PLATFORM_SCHEMA: Incomplete
 PLATFORM_SCHEMA_BASE: Incomplete

homeassistant-stubs/components/renault/device_tracker.pyi

@@ -1,18 +1,22 @@
 from . import RenaultConfigEntry as RenaultConfigEntry
 from .entity import RenaultDataEntity as RenaultDataEntity, RenaultDataEntityDescription as RenaultDataEntityDescription
-from homeassistant.components.device_tracker import SourceType as SourceType, TrackerEntity as TrackerEntity
+from dataclasses import dataclass
+from homeassistant.components.device_tracker import TrackerEntity as TrackerEntity, TrackerEntityDescription as TrackerEntityDescription
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from renault_api.kamereon.models import KamereonVehicleLocationData
 
+@dataclass(frozen=True, kw_only=True)
+class RenaultTrackerEntityDescription(TrackerEntityDescription, RenaultDataEntityDescription):
+    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=...) -> None: ...
+
 async def async_setup_entry(hass: HomeAssistant, config_entry: RenaultConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class RenaultDeviceTracker(RenaultDataEntity[KamereonVehicleLocationData], TrackerEntity):
+    entity_description: RenaultTrackerEntityDescription
     @property
     def latitude(self) -> float | None: ...
     @property
     def longitude(self) -> float | None: ...
-    @property
-    def source_type(self) -> SourceType: ...
 
-DEVICE_TRACKER_TYPES: tuple[RenaultDataEntityDescription, ...]
+DEVICE_TRACKER_TYPES: tuple[RenaultTrackerEntityDescription, ...]

homeassistant-stubs/components/renault/renault_hub.pyi

@@ -3,7 +3,7 @@ from .renault_vehicle import RenaultVehicleProxy as RenaultVehicleProxy
 from _typeshed import Incomplete
 from datetime import timedelta
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import ATTR_IDENTIFIERS as ATTR_IDENTIFIERS, ATTR_MANUFACTURER as ATTR_MANUFACTURER, ATTR_MODEL as ATTR_MODEL, ATTR_NAME as ATTR_NAME, ATTR_SW_VERSION as ATTR_SW_VERSION
+from homeassistant.const import ATTR_IDENTIFIERS as ATTR_IDENTIFIERS, ATTR_MANUFACTURER as ATTR_MANUFACTURER, ATTR_MODEL as ATTR_MODEL, ATTR_MODEL_ID as ATTR_MODEL_ID, ATTR_NAME as ATTR_NAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers import device_registry as dr

homeassistant-stubs/components/renault/renault_vehicle.pyi

@@ -42,6 +42,8 @@ class RenaultVehicleProxy:
     async def set_charge_stop(self) -> models.KamereonVehicleChargingStartActionData: ...
     async def set_ac_stop(self) -> models.KamereonVehicleHvacStartActionData: ...
     async def set_ac_start(self, temperature: float, when: datetime | None = None) -> models.KamereonVehicleHvacStartActionData: ...
+    async def get_hvac_settings(self) -> models.KamereonVehicleHvacSettingsData: ...
+    async def set_hvac_schedules(self, schedules: list[models.HvacSchedule]) -> models.KamereonVehicleHvacScheduleActionData: ...
     async def get_charging_settings(self) -> models.KamereonVehicleChargingSettingsData: ...
     async def set_charge_schedules(self, schedules: list[models.ChargeSchedule]) -> models.KamereonVehicleChargeScheduleActionData: ...
 

homeassistant-stubs/components/renault/services.pyi

@@ -15,9 +15,13 @@ SERVICE_AC_START_SCHEMA: Incomplete
 SERVICE_CHARGE_SET_SCHEDULE_DAY_SCHEMA: Incomplete
 SERVICE_CHARGE_SET_SCHEDULE_SCHEMA: Incomplete
 SERVICE_CHARGE_SET_SCHEDULES_SCHEMA: Incomplete
+SERVICE_AC_SET_SCHEDULE_DAY_SCHEMA: Incomplete
+SERVICE_AC_SET_SCHEDULE_SCHEMA: Incomplete
+SERVICE_AC_SET_SCHEDULES_SCHEMA: Incomplete
 SERVICE_AC_CANCEL: str
 SERVICE_AC_START: str
 SERVICE_CHARGE_SET_SCHEDULES: str
+SERVICE_AC_SET_SCHEDULES: str
 SERVICES: Incomplete
 
 def setup_services(hass: HomeAssistant) -> None: ...

homeassistant-stubs/components/rfxtrx/__init__.pyi

@@ -1,5 +1,5 @@
 import RFXtrx as rfxtrxmod
-from .const import ATTR_EVENT as ATTR_EVENT, COMMAND_GROUP_LIST as COMMAND_GROUP_LIST, CONF_AUTOMATIC_ADD as CONF_AUTOMATIC_ADD, CONF_DATA_BITS as CONF_DATA_BITS, CONF_PROTOCOLS as CONF_PROTOCOLS, DATA_RFXOBJECT as DATA_RFXOBJECT, DEVICE_PACKET_TYPE_LIGHTING4 as DEVICE_PACKET_TYPE_LIGHTING4, DOMAIN as DOMAIN, EVENT_RFXTRX_EVENT as EVENT_RFXTRX_EVENT, SERVICE_SEND as SERVICE_SEND
+from .const import ATTR_EVENT as ATTR_EVENT, CONF_AUTOMATIC_ADD as CONF_AUTOMATIC_ADD, CONF_DATA_BITS as CONF_DATA_BITS, CONF_PROTOCOLS as CONF_PROTOCOLS, DATA_RFXOBJECT as DATA_RFXOBJECT, DEVICE_PACKET_TYPE_LIGHTING4 as DEVICE_PACKET_TYPE_LIGHTING4, DOMAIN as DOMAIN, EVENT_RFXTRX_EVENT as EVENT_RFXTRX_EVENT, SERVICE_SEND as SERVICE_SEND, SIGNAL_EVENT as SIGNAL_EVENT
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Mapping
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
@@ -7,15 +7,13 @@ from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_DEVICE as
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, ServiceCall as ServiceCall, callback as callback
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers import device_registry as dr
-from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo, EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
+from homeassistant.helpers.device_registry import EventDeviceRegistryUpdatedData as EventDeviceRegistryUpdatedData
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect, async_dispatcher_send as async_dispatcher_send
 from homeassistant.helpers.entity import Entity as Entity
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
 from typing import Any, NamedTuple
 
 DEFAULT_OFF_DELAY: float
-SIGNAL_EVENT: Incomplete
 CONNECT_TIMEOUT: float
 _LOGGER: Incomplete
 
@@ -42,27 +40,4 @@ def get_device_data_bits(device: rfxtrxmod.RFXtrxDevice, devices: dict[DeviceTup
 def find_possible_pt2262_device(device_ids: set[str], device_id: str) -> str | None: ...
 def get_device_id(device: rfxtrxmod.RFXtrxDevice, data_bits: int | None = None) -> DeviceTuple: ...
 def get_device_tuple_from_identifiers(identifiers: set[tuple[str, str]]) -> DeviceTuple | None: ...
-def get_identifiers_from_device_tuple(device_tuple: DeviceTuple) -> set[tuple[str, str]]: ...
 async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: ConfigEntry, device_entry: dr.DeviceEntry) -> bool: ...
-
-class RfxtrxEntity(RestoreEntity):
-    _attr_assumed_state: bool
-    _attr_has_entity_name: bool
-    _attr_should_poll: bool
-    _device: rfxtrxmod.RFXtrxDevice
-    _event: rfxtrxmod.RFXtrxEvent | None
-    _attr_device_info: Incomplete
-    _attr_unique_id: Incomplete
-    _device_id: Incomplete
-    def __init__(self, device: rfxtrxmod.RFXtrxDevice, device_id: DeviceTuple, event: rfxtrxmod.RFXtrxEvent | None = None) -> None: ...
-    async def async_added_to_hass(self) -> None: ...
-    @property
-    def extra_state_attributes(self) -> dict[str, str] | None: ...
-    def _event_applies(self, event: rfxtrxmod.RFXtrxEvent, device_id: DeviceTuple) -> bool: ...
-    def _apply_event(self, event: rfxtrxmod.RFXtrxEvent) -> None: ...
-    def _handle_event(self, event: rfxtrxmod.RFXtrxEvent, device_id: DeviceTuple) -> None: ...
-
-class RfxtrxCommandEntity(RfxtrxEntity):
-    _attr_name: Incomplete
-    def __init__(self, device: rfxtrxmod.RFXtrxDevice, device_id: DeviceTuple, event: rfxtrxmod.RFXtrxEvent | None = None) -> None: ...
-    async def _async_send(self, fun: Callable[[rfxtrxmod.PySerialTransport, Unpack[_Ts]], None], *args: Unpack[_Ts]) -> None: ...

homeassistant-stubs/components/rfxtrx/binary_sensor.pyi

@@ -1,6 +1,7 @@
 import RFXtrx as rfxtrxmod
-from . import DeviceTuple as DeviceTuple, RfxtrxEntity as RfxtrxEntity, async_setup_platform_entry as async_setup_platform_entry, get_pt2262_cmd as get_pt2262_cmd
+from . import DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry, get_pt2262_cmd as get_pt2262_cmd
 from .const import COMMAND_OFF_LIST as COMMAND_OFF_LIST, COMMAND_ON_LIST as COMMAND_ON_LIST, CONF_DATA_BITS as CONF_DATA_BITS, CONF_OFF_DELAY as CONF_OFF_DELAY, DEVICE_PACKET_TYPE_LIGHTING4 as DEVICE_PACKET_TYPE_LIGHTING4
+from .entity import RfxtrxEntity as RfxtrxEntity
 from _typeshed import Incomplete
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/rfxtrx/const.pyi

@@ -18,3 +18,4 @@ DEVICE_PACKET_TYPE_LIGHTING4: int
 EVENT_RFXTRX_EVENT: str
 DATA_RFXOBJECT: str
 DOMAIN: str
+SIGNAL_EVENT: Incomplete

homeassistant-stubs/components/rfxtrx/cover.pyi

@@ -1,6 +1,7 @@
 import RFXtrx as rfxtrxmod
-from . import DeviceTuple as DeviceTuple, RfxtrxCommandEntity as RfxtrxCommandEntity, async_setup_platform_entry as async_setup_platform_entry
+from . import DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry
 from .const import COMMAND_OFF_LIST as COMMAND_OFF_LIST, COMMAND_ON_LIST as COMMAND_ON_LIST, CONF_VENETIAN_BLIND_MODE as CONF_VENETIAN_BLIND_MODE, CONST_VENETIAN_BLIND_MODE_EU as CONST_VENETIAN_BLIND_MODE_EU, CONST_VENETIAN_BLIND_MODE_US as CONST_VENETIAN_BLIND_MODE_US
+from .entity import RfxtrxCommandEntity as RfxtrxCommandEntity
 from _typeshed import Incomplete
 from homeassistant.components.cover import CoverEntity as CoverEntity, CoverEntityFeature as CoverEntityFeature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/rfxtrx/device_action.pyi

@@ -2,7 +2,7 @@ from . import DATA_RFXOBJECT as DATA_RFXOBJECT, DOMAIN as DOMAIN
 from .helpers import async_get_device_object as async_get_device_object
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable
-from homeassistant.components.device_automation.exceptions import InvalidDeviceAutomationConfig as InvalidDeviceAutomationConfig
+from homeassistant.components.device_automation import InvalidDeviceAutomationConfig as InvalidDeviceAutomationConfig
 from homeassistant.const import CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_DOMAIN as CONF_DOMAIN, CONF_TYPE as CONF_TYPE
 from homeassistant.core import Context as Context, HomeAssistant as HomeAssistant
 from homeassistant.helpers.typing import ConfigType as ConfigType, TemplateVarsType as TemplateVarsType

homeassistant-stubs/components/rfxtrx/device_trigger.pyi

@@ -2,8 +2,7 @@ from . import DOMAIN as DOMAIN
 from .const import EVENT_RFXTRX_EVENT as EVENT_RFXTRX_EVENT
 from .helpers import async_get_device_object as async_get_device_object
 from _typeshed import Incomplete
-from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA as DEVICE_TRIGGER_BASE_SCHEMA
-from homeassistant.components.device_automation.exceptions import InvalidDeviceAutomationConfig as InvalidDeviceAutomationConfig
+from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA as DEVICE_TRIGGER_BASE_SCHEMA, InvalidDeviceAutomationConfig as InvalidDeviceAutomationConfig
 from homeassistant.const import ATTR_DEVICE_ID as ATTR_DEVICE_ID, CONF_DEVICE_ID as CONF_DEVICE_ID, CONF_DOMAIN as CONF_DOMAIN, CONF_PLATFORM as CONF_PLATFORM, CONF_TYPE as CONF_TYPE
 from homeassistant.core import CALLBACK_TYPE as CALLBACK_TYPE, HomeAssistant as HomeAssistant
 from homeassistant.helpers.trigger import TriggerActionType as TriggerActionType, TriggerInfo as TriggerInfo

homeassistant-stubs/components/rfxtrx/entity.pyi

@@ -0,0 +1,33 @@
+import RFXtrx as rfxtrxmod
+from . import DeviceTuple as DeviceTuple
+from .const import ATTR_EVENT as ATTR_EVENT, COMMAND_GROUP_LIST as COMMAND_GROUP_LIST, DATA_RFXOBJECT as DATA_RFXOBJECT, DOMAIN as DOMAIN, SIGNAL_EVENT as SIGNAL_EVENT
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from homeassistant.core import callback as callback
+from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
+from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
+from homeassistant.helpers.restore_state import RestoreEntity as RestoreEntity
+
+def _get_identifiers_from_device_tuple(device_tuple: DeviceTuple) -> set[tuple[str, str]]: ...
+
+class RfxtrxEntity(RestoreEntity):
+    _attr_assumed_state: bool
+    _attr_has_entity_name: bool
+    _attr_should_poll: bool
+    _device: rfxtrxmod.RFXtrxDevice
+    _event: rfxtrxmod.RFXtrxEvent | None
+    _attr_device_info: Incomplete
+    _attr_unique_id: Incomplete
+    _device_id: Incomplete
+    def __init__(self, device: rfxtrxmod.RFXtrxDevice, device_id: DeviceTuple, event: rfxtrxmod.RFXtrxEvent | None = None) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    @property
+    def extra_state_attributes(self) -> dict[str, str] | None: ...
+    def _event_applies(self, event: rfxtrxmod.RFXtrxEvent, device_id: DeviceTuple) -> bool: ...
+    def _apply_event(self, event: rfxtrxmod.RFXtrxEvent) -> None: ...
+    def _handle_event(self, event: rfxtrxmod.RFXtrxEvent, device_id: DeviceTuple) -> None: ...
+
+class RfxtrxCommandEntity(RfxtrxEntity):
+    _attr_name: Incomplete
+    def __init__(self, device: rfxtrxmod.RFXtrxDevice, device_id: DeviceTuple, event: rfxtrxmod.RFXtrxEvent | None = None) -> None: ...
+    async def _async_send(self, fun: Callable[[rfxtrxmod.PySerialTransport, Unpack[_Ts]], None], *args: Unpack[_Ts]) -> None: ...

homeassistant-stubs/components/rfxtrx/event.pyi

@@ -1,5 +1,6 @@
-from . import DeviceTuple as DeviceTuple, RfxtrxEntity as RfxtrxEntity, async_setup_platform_entry as async_setup_platform_entry
+from . import DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry
 from .const import DEVICE_PACKET_TYPE_LIGHTING4 as DEVICE_PACKET_TYPE_LIGHTING4
+from .entity import RfxtrxEntity as RfxtrxEntity
 from RFXtrx import RFXtrxDevice as RFXtrxDevice, RFXtrxEvent as RFXtrxEvent
 from _typeshed import Incomplete
 from homeassistant.components.event import EventEntity as EventEntity

homeassistant-stubs/components/rfxtrx/light.pyi

@@ -1,6 +1,7 @@
 import RFXtrx as rfxtrxmod
-from . import DeviceTuple as DeviceTuple, RfxtrxCommandEntity as RfxtrxCommandEntity, async_setup_platform_entry as async_setup_platform_entry
+from . import DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry
 from .const import COMMAND_OFF_LIST as COMMAND_OFF_LIST, COMMAND_ON_LIST as COMMAND_ON_LIST
+from .entity import RfxtrxCommandEntity as RfxtrxCommandEntity
 from _typeshed import Incomplete
 from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ColorMode as ColorMode, LightEntity as LightEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/rfxtrx/sensor.pyi

@@ -1,5 +1,6 @@
-from . import DeviceTuple as DeviceTuple, RfxtrxEntity as RfxtrxEntity, async_setup_platform_entry as async_setup_platform_entry, get_rfx_object as get_rfx_object
+from . import DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry, get_rfx_object as get_rfx_object
 from .const import ATTR_EVENT as ATTR_EVENT
+from .entity import RfxtrxEntity as RfxtrxEntity
 from RFXtrx import RFXtrxDevice as RFXtrxDevice, RFXtrxEvent as RFXtrxEvent
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable

homeassistant-stubs/components/rfxtrx/siren.pyi

@@ -1,6 +1,7 @@
 import RFXtrx as rfxtrxmod
-from . import DEFAULT_OFF_DELAY as DEFAULT_OFF_DELAY, DeviceTuple as DeviceTuple, RfxtrxCommandEntity as RfxtrxCommandEntity, async_setup_platform_entry as async_setup_platform_entry
+from . import DEFAULT_OFF_DELAY as DEFAULT_OFF_DELAY, DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry
 from .const import CONF_OFF_DELAY as CONF_OFF_DELAY
+from .entity import RfxtrxCommandEntity as RfxtrxCommandEntity
 from _typeshed import Incomplete
 from homeassistant.components.siren import ATTR_TONE as ATTR_TONE, SirenEntity as SirenEntity, SirenEntityFeature as SirenEntityFeature
 from homeassistant.config_entries import ConfigEntry as ConfigEntry

homeassistant-stubs/components/rfxtrx/switch.pyi

@@ -1,6 +1,7 @@
 import RFXtrx as rfxtrxmod
-from . import DOMAIN as DOMAIN, DeviceTuple as DeviceTuple, RfxtrxCommandEntity as RfxtrxCommandEntity, async_setup_platform_entry as async_setup_platform_entry, get_pt2262_cmd as get_pt2262_cmd
-from .const import COMMAND_OFF_LIST as COMMAND_OFF_LIST, COMMAND_ON_LIST as COMMAND_ON_LIST, CONF_DATA_BITS as CONF_DATA_BITS, DEVICE_PACKET_TYPE_LIGHTING4 as DEVICE_PACKET_TYPE_LIGHTING4
+from . import DeviceTuple as DeviceTuple, async_setup_platform_entry as async_setup_platform_entry, get_pt2262_cmd as get_pt2262_cmd
+from .const import COMMAND_OFF_LIST as COMMAND_OFF_LIST, COMMAND_ON_LIST as COMMAND_ON_LIST, CONF_DATA_BITS as CONF_DATA_BITS, DEVICE_PACKET_TYPE_LIGHTING4 as DEVICE_PACKET_TYPE_LIGHTING4, DOMAIN as DOMAIN
+from .entity import RfxtrxCommandEntity as RfxtrxCommandEntity
 from _typeshed import Incomplete
 from homeassistant.components.switch import SwitchEntity as SwitchEntity
 from homeassistant.config_entries import ConfigEntry as ConfigEntry