from modeltranslation.translator import TranslationOptions, register

from .models import Category, Course, CourseFAQ, CourseHighlight, Lesson, Section


@register(Category)
class CategoryTranslationOptions(TranslationOptions):
    fields = ("name",)


@register(Course)
class CourseTranslationOptions(TranslationOptions):
    fields = ("title", "tagline", "description")


@register(CourseHighlight)
class CourseHighlightTranslationOptions(TranslationOptions):
    fields = ("text",)


@register(CourseFAQ)
class CourseFAQTranslationOptions(TranslationOptions):
    fields = ("question", "answer")


@register(Section)
class SectionTranslationOptions(TranslationOptions):
    fields = ("title",)


@register(Lesson)
class LessonTranslationOptions(TranslationOptions):
    fields = ("title", "body")