from django.contrib import admin
from mptt.admin import MPTTModelAdmin, DraggableMPTTAdmin

from .models import Course, Category, Lesson, Material, CourseComment, LessonComment


@admin.register(Course)
class CourseAdmin(admin.ModelAdmin):
    list_display = (
        'author',
        'title',
        'price',
        'is_infinite',
        'category',
        'status',
        'created_at',
        'update_at',
    )


@admin.register(Category)
class CategoryAdmin(admin.ModelAdmin):
    pass


@admin.register(Lesson)
class LessonAdmin(admin.ModelAdmin):
    list_display = (
        'title',
        'course',
    )


@admin.register(Material)
class MaterialAdmin(admin.ModelAdmin):
    list_display = (
        'title',
    )


@admin.register(CourseComment)
class CourseCommentAdmin(DraggableMPTTAdmin):
    pass


@admin.register(LessonComment)
class LessonCommentAdmin(DraggableMPTTAdmin):
    pass