"""Translation classes."""

from modeltranslation.translator import register, TranslationOptions
from webapp.models import BaseFeed, Feed, Tag, Event, Registration
from webapp.models import PresetRole, BaseRole, PresetKaehmyRole


@register(BaseFeed)
class BaseFeedTranslationOptions(TranslationOptions):
    """Class for base feed translation options."""

    fields = ('title', 'description', 'content')


@register(Feed)
class FeedTranslationOptions(TranslationOptions):
    """Class for feed translation options."""

    fields = ()


@register(Event)
class EventTranslationOptions(TranslationOptions):
    """Class for event translation options."""

    fields = ()


@register(Tag)
class TagTranslationOptions(TranslationOptions):
    """Class for tag translation options."""

    fields = ('name',)


@register(Registration)
class RegistrationTranslationOptions(TranslationOptions):
    """Class for registration translation options."""

    fields = ('name',)


@register(BaseRole)
class BaseRoleTranslationOptions(TranslationOptions):
    """Class for base role translation options"""

    fields = ('name',)


@register(PresetRole)
class PresetRoleTranslationOptions(TranslationOptions):
    """Class for PresetRole translation options."""

    fields = ('description',)


@register(PresetKaehmyRole)
class PresetKaehmyRoleTranslationOptions(TranslationOptions):
    """ Class for PresetKaehmyRole translation options"""

    fields = ()