from django.contrib import admin
from training import models
from reversion.admin import VersionAdmin


admin.site.register(models.TrainingCategory, VersionAdmin)
admin.site.register(models.TrainingLevel, VersionAdmin)
admin.site.register(models.TrainingLevelQualification, VersionAdmin)
admin.site.register(models.TrainingLevelRequirement, VersionAdmin)


@admin.register(models.TrainingItemQualification)
class TrainingItemQualificationAdmin(VersionAdmin):
    list_display = ["__str__", "trainee"]


@admin.register(models.TrainingItem)
class TrainingItemAdmin(VersionAdmin):
    list_display = ["__str__", "category", "active", "technician_can_train"]
    list_filter = ["category", "active", "technician_can_train"]