forked from LBRYCommunity/lbry-sdk
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
import sqlite3
|
|
import os
|
|
import time
|
|
|
|
|
|
def do_migration(conf):
|
|
db_path = os.path.join(conf.data_dir, 'lbrynet.sqlite')
|
|
connection = sqlite3.connect(db_path)
|
|
connection.row_factory = sqlite3.Row
|
|
cursor = connection.cursor()
|
|
|
|
current_columns = []
|
|
for col_info in cursor.execute("pragma table_info('file');").fetchall():
|
|
current_columns.append(col_info[1])
|
|
|
|
if 'added_on' in current_columns:
|
|
connection.close()
|
|
print('already migrated')
|
|
return
|
|
|
|
# follow 12 step schema change procedure
|
|
cursor.execute("pragma foreign_keys=off")
|
|
|
|
# we don't have any indexes, views or triggers, so step 3 is skipped.
|
|
cursor.execute("drop table if exists new_file")
|
|
cursor.execute("""
|
|
create table if not exists new_file (
|
|
stream_hash text not null primary key references stream,
|
|
file_name text,
|
|
download_directory text,
|
|
blob_data_rate text not null,
|
|
status text not null,
|
|
saved_file integer not null,
|
|
content_fee text,
|
|
added_on integer not null
|
|
);
|
|
|
|
|
|
""")
|
|
|
|
# step 5: transfer content from old to new
|
|
select = "select * from file"
|
|
for (stream_hash, file_name, download_dir, blob_rate, status, saved_file, fee) \
|
|
in cursor.execute(select).fetchall():
|
|
added_on = int(time.time())
|
|
cursor.execute(
|
|
"insert into new_file values (?, ?, ?, ?, ?, ?, ?, ?)",
|
|
(stream_hash, file_name, download_dir, blob_rate, status, saved_file, fee, added_on)
|
|
)
|
|
|
|
# step 6: drop old table
|
|
cursor.execute("drop table file")
|
|
|
|
# step 7: rename new table to old table
|
|
cursor.execute("alter table new_file rename to file")
|
|
|
|
# step 8: we aren't using indexes, views or triggers so skip
|
|
# step 9: no views so skip
|
|
# step 10: foreign key check
|
|
cursor.execute("pragma foreign_key_check;")
|
|
|
|
# step 11: commit transaction
|
|
connection.commit()
|
|
|
|
# step 12: re-enable foreign keys
|
|
connection.execute("pragma foreign_keys=on;")
|
|
|
|
# done :)
|
|
connection.close()
|