| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
- import argparse
- import os
- import sqlite3
- import sys
- sys.path.append(os.path.dirname(os.path.dirname(__file__)))
- from timesearch_modules import tsdb
- def upgrade_1_to_2(db):
- '''
- In this version, many of the timesearch modules were renamed, including
- redmash -> index. This update will rename the existing `redmash` folder
- to `index`.
- '''
- cur = db.sql.cursor()
- redmash_dir = db.index_dir.parent.with_child('redmash')
- if redmash_dir.exists:
- redmash_dir.assert_is_directory()
- print('Renaming redmash folder to index.')
- os.rename(redmash_dir, db.index_dir)
- def upgrade_all(database_filename):
- '''
- Given the filename of a database, apply all of the needed
- upgrade_x_to_y functions in order.
- '''
- db = tsdb.TSDB(database_filename, do_create=False, skip_version_check=True)
- cur = db.sql.cursor()
- cur.execute('PRAGMA user_version')
- current_version = cur.fetchone()[0]
- needed_version = tsdb.DATABASE_VERSION
- if current_version == needed_version:
- print('Already up to date with version %d.' % needed_version)
- return
- for version_number in range(current_version + 1, needed_version + 1):
- print('Upgrading from %d to %d' % (current_version, version_number))
- upgrade_function = 'upgrade_%d_to_%d' % (current_version, version_number)
- upgrade_function = eval(upgrade_function)
- upgrade_function(db)
- db.sql.cursor().execute('PRAGMA user_version = %d' % version_number)
- db.sql.commit()
- current_version = version_number
- print('Upgrades finished.')
- def upgrade_all_argparse(args):
- return upgrade_all(database_filename=args.database_filename)
- def main(argv):
- parser = argparse.ArgumentParser()
- parser.add_argument('database_filename')
- parser.set_defaults(func=upgrade_all_argparse)
- args = parser.parse_args(argv)
- return args.func(args)
- if __name__ == '__main__':
- raise SystemExit(main(sys.argv[1:]))
|
