Fixes to the downgrade migration for user_id in sqlatables

apache · Dec 17, 2015 · f24e58a · f24e58a
1 parent 6f948f0
commit f24e58a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/panoramix/migrations/versions/2591d77e9831_user_id.py b/panoramix/migrations/versions/2591d77e9831_user_id.py
@@ -21,7 +21,6 @@ def upgrade():
 
 
 def downgrade():
-  with op.batch_alter_table('tables'):
-    op.drop_constraint(None, 'tables', type_='foreignkey')
-    op.drop_column('tables', 'user_id')
-
+  with op.batch_alter_table('tables') as batch_op:
+    batch_op.drop_constraint('user_id', type_='foreignkey')
+    batch_op.drop_column('user_id')