diff options
Diffstat (limited to 'mediagoblin/db/util.py')
-rw-r--r-- | mediagoblin/db/util.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/mediagoblin/db/util.py b/mediagoblin/db/util.py index a7ee8dd1..c8932755 100644 --- a/mediagoblin/db/util.py +++ b/mediagoblin/db/util.py @@ -119,6 +119,9 @@ MIGRATIONS = {} class RegisterMigration(object): + """ + Tool for registering migrations + """ def __init__(self, migration_number, migration_registry=MIGRATIONS): self.migration_number = migration_number self.migration_registry = migration_registry @@ -161,13 +164,14 @@ class MigrationManager(object): def latest_migration(self): """ - Return a tuple like: - (migration_number, migration_func) - - Where migration_number is the number of the latest migration - and migration func is the actual function that would be run. + Return a migration number for the latest migration, or 0 if + there are no migrations. """ - return self.sorted_migrations[-1] + if self.sorted_migrations: + return self.sorted_migrations[-1][0] + else: + # If no migrations have been set, we start at 0. + return 0 def set_current_migration(self, migration_number=None): """ |