From: Christopher Allan Webber <cwebber@dustycloud.org>
Date: Sat, 9 Jul 2011 20:07:40 +0000 (-0500)
Subject: latest_migration now returns migration numbers only, and 0 if no migrations.
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=dca6406a90227c72dd6e0661d8cd26556a44606b;p=mediagoblin.git

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 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):
         """