-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdb_migrate.py
More file actions
executable file
·28 lines (21 loc) · 968 Bytes
/
db_migrate.py
File metadata and controls
executable file
·28 lines (21 loc) · 968 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/Users/mbasanta/virtualenvs/QR5Server/bin/python
import imp
from migrate.versioning import api
from qr5server import db
from config import BaseConfiguration
DATABASE_URI = BaseConfiguration.SQLALCHEMY_DATABASE_URI
MIGRATE_REPO = BaseConfiguration.SQLALCHEMY_MIGRATE_REPO
v = api.db_version(DATABASE_URI, MIGRATE_REPO)
migration = MIGRATE_REPO + ('/versions/%03d_migration.py' % (v+1))
tmp_module = imp.new_module('old_model')
old_model = api.create_model(DATABASE_URI, MIGRATE_REPO)
exec(old_model, tmp_module.__dict__)
script = api.make_update_script_for_model(DATABASE_URI,
MIGRATE_REPO,
tmp_module.meta,
db.metadata)
open(migration, "wt").write(script)
api.upgrade(DATABASE_URI, MIGRATE_REPO)
v = api.db_version(DATABASE_URI, MIGRATE_REPO)
print('New migration saved as ' + migration)
print('Current database version: ' + str(v))