homeassistant-stubs 2024.9.3__py3-none-any.whl → 2024.10.0b0__py3-none-any.whl

homeassistant-stubs/components/recorder/const.pyi

@@ -22,15 +22,15 @@ ATTR_KEEP_DAYS: str
 ATTR_REPACK: str
 ATTR_APPLY_FILTER: str
 KEEPALIVE_TIME: int
-STATISTICS_ROWS_SCHEMA_VERSION: int
 CONTEXT_ID_AS_BINARY_SCHEMA_VERSION: int
 EVENT_TYPE_IDS_SCHEMA_VERSION: int
 STATES_META_SCHEMA_VERSION: int
 LAST_REPORTED_SCHEMA_VERSION: int
 LEGACY_STATES_EVENT_ID_INDEX_SCHEMA_VERSION: int
 INTEGRATION_PLATFORM_COMPILE_STATISTICS: str
-INTEGRATION_PLATFORM_VALIDATE_STATISTICS: str
 INTEGRATION_PLATFORM_LIST_STATISTIC_IDS: str
+INTEGRATION_PLATFORM_UPDATE_STATISTICS_ISSUES: str
+INTEGRATION_PLATFORM_VALIDATE_STATISTICS: str
 INTEGRATION_PLATFORM_METHODS: Incomplete
 
 class SupportedDialect(StrEnum):

homeassistant-stubs/components/recorder/core.pyi

@@ -1,7 +1,7 @@
 import asyncio
 import threading
 from . import migration as migration, statistics as statistics
-from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, 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_MAX_BIND_VARS as SQLITE_MAX_BIND_VARS, SQLITE_URL_PREFIX as SQLITE_URL_PREFIX, STATISTICS_ROWS_SCHEMA_VERSION as STATISTICS_ROWS_SCHEMA_VERSION, SupportedDialect as SupportedDialect
+from .const import DB_WORKER_PREFIX as DB_WORKER_PREFIX, 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_MAX_BIND_VARS as SQLITE_MAX_BIND_VARS, 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 .migration import EntityIDMigration as EntityIDMigration, EventIDPostMigration as EventIDPostMigration, EventTypeIDMigration as EventTypeIDMigration, EventsContextIDMigration as EventsContextIDMigration, StatesContextIDMigration as StatesContextIDMigration
@@ -87,7 +87,6 @@ class Recorder(threading.Thread):
     event_session: Incomplete
     _get_session: Incomplete
     _completed_first_database_setup: Incomplete
-    async_migration_event: Incomplete
     migration_in_progress: bool
     migration_is_live: bool
     use_legacy_events_index: bool
@@ -157,7 +156,6 @@ class Recorder(threading.Thread):
     def _guarded_process_one_task_or_event_or_recover(self, task: RecorderTask | Event) -> None: ...
     def _process_one_task_or_event_or_recover(self, task: RecorderTask | Event) -> None: ...
     def _setup_recorder(self) -> bool: ...
-    def _async_migration_started(self) -> None: ...
     def _migrate_schema_offline(self, schema_status: migration.SchemaValidationStatus) -> tuple[bool, migration.SchemaValidationStatus]: ...
     def _migrate_schema_live(self, schema_status: migration.SchemaValidationStatus) -> tuple[bool, migration.SchemaValidationStatus]: ...
     def _migrate_schema(self, schema_status: migration.SchemaValidationStatus, live: bool) -> tuple[bool, migration.SchemaValidationStatus]: ...
@@ -172,8 +170,6 @@ class Recorder(threading.Thread):
     def _close_event_session(self) -> None: ...
     def _reopen_event_session(self) -> None: ...
     def _open_event_session(self) -> None: ...
-    def _post_schema_migration(self, old_version: int, new_version: int) -> None: ...
-    def _post_migrate_entity_ids(self) -> bool: ...
     def _send_keep_alive(self) -> None: ...
     async def async_block_till_done(self) -> None: ...
     def block_till_done(self) -> None: ...

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

