# tables.py
import django_tables2 as tables
from .models import Task

# render the StatusColumn based on the style.css
class StatusColumn(tables.Column):
    attrs = {
        "td": {
            "class": lambda record: record.status
        },
        "tr": {
            "class": lambda record: record.status
        }
    }
    def render(self, record):
        return "{}".format(record.status)

# render the StatusColumn based on the style.css
class ResumeColumn(tables.BooleanColumn):
    def render(self, record):
        return record.resume

class TaskTable(tables.Table):

    class Meta:
        model = Task
        template_name = "django_tables2/bootstrap4.html"
        fields = ("id", "workflow","filter","priority","status","quality","project","sas_id","resume","actions","buttons")

    # columns that need specific rendering
    status = StatusColumn()
    #resume = ResumeColumn()

    actions = tables.TemplateColumn(verbose_name='Details',
                                    template_name='query/action_buttons_per_row.html',
                                    orderable=False)  # orderable not sortable

    buttons = tables.TemplateColumn(verbose_name='Set Status',
                                    template_name='query/status_buttons_per_row.html',
                                    orderable=False)  # orderable not sortable