from django.contrib import admin

from .models import SignoffEvidence, TrainerSignoff, TrainingRecord


class TrainerSignoffInline(admin.TabularInline):
    model = TrainerSignoff
    extra = 0
    readonly_fields = ("signed_off_at",)


class SignoffEvidenceInline(admin.TabularInline):
    model = SignoffEvidence
    extra = 0
    readonly_fields = ("uploaded_at",)


@admin.register(TrainingRecord)
class TrainingRecordAdmin(admin.ModelAdmin):
    list_display = ("id", "enrollment", "completion_mode", "status", "completed_at")
    list_filter = ("completion_mode", "status")
    inlines = [TrainerSignoffInline, SignoffEvidenceInline]


@admin.register(TrainerSignoff)
class TrainerSignoffAdmin(admin.ModelAdmin):
    list_display = ("training_record", "trainer", "decision", "signed_off_at")
    list_filter = ("decision",)
    raw_id_fields = ("trainer",)