@@ -1,16 +1,15 @@
-from ... import recorder as recorder
 from ..db_schema import RecorderRuns as RecorderRuns, StateAttributes as StateAttributes, States as States
 from ..filters import Filters as Filters
-from ..models 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 ..models.legacy import LegacyLazyState as LegacyLazyState, LegacyLazyStatePreSchema31 as LegacyLazyStatePreSchema31, legacy_row_to_compressed_state as legacy_row_to_compressed_state, legacy_row_to_compressed_state_pre_schema_31 as legacy_row_to_compressed_state_pre_schema_31
+from ..models import process_timestamp as process_timestamp, process_timestamp_to_utc_isoformat as process_timestamp_to_utc_isoformat
+from ..models.legacy import LegacyLazyState as LegacyLazyState, legacy_row_to_compressed_state as legacy_row_to_compressed_state
 from ..util import execute_stmt_lambda_element as execute_stmt_lambda_element, session_scope as session_scope
-from .common import _schema_version as _schema_version
 from .const import LAST_CHANGED_KEY as LAST_CHANGED_KEY, NEED_ATTRIBUTE_DOMAINS as NEED_ATTRIBUTE_DOMAINS, SIGNIFICANT_DOMAINS as SIGNIFICANT_DOMAINS, SIGNIFICANT_DOMAINS_ENTITY_ID_LIKE as SIGNIFICANT_DOMAINS_ENTITY_ID_LIKE, STATE_KEY as STATE_KEY
 from _typeshed import Incomplete
 from collections.abc import Callable as Callable, Iterable
 from datetime import datetime
 from homeassistant.const import COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, split_entity_id as split_entity_id
+from homeassistant.helpers.recorder import get_instance as get_instance
 from sqlalchemy import Column as Column
 from sqlalchemy.engine.row import Row as Row
 from sqlalchemy.orm.session import Session as Session
@@ -34,16 +33,16 @@ _QUERY_STATES_NO_LAST_CHANGED: Incomplete
 _FIELD_MAP: Incomplete
 _FIELD_MAP_PRE_SCHEMA_31: Incomplete
 
-def _lambda_stmt_and_join_attributes(schema_version: int, no_attributes: bool, include_last_changed: bool = True) -> tuple[StatementLambdaElement, bool]: ...
+def _lambda_stmt_and_join_attributes(no_attributes: bool, include_last_changed: bool = True) -> tuple[StatementLambdaElement, bool]: ...
 def get_significant_states(hass: HomeAssistant, start_time: datetime, end_time: datetime | None = None, entity_ids: list[str] | None = None, filters: Filters | None = None, include_start_time_state: bool = True, significant_changes_only: bool = True, minimal_response: bool = False, no_attributes: bool = False, compressed_state_format: bool = False) -> dict[str, list[State | dict[str, Any]]]: ...
-def _significant_states_stmt(schema_version: int, start_time: datetime, end_time: datetime | None, entity_ids: list[str], significant_changes_only: bool, no_attributes: bool) -> StatementLambdaElement: ...
+def _significant_states_stmt(start_time: datetime, end_time: datetime | None, entity_ids: list[str], significant_changes_only: bool, no_attributes: bool) -> StatementLambdaElement: ...
 def get_significant_states_with_session(hass: HomeAssistant, session: Session, start_time: datetime, end_time: datetime | None = None, entity_ids: list[str] | None = None, filters: Filters | None = None, include_start_time_state: bool = True, significant_changes_only: bool = True, minimal_response: bool = False, no_attributes: bool = False, compressed_state_format: bool = False) -> dict[str, list[State | dict[str, Any]]]: ...
 def get_full_significant_states_with_session(hass: HomeAssistant, session: Session, start_time: datetime, end_time: datetime | None = None, entity_ids: list[str] | None = None, filters: Filters | None = None, include_start_time_state: bool = True, significant_changes_only: bool = True, no_attributes: bool = False) -> dict[str, list[State]]: ...
