evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1163 from balloob/migration-5-fix

Browse source 
Fix recorder migration 5
This commit is contained in:
Paulus Schoutsen 2016-02-07 10:16:24 -08:00
commit 27dc2f61fb
1 changed files with 38 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

67
homeassistant/components/recorder.py
View file

@ -285,7 +285,8 @@ class Recorder(threading.Thread):
else: else:
return cur.fetchall() return cur.fetchall()
except sqlite3.IntegrityError: except (sqlite3.IntegrityError, sqlite3.OperationalError,
sqlite3.ProgrammingError):
_LOGGER.exception( _LOGGER.exception(
"Error querying the database using: %s", sql_query) "Error querying the database using: %s", sql_query)
return [] return []
@ -329,7 +330,7 @@ class Recorder(threading.Thread):
migration_id = 0 migration_id = 0
if migration_id < 1: if migration_id < 1:
self.query(""" cur.execute("""
CREATE TABLE recorder_runs ( CREATE TABLE recorder_runs (
run_id integer primary key, run_id integer primary key,
start integer, start integer,
@ -338,7 +339,7 @@ class Recorder(threading.Thread):
created integer) created integer)
""") """)
self.query(""" cur.execute("""
CREATE TABLE events ( CREATE TABLE events (
event_id integer primary key, event_id integer primary key,
event_type text, event_type text,
@ -346,10 +347,10 @@ class Recorder(threading.Thread):
origin text, origin text,
created integer) created integer)
""") """)
self.query( cur.execute(
'CREATE INDEX events__event_type ON events(event_type)') 'CREATE INDEX events__event_type ON events(event_type)')
self.query(""" cur.execute("""
CREATE TABLE states ( CREATE TABLE states (
state_id integer primary key, state_id integer primary key,
entity_id text, entity_id text,
@ -359,51 +360,51 @@ class Recorder(threading.Thread):
last_updated integer, last_updated integer,
created integer) created integer)
""") """)
self.query('CREATE INDEX states__entity_id ON states(entity_id)') cur.execute('CREATE INDEX states__entity_id ON states(entity_id)')
save_migration(1) save_migration(1)
if migration_id < 2: if migration_id < 2:
self.query(""" cur.execute("""
ALTER TABLE events ALTER TABLE events
ADD COLUMN time_fired integer ADD COLUMN time_fired integer
""") """)
self.query('UPDATE events SET time_fired=created') cur.execute('UPDATE events SET time_fired=created')
save_migration(2) save_migration(2)
if migration_id < 3: if migration_id < 3:
utc_offset = self.utc_offset utc_offset = self.utc_offset
self.query(""" cur.execute("""
ALTER TABLE recorder_runs ALTER TABLE recorder_runs
ADD COLUMN utc_offset integer ADD COLUMN utc_offset integer
""") """)
self.query(""" cur.execute("""
ALTER TABLE events ALTER TABLE events
ADD COLUMN utc_offset integer ADD COLUMN utc_offset integer
""") """)
self.query(""" cur.execute("""
ALTER TABLE states ALTER TABLE states
ADD COLUMN utc_offset integer ADD COLUMN utc_offset integer
""") """)
self.query("UPDATE recorder_runs SET utc_offset=?", [utc_offset]) cur.execute("UPDATE recorder_runs SET utc_offset=?", [utc_offset])
self.query("UPDATE events SET utc_offset=?", [utc_offset]) cur.execute("UPDATE events SET utc_offset=?", [utc_offset])
self.query("UPDATE states SET utc_offset=?", [utc_offset]) cur.execute("UPDATE states SET utc_offset=?", [utc_offset])
save_migration(3) save_migration(3)
if migration_id < 4: if migration_id < 4:
# We had a bug where we did not save utc offset for recorder runs # We had a bug where we did not save utc offset for recorder runs
self.query( cur.execute(
"""UPDATE recorder_runs SET utc_offset=? """UPDATE recorder_runs SET utc_offset=?
WHERE utc_offset IS NULL""", [self.utc_offset]) WHERE utc_offset IS NULL""", [self.utc_offset])
self.query(""" cur.execute("""
ALTER TABLE states ALTER TABLE states
ADD COLUMN event_id integer ADD COLUMN event_id integer
""") """)
@ -412,25 +413,33 @@ class Recorder(threading.Thread):
if migration_id < 5: if migration_id < 5:
# Add domain so that thermostat graphs look right # Add domain so that thermostat graphs look right
self.query(""" try:
ALTER TABLE states cur.execute("""
ADD COLUMN domain text ALTER TABLE states
""") ADD COLUMN domain text
""")
except sqlite3.OperationalError:
# We had a bug in this migration for a while on dev
# Without this, dev-users will have to throw away their db
pass
# TravisCI has Python compiled against an old version of SQLite3
# which misses the instr method.
self.conn.create_function(
"instr", 2,
lambda string, substring: string.find(substring) + 1)
# populate domain with defaults # populate domain with defaults
rows = self.query("select distinct entity_id from states") cur.execute("""
for row in rows: UPDATE states
entity_id = row[0] set domain=substr(entity_id, 0, instr(entity_id, '.'))
domain = entity_id.split(".")[0] """)
self.query(
"UPDATE states set domain=? where entity_id=?",
domain, entity_id)
# add indexes we are going to use a lot on selects # add indexes we are going to use a lot on selects
self.query(""" cur.execute("""
CREATE INDEX states__state_changes ON CREATE INDEX states__state_changes ON
states (last_changed, last_updated, entity_id)""") states (last_changed, last_updated, entity_id)""")
self.query(""" cur.execute("""
CREATE INDEX states__significant_changes ON CREATE INDEX states__significant_changes ON
states (domain, last_updated, entity_id)""") states (domain, last_updated, entity_id)""")
save_migration(5) save_migration(5)