Another db->db.engine because I'm bad at things ;)

Thanks again Elrond.
This commit is contained in:
Christopher Allan Webber 2012-01-02 16:14:55 -06:00
parent 3f2c6f96c1
commit cfbbdcc5ad

View File

@ -50,10 +50,10 @@ class MigrationManager(object):
self.printer = printer self.printer = printer
# For convenience # For convenience
from mediagoblin.db.sql.models import MigrationData from mediagoblin.db.sql.models import MigrationRecord
self.migration_model = MigrationData self.migration_model = MigrationRecord
self.migration_table = MigrationData.__table__ self.migration_table = MigrationRecord.__table__
@property @property
def sorted_migrations(self): def sorted_migrations(self):
@ -251,8 +251,8 @@ def assure_migrations_table_setup(db):
""" """
Make sure the migrations table is set up in the database. Make sure the migrations table is set up in the database.
""" """
from mediagoblin.db.sql.models import MigrationData from mediagoblin.db.sql.models import MigrationRecord
if not MigrationData.__table__.exists(db): if not MigrationRecord.__table__.exists(db.engine):
MigrationData.metadata.create_all( MigrationRecord.metadata.create_all(
db, tables=[MigrationData.__table__]) db, tables=[MigrationRecord.__table__])