diff --git a/superset/migrations/versions/4736ec66ce19_.py b/superset/migrations/versions/4736ec66ce19_.py index 2d560d57dfd21..de86a392292ad 100644 --- a/superset/migrations/versions/4736ec66ce19_.py +++ b/superset/migrations/versions/4736ec66ce19_.py @@ -34,11 +34,10 @@ sa.Column('datasource_name', sa.String(255)), ) -bind = op.get_bind() -insp = sa.engine.reflection.Inspector.from_engine(bind) - def upgrade(): + bind = op.get_bind() + insp = sa.engine.reflection.Inspector.from_engine(bind) # Add the new less restrictive uniqueness constraint. with op.batch_alter_table('datasources', naming_convention=conv) as batch_op: @@ -114,6 +113,8 @@ def upgrade(): def downgrade(): + bind = op.get_bind() + insp = sa.engine.reflection.Inspector.from_engine(bind) # Add the new more restrictive uniqueness constraint which is required by # the foreign key constraints. Note this operation will fail if the