forked from LBRYCommunity/lbry-sdk
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
import sqlite3
|
|
import os
|
|
import logging
|
|
|
|
log = logging.getLogger(__name__)
|
|
UNSET_NOUT = -1
|
|
|
|
def do_migration(db_dir):
|
|
log.info("Doing the migration")
|
|
migrate_blockchainname_db(db_dir)
|
|
log.info("Migration succeeded")
|
|
|
|
|
|
def migrate_blockchainname_db(db_dir):
|
|
blockchainname_db = os.path.join(db_dir, "blockchainname.db")
|
|
# skip migration on fresh installs
|
|
if not os.path.isfile(blockchainname_db):
|
|
return
|
|
temp_db = sqlite3.connect(":memory:")
|
|
db_file = sqlite3.connect(blockchainname_db)
|
|
file_cursor = db_file.cursor()
|
|
mem_cursor = temp_db.cursor()
|
|
|
|
mem_cursor.execute("create table if not exists name_metadata ("
|
|
" name text, "
|
|
" txid text, "
|
|
" n integer, "
|
|
" sd_hash text)")
|
|
mem_cursor.execute("create table if not exists claim_ids ("
|
|
" claimId text, "
|
|
" name text, "
|
|
" txid text, "
|
|
" n integer)")
|
|
temp_db.commit()
|
|
|
|
name_metadata = file_cursor.execute("select * from name_metadata").fetchall()
|
|
claim_metadata = file_cursor.execute("select * from claim_ids").fetchall()
|
|
|
|
# fill n as V1_UNSET_NOUT, Wallet.py will be responsible for filling in correct n
|
|
for name, txid, sd_hash in name_metadata:
|
|
mem_cursor.execute(
|
|
"insert into name_metadata values (?, ?, ?, ?) ",
|
|
(name, txid, UNSET_NOUT, sd_hash))
|
|
|
|
for claim_id, name, txid in claim_metadata:
|
|
mem_cursor.execute(
|
|
"insert into claim_ids values (?, ?, ?, ?)",
|
|
(claim_id, name, txid, UNSET_NOUT))
|
|
temp_db.commit()
|
|
|
|
new_name_metadata = mem_cursor.execute("select * from name_metadata").fetchall()
|
|
new_claim_metadata = mem_cursor.execute("select * from claim_ids").fetchall()
|
|
|
|
file_cursor.execute("drop table name_metadata")
|
|
file_cursor.execute("create table name_metadata ("
|
|
" name text, "
|
|
" txid text, "
|
|
" n integer, "
|
|
" sd_hash text)")
|
|
|
|
for name, txid, n, sd_hash in new_name_metadata:
|
|
file_cursor.execute(
|
|
"insert into name_metadata values (?, ?, ?, ?) ", (name, txid, n, sd_hash))
|
|
|
|
file_cursor.execute("drop table claim_ids")
|
|
file_cursor.execute("create table claim_ids ("
|
|
" claimId text, "
|
|
" name text, "
|
|
" txid text, "
|
|
" n integer)")
|
|
|
|
for claim_id, name, txid, n in new_claim_metadata:
|
|
file_cursor.execute("insert into claim_ids values (?, ?, ?, ?)", (claim_id, name, txid, n))
|
|
|
|
db_file.commit()
|
|
db_file.close()
|
|
temp_db.close()
|