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 for migration_number, migration_func in self.sorted_migrations
if migration_number > db_current_migration] 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 Includes two optional args:
after you run it. - pre_callback: if called, this is a callback on something to
run pre-migration. Takes (migration_number, migration_func)
Each time you loop over this, it'll return the migration as arguments
number and migration function. - 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(): for migration_number, migration_func in self.migrations_to_run():
if pre_callback:
pre_callback(migration_number, migration_func)
migration_func(self.database) migration_func(self.database)
self.set_current_migration(migration_number) self.set_current_migration(migration_number)
yield migration_number, migration_func if post_callback:
post_callback(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