heckyel/mediagoblin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simpler run_migrations method.

Browse Source 
Allows for calbacks, should be useful for printing output and catching
things in tests.

Gets rid of the generator stuff which now that I think of it is a
messy idea.
This commit is contained in:
Christopher Allan Webber 2011-07-10 15:41:18 -05:00
parent 42fe078029
commit d0ee0003a2
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
mediagoblin/db/util.py
View File

@ -232,25 +232,22 @@ class MigrationManager(object):
for migration_number, migration_func in self.sorted_migrations
if migration_number > db_current_migration]
def iteratively_migrate(self):
def migrate_new(self, pre_callback=None, post_callback=None):
"""
Iteratively run all migrations.
Run all migrations.
Useful if you need to print some message about each migration
after you run it.
Each time you loop over this, it'll return the migration
number and migration function.
Includes two optional args:
- pre_callback: if called, this is a callback on something to
run pre-migration. Takes (migration_number, migration_func)
as arguments
- pre_callback: if called, this is a callback on something to
run post-migration. Takes (migration_number, migration_func)
as arguments
"""
for migration_number, migration_func in self.migrations_to_run():
if pre_callback:
pre_callback(migration_number, migration_func)
migration_func(self.database)
self.set_current_migration(migration_number)
yield migration_number, migration_func
def run_new_migrations(self):
"""
Install all migrations that need to be installed, quietly.
"""
for migration_number, migration_func in self.iteratively_migrate():
# No need to say anything... we're just migrating quietly.
pass
if post_callback:
post_callback(migration_number, migration_func)