voussoir/ycdl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add database user_version check to merge_db.

Browse source
This commit is contained in:
voussoir 2020-07-01 16:00:09 -07:00
parent 5d000f87c3
commit 3c432c51f0
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
utilities/merge_db.py
View file

@ -25,6 +25,12 @@ def merge_db(from_db_path, to_db_path, channel):
to_db = sqlite3.connect(to_db_path) to_db = sqlite3.connect(to_db_path)
from_db = sqlite3.connect(from_db_path) from_db = sqlite3.connect(from_db_path)
to_version = to_db.execute('PRAGMA user_version').fetchone()[0]
from_version = from_db.execute('PRAGMA user_version').fetchone()[0]
if to_version != from_version:
raise Exception(f'Databases have different versions: to={to_version}, from={from_version}.')
to_db.execute('ATTACH DATABASE "%s" AS other' % from_db_path) to_db.execute('ATTACH DATABASE "%s" AS other' % from_db_path)
if channel == '*': if channel == '*':
_migrate_helper(to_db, 'channels') _migrate_helper(to_db, 'channels')