From b0ec21bff3fd01dc5e1d217560ca6479aadaae8b Mon Sep 17 00:00:00 2001 From: Christopher Allan Webber Date: Mon, 2 Jan 2012 09:12:11 -0600 Subject: [PATCH] Add create_new_migration_record method to MigrationManager --- mediagoblin/db/sql/util.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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): """ -- 2.25.1