-def _state_changed_during_period_stmt(schema_version: int, start_time: datetime, end_time: datetime | None, entity_id: str, no_attributes: bool, descending: bool, limit: int | None) -> StatementLambdaElement: ...
+def _state_changed_during_period_stmt(start_time: datetime, end_time: datetime | None, entity_id: str, no_attributes: bool, descending: bool, limit: int | None) -> StatementLambdaElement: ...
 def state_changes_during_period(hass: HomeAssistant, start_time: datetime, end_time: datetime | None = None, entity_id: str | None = None, no_attributes: bool = False, descending: bool = False, limit: int | None = None, include_start_time_state: bool = True) -> dict[str, list[State]]: ...
-def _get_last_state_changes_stmt(schema_version: int, number_of_states: int, entity_id: str) -> StatementLambdaElement: ...
+def _get_last_state_changes_stmt(number_of_states: int, entity_id: str) -> StatementLambdaElement: ...
 def get_last_state_changes(hass: HomeAssistant, number_of_states: int, entity_id: str) -> dict[str, list[State]]: ...
-def _get_states_for_entities_stmt(schema_version: int, run_start: datetime, utc_point_in_time: datetime, entity_ids: list[str], no_attributes: bool) -> StatementLambdaElement: ...
+def _get_states_for_entities_stmt(run_start: datetime, utc_point_in_time: datetime, entity_ids: list[str], no_attributes: bool) -> StatementLambdaElement: ...
 def _get_rows_with_session(hass: HomeAssistant, session: Session, utc_point_in_time: datetime, entity_ids: list[str], run: RecorderRuns | None = None, no_attributes: bool = False) -> Iterable[Row]: ...
-def _get_single_entity_states_stmt(schema_version: int, utc_point_in_time: datetime, entity_id: str, no_attributes: bool = False) -> StatementLambdaElement: ...
+def _get_single_entity_states_stmt(utc_point_in_time: datetime, entity_id: str, no_attributes: bool = False) -> StatementLambdaElement: ...
 def _sorted_states_to_dict(hass: HomeAssistant, session: Session, states: Iterable[Row], start_time: datetime, entity_ids: list[str], include_start_time_state: bool = True, minimal_response: bool = False, no_attributes: bool = False, compressed_state_format: bool = False) -> dict[str, list[State | dict[str, Any]]]: ...

homeassistant-stubs/components/recorder/history/modern.pyi

@@ -1,4 +1,3 @@
-from ... import recorder as recorder
 from ..const import LAST_REPORTED_SCHEMA_VERSION as LAST_REPORTED_SCHEMA_VERSION
 from ..db_schema import SHARED_ATTR_OR_LEGACY_ATTRIBUTES as SHARED_ATTR_OR_LEGACY_ATTRIBUTES, StateAttributes as StateAttributes, States as States
 from ..filters import Filters as Filters
@@ -10,6 +9,7 @@ from collections.abc import Callable as Callable, Iterable
 from datetime import datetime
 from homeassistant.const import COMPRESSED_STATE_LAST_UPDATED as COMPRESSED_STATE_LAST_UPDATED, COMPRESSED_STATE_STATE as COMPRESSED_STATE_STATE
 from homeassistant.core import HomeAssistant as HomeAssistant, State as State, split_entity_id as split_entity_id
+from homeassistant.helpers.recorder import get_instance as get_instance
 from sqlalchemy import CompoundSelect as CompoundSelect, Select as Select, Subquery as Subquery
 from sqlalchemy.engine.row import Row as Row
 from sqlalchemy.orm.session import Session as Session

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

@@ -141,26 +141,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
@@ -23,6 +23,7 @@ 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: ...
+async def ws_update_statistics_issues(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_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: ...

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: ...