| 12345678910111213141516171819202122232425262728293031323334 |
- import os
- from . import common
- from . import tsdb
- MIGRATE_QUERY = '''
- INSERT INTO {tablename}
- SELECT othertable.* FROM other.{tablename} othertable
- LEFT JOIN {tablename} mytable ON mytable.idint == othertable.idint
- WHERE mytable.idint IS NULL;
- '''
- def _migrate_helper(db, tablename):
- query = MIGRATE_QUERY.format(tablename=tablename)
- print(query)
- oldcount = db.cur.execute('SELECT count(*) FROM %s' % tablename).fetchone()[0]
- db.cur.execute(query)
- db.sql.commit()
- newcount = db.cur.execute('SELECT count(*) FROM %s' % tablename).fetchone()[0]
- print('Gained %d items.' % (newcount - oldcount))
- def merge_db(from_db_path, to_db_path):
- to_db = tsdb.TSDB(to_db_path)
- from_db = tsdb.TSDB(from_db_path)
- to_db.cur.execute('ATTACH DATABASE "%s" AS other' % from_db_path)
- _migrate_helper(to_db, 'submissions')
- _migrate_helper(to_db, 'comments')
- def merge_db_argparse(args):
- return merge_db(args.from_db_path, args.to_db_path)
|
