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

diff --git a/mediagoblin/db/util.py b/mediagoblin/db/util.py
index a7ee8dd1687bf7227967e54c34a3a48ad9a977eb..c89327552ce47746a64bc6c59e451a29a17786ea 100644 (file)
--- 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):
         """