Migration with type creation

moschlar · Oct 30, 2013 · 6315b7c · 6315b7c
1 parent 67462b9
commit 6315b7c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/migration/versions/3a8c537e6090_test_visibility.py b/migration/versions/3a8c537e6090_test_visibility.py
@@ -35,15 +35,17 @@
 from sqlalchemy.orm import sessionmaker
 from sauce.model import Test
 
+test_visibility = sa.Enum('invisible', 'result_only', 'data_only', 'visible', name='test_visibility')
+
 
 def upgrade():
     cntxt = context.get_context()
     Session = sessionmaker(bind=cntxt.bind)
 
     op.alter_column('tests', 'visible', existing_type=sa.Boolean(), nullable=True)
 
-    test_visibility = sa.Enum('invisible', 'result_only', 'data_only', 'visible', name='test_visibility')
     test_visibility.create(op.get_bind(), checkfirst=False)
+
     op.add_column('tests', 
         sa.Column('visibility',
             test_visibility,
@@ -62,4 +64,6 @@ def upgrade():
 def downgrade():
     op.drop_column('tests', 'visibility')
 
+    test_visibility.drop(op.get_bind(), checkfirst=False)
+
     op.alter_column('tests', 'visible', existing_type=sa.Boolean(), nullable=False, default=False, server_default='False')
diff --git a/migration/versions/425be68ff414_event_enroll.py b/migration/versions/425be68ff414_event_enroll.py
@@ -31,10 +31,14 @@
 #from alembic.operations import Operations as op
 import sqlalchemy as sa
 
+event_enroll = sa.Enum('event', 'lesson', 'lesson_team', 'team', 'team_new', name='event_enroll')
+
 
 def upgrade():
-    op.add_column('events', sa.Column('enroll', sa.Enum('event', 'lesson', 'lesson_team', 'team', 'team_new', name='event_enroll'), nullable=True))
+    event_enroll.create(op.get_bind(), checkfirst=False)
+    op.add_column('events', sa.Column('enroll', event_enroll, nullable=True))
 
 
 def downgrade():
+    event_enroll.drop(op.get_bind(), checkfirst=False)
     op.drop_column('events', 'enroll')