47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
|
import sqlite3
|
||
|
|
import os
|
||
|
|
import binascii
|
||
|
|
|
||
|
|
|
||
|
|
def do_migration(conf):
|
||
|
|
db_path = os.path.join(conf.data_dir, "lbrynet.sqlite")
|
||
|
|
connection = sqlite3.connect(db_path)
|
||
|
|
cursor = connection.cursor()
|
||
|
|
|
||
|
|
cursor.execute(
|
||
|
|
"pragma foreign_keys=off;"
|
||
|
|
)
|
||
|
|
|
||
|
|
cursor.execute("""
|
||
|
|
create table if not exists new_file (
|
||
|
|
stream_hash text primary key not null references stream,
|
||
|
|
file_name text,
|
||
|
|
download_directory text,
|
||
|
|
blob_data_rate real not null,
|
||
|
|
status text not null,
|
||
|
|
saved_file integer not null,
|
||
|
|
content_fee text
|
||
|
|
);
|
||
|
|
""")
|
||
|
|
for (stream_hash, file_name, download_dir, data_rate, status) in cursor.execute("select * from file").fetchall():
|
||
|
|
saved_file = 0
|
||
|
|
if download_dir != '{stream}' and file_name != '{stream}':
|
||
|
|
try:
|
||
|
|
if os.path.isfile(os.path.join(binascii.unhexlify(download_dir).decode(),
|
||
|
|
binascii.unhexlify(file_name).decode())):
|
||
|
|
saved_file = 1
|
||
|
|
else:
|
||
|
|
download_dir, file_name = None, None
|
||
|
|
except (OSError, ValueError):
|
||
|
|
download_dir, file_name = None, None
|
||
|
|
else:
|
||
|
|
download_dir, file_name = None, None
|
||
|
|
cursor.execute(
|
||
|
|
"insert into new_file values (?, ?, ?, ?, ?, ?, NULL)",
|
||
|
|
(stream_hash, file_name, download_dir, data_rate, status, saved_file)
|
||
|
|
)
|
||
|
|
cursor.execute("drop table file")
|
||
|
|
cursor.execute("alter table new_file rename to file")
|
||
|
|
connection.commit()
|
||
|
|
connection.close()
|