from django.contrib import admin
from .models import (
    AnatomyQuestion,
    Examination,
    QuestionType,
    Answers,
    HalfMarkAnswers,
    IncorrectAnswers,
    UserAnswer,
    Exam,
    Modality,
    BodyPart,
    Structure,
)

from reversion.admin import VersionAdmin

# Register your models here.
class AnatomyAnswersInline(admin.TabularInline):
    model = Answers
    extra = 1


class HalfMarkAnatomyAnswersInline(admin.TabularInline):
    model = HalfMarkAnswers
    extra = 1


class IncorrectAnatomyAnswersInline(admin.TabularInline):
    model = IncorrectAnswers
    extra = 1


class ExamInline(admin.TabularInline):
    model = AnatomyQuestion.exams.through
    extra = 1


class AnatomyAdmin(VersionAdmin):
    inlines = [
        AnatomyAnswersInline,
        HalfMarkAnatomyAnswersInline,
        IncorrectAnatomyAnswersInline,
        ExamInline,
    ]


admin.site.register(AnatomyQuestion, AnatomyAdmin)
admin.site.register(Examination)
admin.site.register(QuestionType)
admin.site.register(UserAnswer)
admin.site.register(Exam)
admin.site.register(Modality)
admin.site.register(BodyPart)
admin.site.register(Structure)