tables.py

from __future__ import annotations

from typing import Optional

from django.utils.translation import ugettext_lazy as _

import django_tables2 as tables
from django_tables2.utils import A

from .models import LessonPeriod


def _css_class_from_lesson_state(
    record: Optional[LessonPeriod] = None, table: Optional[LessonsTable] = None
) -> str:
    if record.get_substitution(record._week):
        if record.get_substitution(record._week).cancelled:
            return "table-danger"
        else:
            return "table-warning"
    else:
        return ""


class LessonsTable(tables.Table):
    class Meta:
        attrs = {"class": "table table-striped table-bordered table-hover table-responsive-xl"}
        row_attrs = {"class": _css_class_from_lesson_state}

    period__period = tables.Column(accessor="period.period")
    lesson__groups = tables.Column(accessor="lesson.group_names", verbose_name=_("Groups"))
    lesson__teachers = tables.Column(accessor="lesson.teacher_names", verbose_name=_("Teachers"))
    lesson__subject = tables.Column(accessor="lesson.subject")
    room = tables.Column(accessor="room")
    edit_substitution = tables.LinkColumn(
        "edit_substitution", args=[A("id"), A("_week")], text=_("Substitution")
    )


class SubstitutionsTable(tables.Table):
    class Meta:
        attrs = {"class": "table table-striped table-bordered table-hover table-responsive-xl"}

    lesson_period = tables.Column(verbose_name=_("Lesson"))
    lesson__groups = tables.Column(
        accessor="lesson_period.lesson.group_names", verbose_name=_("Groups")
    )
    lesson__teachers = tables.Column(
        accessor="lesson_period.get_teacher_names", verbose_name=_("Teachers")
    )
    lesson__subject = tables.Column(accessor="subject")
    room = tables.Column(accessor="room")
    cancelled = tables.BooleanColumn(accessor="cancelled", verbose_name=_("Cancelled"))