Elrond's suggestion: have set_current_migration execute after each migration run.

diff --git a/mediagoblin/db/sql/util.py b/mediagoblin/db/sql/util.py
index c6d8562e2fb0cce87f84d32bbe732cc9841be5f9..92866149476f87f29c6fd36edd2bb83bdbdee7b6 100644 (file)
--- a/mediagoblin/db/sql/util.py
+++ b/mediagoblin/db/sql/util.py
@@ -217,9 +217,9 @@ class MigrationManager(object):
                     u'   + Running migration %s, "%s"... ' % (
                         migration_number, migration_func.func_name))
                 migration_func(self.session)
+                self.set_current_migration()
                 self.printer('done.\n')
 
-            self.set_current_migration()
             return u'migrated'
 
         # Otherwise return None.  Well it would do this anyway, but