from django.contrib import admin

from university.models import (
    User,
    Course,
    Chapter,
    Lesson,
    Upload,
    Badge,
    UserBadge,
    Exam,
    UserExam,
    Question,
    Quiz,
    UserQuiz
)


class CourseAdmin(admin.ModelAdmin):
    list_display = ('title', 'slug', 'cover_image')


class ChapterAdmin(admin.ModelAdmin):
    list_display = ('title', 'description', 'course')


class LessonAdmin(admin.ModelAdmin):
    list_display = ('id', 'title', 'speaker', 'periods', 'sequence', 'cover_image', 'chapter', 'course_title')

    def course_title(self, obj):
        return obj.chapter.course.title


class UploadAdmin(admin.ModelAdmin):
    list_display = ('path', 'original_filename', 'filesize', 'extension', 'user')


class UserBadgeAdmin(admin.ModelAdmin):
    list_display = ('user', 'badge', 'granted_at')


admin.site.register(User)
admin.site.register(Badge)
admin.site.register(Exam)
admin.site.register(UserExam)
admin.site.register(Question)
admin.site.register(Quiz)
admin.site.register(UserQuiz)
admin.site.register(UserBadge, UserBadgeAdmin)
admin.site.register(Course, CourseAdmin)
admin.site.register(Chapter, ChapterAdmin)
admin.site.register(Lesson, LessonAdmin)
admin.site.register(Upload, UploadAdmin)