diff --git a/migrate/versioning/genmodel.py b/migrate/versioning/genmodel.py index d929d92..a2170e2 100644 --- a/migrate/versioning/genmodel.py +++ b/migrate/versioning/genmodel.py @@ -125,11 +125,11 @@ class ModelGenerator(object): def toUpgradeDowngradePython(self, indent=' '): ''' Assume model is most current and database is out-of-date. ''' - decls = ['from migrate.changeset import schema', 'meta = MetaData(migrate_engine)'] for table in self.diff.tablesMissingInModel + \ - self.diff.tablesMissingInDatabase: + self.diff.tablesMissingInDatabase + \ + self.diff.tablesWithDiff: decls.extend(self.getTableDefn(table)) upgradeCommands, downgradeCommands = [], []