diff options
| author | Elrond <elrond+mediagoblin.org@samba-tng.org> | 2012-12-13 12:35:27 +0100 |
|---|---|---|
| committer | Elrond <elrond+mediagoblin.org@samba-tng.org> | 2013-01-08 22:50:01 +0100 |
| commit | c4466cb4dcf194f634b6fe855f5e5fad09e70df1 (patch) | |
| tree | 67adca82eb905486c598aa2de861527e4496ce0f /mediagoblin/db/migrations.py | |
| parent | a050e776c6d9c4970dd241e92b2f24a8c9deb36d (diff) | |
| download | mediagoblin-c4466cb4dcf194f634b6fe855f5e5fad09e70df1.tar.lz mediagoblin-c4466cb4dcf194f634b6fe855f5e5fad09e70df1.tar.xz mediagoblin-c4466cb4dcf194f634b6fe855f5e5fad09e70df1.zip | |
Add inspect_table.
In our current scheme of migrations, we reflect the current
sql schema into an SQLAlchemy schema. So let's have a tool
function for this.
Diffstat (limited to 'mediagoblin/db/migrations.py')
| -rw-r--r-- | mediagoblin/db/migrations.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/mediagoblin/db/migrations.py b/mediagoblin/db/migrations.py index 95270198..469d2874 100644 --- a/mediagoblin/db/migrations.py +++ b/mediagoblin/db/migrations.py @@ -23,7 +23,7 @@ from sqlalchemy.exc import ProgrammingError from sqlalchemy.ext.declarative import declarative_base from migrate.changeset.constraint import UniqueConstraint -from mediagoblin.db.sql.migration_tools import RegisterMigration +from mediagoblin.db.sql.migration_tools import RegisterMigration, inspect_table from mediagoblin.db.models import MediaEntry, Collection, User MIGRATIONS = {} @@ -60,8 +60,7 @@ def add_wants_notification_column(db_conn): def add_transcoding_progress(db_conn): metadata = MetaData(bind=db_conn.bind) - media_entry = Table('core__media_entries', metadata, autoload=True, - autoload_with=db_conn.bind) + media_entry = inspect_table(metadata, 'core__media_entries') col = Column('transcoding_progress', SmallInteger) col.create(media_entry) @@ -115,8 +114,7 @@ def add_collection_tables(db_conn): def add_mediaentry_collected(db_conn): metadata = MetaData(bind=db_conn.bind) - media_entry = Table('core__media_entries', metadata, autoload=True, - autoload_with=db_conn.bind) + media_entry = inspect_table(metadata, 'core__media_entries') col = Column('collected', Integer, default=0) col.create(media_entry) @@ -172,8 +170,7 @@ def fix_CollectionItem_v0_constraint(db_conn): metadata = MetaData(bind=db_conn.bind) - CollectionItem_table = Table('core__collection_items', - metadata, autoload=True, autoload_with=db_conn.bind) + CollectionItem_table = inspect_table(metadata, 'core__collection_items') constraint = UniqueConstraint('collection', 'media_entry', name='core__collection_items_collection_media_entry_key', |