| import common |
| from autotest_lib.database import db_utils, migrate |
| |
| TKO_MIGRATION_NAME = '031_rename_tko_tables' |
| migrations_module = __import__('autotest_lib.tko.migrations', globals(), |
| locals(), [TKO_MIGRATION_NAME]) |
| tko_migration = getattr(migrations_module, TKO_MIGRATION_NAME) |
| |
| TABLE_NAMES = tko_migration.RENAMES_UP.values() |
| |
| |
| def migrate_up(manager): |
| tko_manager = migrate.get_migration_manager(db_name='TKO', debug=False, |
| force=False) |
| if tko_manager.get_db_version() < 31: |
| raise Exception('You must update the TKO database to at least version ' |
| '31 before applying AUTOTEST_WEB migration 46') |
| |
| if manager.simulate: |
| tko_manager.initialize_and_fill_test_db() |
| |
| if not manager.force: |
| response = raw_input( |
| 'This migration will merge the autotest_web and tko databases. ' |
| 'Following the migration, the tko database will be dropped. ' |
| 'Any user-added tables in tko will NOT be migrated. This ' |
| 'migration is NOT reversible. Are you sure you want to ' |
| 'continue? (yes/no) ') |
| if response != 'yes': |
| raise Exception('User has chosen to abort migration') |
| |
| db_utils.move_tables(manager, tko_manager, TABLE_NAMES) |
| db_utils.drop_database(tko_manager) |
| manager.execute_script(tko_migration.RECREATE_VIEWS_UP) |
| |
| |
| def migrate_down(manager): |
| raise Exception('Migration 46 is not reversible!') |
