from longs.models import Exam
from django.contrib import admin

# Register your models here.
from .models import (
    CidUserGroup,
    Examination,
    #Sign,
    Modality,
    Plane,
    Contrast,
    QuestionNote,
    NoteType,
    CidUser,
    CidUserExam,
    ExamUserStatus,
    UserGrades,
    UserProfile,
    Supervisor,
    ExamCollection,
    Site
)

# from .models import Examination, Sign, Site, Condition

admin.site.register(Examination)
#admin.site.register(Sign)
admin.site.register(Site)
#admin.site.register(Condition)
admin.site.register(Plane)
admin.site.register(Contrast)
admin.site.register(QuestionNote)
admin.site.register(NoteType)
admin.site.register(CidUser)
admin.site.register(CidUserGroup)
admin.site.register(UserProfile)
admin.site.register(UserGrades)
admin.site.register(Supervisor)
admin.site.register(Modality)
admin.site.register(ExamCollection)

class CidUserExamAdmin(admin.ModelAdmin):
    list_filter = ('cid_user', 'user_user', 'completed')

    def get_queryset(self, request):
        qs = super().get_queryset(request).prefetch_related('cid_user', 'user_user', 'exam')
        return qs


admin.site.register(CidUserExam, CidUserExamAdmin)

class ExamUserStatusAdmin(admin.ModelAdmin):

    list_filter = ('status', 'cid_user_exam__user_user', 'cid_user_exam__cid_user')

    def get_queryset(self, request):
        qs = super().get_queryset(request).prefetch_related('cid_user_exam', 'exam')
        return qs

admin.site.register(ExamUserStatus, ExamUserStatusAdmin)