import unfold
from unfold.admin import ModelAdmin as UnfoldModelAdmin
from django.contrib import admin
from .models import TrainingTemplate, TemplateExercise


@admin.register(TrainingTemplate)
class TrainingTemplateAdmin(UnfoldModelAdmin):
    list_display = ['name', 'category', 'is_active', 'created_at']
    list_filter = ['category', 'is_active']
    search_fields = ['name', 'description']
    readonly_fields = ['created_at', 'updated_at']
    filter_horizontal = ['exercises']


@admin.register(TemplateExercise)
class TemplateExerciseAdmin(UnfoldModelAdmin):
    list_display = ['template', 'exercise', 'order', 'default_value']
    list_filter = ['template']
    raw_id_fields = ['template', 'exercise']