diff --git a/migrations/versions/89a74e347d85_next_generation_bridge_management.py b/migrations/versions/89a74e347d85_next_generation_bridge_management.py index 1bfa0d8..9150056 100644 --- a/migrations/versions/89a74e347d85_next_generation_bridge_management.py +++ b/migrations/versions/89a74e347d85_next_generation_bridge_management.py @@ -17,10 +17,12 @@ depends_on = None def upgrade(): - op.drop_constraint('fk_bridge_conf_id_bridge_conf', 'bridge') - op.drop_constraint('fk_bridge_conf_group_id_group', 'bridge_conf') - op.drop_constraint('pk_bridge', 'bridge') - op.drop_constraint('pk_bridge_conf', 'bridge_conf') + bind = op.get_bind() + if bind.engine.name != 'sqlite': + op.drop_constraint('fk_bridge_conf_id_bridge_conf', 'bridge') + op.drop_constraint('fk_bridge_conf_group_id_group', 'bridge_conf') + op.drop_constraint('pk_bridge', 'bridge') + op.drop_constraint('pk_bridge_conf', 'bridge_conf') op.rename_table('bridge', 'original_bridge') op.rename_table('bridge_conf', 'original_bridge_conf') op.create_table('bridge_conf', @@ -58,11 +60,14 @@ def upgrade(): def downgrade(): + bind = op.get_bind() op.drop_table('bridge') op.drop_table('bridge_conf') op.rename_table('original_bridge_conf', 'bridge_conf') - op.create_primary_key('pk_bridge_conf', 'bridge_conf', ['id']) - op.create_foreign_key('fk_bridge_conf_group_id_group', 'bridge_conf', 'group', ['group_id'], ['id']) + if bind.engine.name != 'sqlite': + op.create_primary_key('pk_bridge_conf', 'bridge_conf', ['id']) + op.create_foreign_key('fk_bridge_conf_group_id_group', 'bridge_conf', 'group', ['group_id'], ['id']) op.rename_table('original_bridge', 'bridge') - op.create_primary_key('pk_bridge', 'bridge', ['id']) - op.create_foreign_key('fk_bridge_conf_id_bridge_conf', 'bridge', 'bridge_conf', ['conf_id'], ['id']) + if bind.engine.name != 'sqlite': + op.create_primary_key('pk_bridge', 'bridge', ['id']) + op.create_foreign_key('fk_bridge_conf_id_bridge_conf', 'bridge', 'bridge_conf', ['conf_id'], ['id'])