a86472389a70_remove_hierarchy_for_level_values.py 文件源码

def upgrade():
    table_prefix = context.config.get_main_option('table_prefix')
    table_name = table_prefix + 'environment_hierarchy_level_value'
    with op.batch_alter_table(table_name) as batch:
        batch.drop_column('parent_id')

        batch.drop_constraint(
            table_name + '_level_id_fkey',
            type_='foreignkey'
        )
        batch.create_foreign_key(
            table_name + '_level_id_fkey',
            table_prefix + 'environment_hierarchy_level',
            ['level_id'], ['id'], ondelete='CASCADE'
        )

        batch.create_unique_constraint(
            table_name + '_level_id_value_unique',
            ['level_id', 'value']
        )