From: Christopher Allan Webber <cwebber@dustycloud.org>
Date: Mon, 2 Jan 2012 15:12:11 +0000 (-0600)
Subject: Add create_new_migration_record method to MigrationManager
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=b0ec21bff3fd01dc5e1d217560ca6479aadaae8b;p=mediagoblin.git

Add create_new_migration_record method to MigrationManager
---

diff --git a/mediagoblin/db/sql/util.py b/mediagoblin/db/sql/util.py
index 59e8eb8b..a42d992f 100644
--- a/mediagoblin/db/sql/util.py
+++ b/mediagoblin/db/sql/util.py
@@ -126,6 +126,7 @@ class MigrationManager(object):
         """
         # sanity check before we proceed, none of these should be created
         for model in self.models:
+            # Maybe in the future just print out a "Yikes!" or something?
             assert not model.__table__.exists(self.database)
 
         self.migration_model.metadata.create_all(
@@ -133,8 +134,12 @@ class MigrationManager(object):
             tables=[model.__table__ for model in self.models])
 
     def create_new_migration_record(self):
-        ## TODO
-        pass
+        """
+        Create a new migration record for this migration set
+        """
+        self.migration_model(
+            name=self.name,
+            version=self.latest_migration())
 
     def dry_run(self):
         """