from modeltranslation.decorators import register
from wagtail_modeltranslation.translator import WagtailTranslationOptions

from .models import (
    COFActuIndexPage,
    COFActuPage,
    COFDirectoryEntryPage,
    COFDirectoryPage,
    COFPage,
    COFRootPage,
)


@register(COFRootPage)
class COFRootPageTr(WagtailTranslationOptions):
    fields = ("introduction",)


@register(COFPage)
class COFPageTr(WagtailTranslationOptions):
    fields = ("body",)


@register(COFActuIndexPage)
class COFActuIndexPageTr(WagtailTranslationOptions):
    fields = ()


@register(COFActuPage)
class COFActuPageTr(WagtailTranslationOptions):
    fields = ("chapo", "body")


@register(COFDirectoryPage)
class COFDirectoryPageTr(WagtailTranslationOptions):
    fields = ("introduction",)


@register(COFDirectoryEntryPage)
class COFDirectoryEntryPageTr(WagtailTranslationOptions):
    fields = ("body", "links")