import unfold
from unfold.admin import ModelAdmin as UnfoldModelAdmin
from django.contrib import admin
from .models import Training, Attendance


@admin.register(Training)
class TrainingAdmin(UnfoldModelAdmin):
    list_display = ['date', 'start_time', 'group', 'location', 'is_completed']
    list_filter = ['group', 'is_completed', 'date']
    search_fields = ['notes']
    readonly_fields = ['created_at', 'updated_at']
    raw_id_fields = ['location', 'template']
    filter_horizontal = ['trainers']
    date_hierarchy = 'date'


@admin.register(Attendance)
class AttendanceAdmin(UnfoldModelAdmin):
    list_display = ['training', 'wrestler', 'created_at']
    list_filter = ['training']
    raw_id_fields = ['training', 'wrestler']