Freeze integration setup timeout for recorder during non-live data migration (#131998)

This commit is contained in:
Erik Montnemery 2024-12-01 16:17:55 +01:00 committed by GitHub
parent 8878d0f0e1
commit 598ce1f3b0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -740,7 +740,7 @@ class Recorder(threading.Thread):
self.schema_version = schema_status.current_version
# Do non-live data migration
migration.migrate_data_non_live(self, self.get_session, schema_status)
self._migrate_data_offline(schema_status)
# Non-live migration is now completed, remaining steps are live
self.migration_is_live = True
@ -916,6 +916,13 @@ class Recorder(threading.Thread):
return False
def _migrate_data_offline(
self, schema_status: migration.SchemaValidationStatus
) -> None:
"""Migrate data."""
with self.hass.timeout.freeze(DOMAIN):
migration.migrate_data_non_live(self, self.get_session, schema_status)
def _migrate_schema_offline(
self, schema_status: migration.SchemaValidationStatus
) -> tuple[bool, migration.SchemaValidationStatus]: