Split up run_upgrade

pull/307/head
Erik Johnston 2015-10-16 11:19:44 +01:00
parent bcfb653816
commit 22a8c91448
1 changed files with 11 additions and 5 deletions

View File

@ -55,16 +55,24 @@ CREATE INDEX event_search_ev_ridx ON event_search(room_id);
SQLITE_TABLE = ( SQLITE_TABLE = (
"CREATE VIRTUAL TABLE event_search USING fts3 ( event_id, room_id, key, value)" "CREATE VIRTUAL TABLE event_search USING fts3 ( event_id, room_id, key, value)"
) )
SQLITE_INDEX = "CREATE INDEX event_search_ev_idx ON event_search(event_id)"
def run_upgrade(cur, database_engine, *args, **kwargs): def run_upgrade(cur, database_engine, *args, **kwargs):
if isinstance(database_engine, PostgresEngine): if isinstance(database_engine, PostgresEngine):
for statement in get_statements(POSTGRES_SQL.splitlines()): run_postgres_upgrade(cur)
cur.execute(statement)
return return
if isinstance(database_engine, Sqlite3Engine): if isinstance(database_engine, Sqlite3Engine):
run_sqlite_upgrade(cur)
return
def run_postgres_upgrade(cur):
for statement in get_statements(POSTGRES_SQL.splitlines()):
cur.execute(statement)
def run_sqlite_upgrade(cur):
cur.execute(SQLITE_TABLE) cur.execute(SQLITE_TABLE)
rowid = -1 rowid = -1
@ -113,5 +121,3 @@ def run_upgrade(cur, database_engine, *args, **kwargs):
" VALUES (?,?,?,?)", " VALUES (?,?,?,?)",
rows rows
) )
# cur.execute(SQLITE_INDEX)