latest_migration now returns migration numbers only, and 0 if no migrations.

This commit is contained in:
Christopher Allan Webber 2011-07-09 15:07:40 -05:00
parent 993f0a9321
commit dca6406a90

View File

@ -119,6 +119,9 @@ MIGRATIONS = {}
class RegisterMigration(object): class RegisterMigration(object):
"""
Tool for registering migrations
"""
def __init__(self, migration_number, migration_registry=MIGRATIONS): def __init__(self, migration_number, migration_registry=MIGRATIONS):
self.migration_number = migration_number self.migration_number = migration_number
self.migration_registry = migration_registry self.migration_registry = migration_registry
@ -161,13 +164,14 @@ class MigrationManager(object):
def latest_migration(self): def latest_migration(self):
""" """
Return a tuple like: Return a migration number for the latest migration, or 0 if
(migration_number, migration_func) there are no migrations.
Where migration_number is the number of the latest migration
and migration func is the actual function that would be run.
""" """
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): def set_current_migration(self, migration_number=None):
""" """