123456789101112131415 |
- #!/usr/bin/env python2
- import imp
- from migrate.versioning import api
- from flask_app import db, config
- migration = config['SQLALCHEMY_MIGRATE_REPO'] + '/versions/%03d_migration.py' % (api.db_version(config['SQLALCHEMY_DATABASE_URI'], config['SQLALCHEMY_MIGRATE_REPO']) + 1)
- tmp_module = imp.new_module('old_model')
- old_model = api.create_model(config['SQLALCHEMY_DATABASE_URI'], config['SQLALCHEMY_MIGRATE_REPO'])
- exec old_model in tmp_module.__dict__
- script = api.make_update_script_for_model(config['SQLALCHEMY_DATABASE_URI'], config['SQLALCHEMY_MIGRATE_REPO'], tmp_module.meta, db.metadata)
- open(migration, "wt").write(script)
- api.upgrade(config['SQLALCHEMY_DATABASE_URI'], config['SQLALCHEMY_MIGRATE_REPO'])
- print 'New migration saved as ' + migration
- print 'Current database version: ' + str(api.db_version(config['SQLALCHEMY_DATABASE_URI'], config['SQLALCHEMY_MIGRATE_REPO']))
|