from wagtail.admin.ui.tables import BaseColumn


class OrderableColumn(BaseColumn):
    header_template_name = "wagtailadmin/orderable/orderable_header.html"
    cell_template_name = "wagtailadmin/orderable/orderable_cell.html"

    def get_cell_context_data(self, instance, parent_context):
        context = super().get_cell_context_data(instance, parent_context)
        context["instance_id"] = instance.pk
        return context
