Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix Statistics recorder migration path by dropping in pairs (#52453)

pull/52545/head
Franck Nijhof 2021-07-05 13:34:40 +02:00 committed by GitHub
parent 5e9127ef7a
commit 74029a0948
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
homeassistant/components/recorder/migration.py
View File

@ -461,14 +461,19 @@ def _apply_update(engine, session, new_version, old_version):
# This dropped the statistics table, done again in version 18.
pass
elif new_version == 18:
# Recreate the statisticsmeta tables
if sqlalchemy.inspect(engine).has_table(StatisticsMeta.__tablename__):
StatisticsMeta.__table__.drop(engine)
StatisticsMeta.__table__.create(engine)
# Recreate the statistics and statistics meta tables.
#
# Order matters! Statistics has a relation with StatisticsMeta,
# so statistics need to be deleted before meta (or in pair depending
# on the SQL backend); and meta needs to be created before statistics.
if sqlalchemy.inspect(engine).has_table(
StatisticsMeta.__tablename__
) or sqlalchemy.inspect(engine).has_table(Statistics.__tablename__):
Base.metadata.drop_all(
bind=engine, tables=[Statistics.__table__, StatisticsMeta.__table__]
)
# Recreate the statistics table
if sqlalchemy.inspect(engine).has_table(Statistics.__tablename__):
Statistics.__table__.drop(engine)
StatisticsMeta.__table__.create(engine)
Statistics.__table__.create(engine)
else:
raise ValueError(f"No schema migration defined for version {new_version}")