from .models import COFRootPage, COFPage, COFEvent, COFActuIndexPage, COFActuPage, COFDirectoryPage, COFDirectoryEntryPage

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

@register(COFRootPage)
class COFPageTr(WagtailTranslationOptions):
    fields = (
        'introduction',
    )

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

@register(COFEvent)
class COFEventTr(WagtailTranslationOptions):
    fields = (
        'title',
        'description',
    )

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

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

@register(COFDirectoryPage)
class COFDirectoryPageTr(WagtailTranslationOptions):
    fields = (
        'introduction',
    )
    
@register(COFDirectoryEntryPage)
class COFDirectoryEntryPageTr(WagtailTranslationOptions):
    fields = (
        'body',
        'links',
    )