from modeltranslation.translator import TranslationOptions, register
from .models import *



@register(Menu)
class MenuranslationOptions(TranslationOptions):
    fields = ('title',)


@register(Hero)
class HeroTranslationOptions(TranslationOptions):
    fields = ('title', 'subtitle', 'description')


@register(About)
class AboutTranslationOptions(TranslationOptions):
    fields = ('description',)


@register(AboutSlide)
class AboutSlideTranslationOptions(TranslationOptions):
    fields = ('title', 'description', 'years_label')


@register(AboutSkill)
class AboutSkillTranslationOptions(TranslationOptions):
    fields = ('title', 'title_akcent', 'title_counter', 'text', 'description')


@register(Movie)
class MovieTranslationOptions(TranslationOptions):
    fields = ('title', 'description')


@register(AudioTrack)
class AudioTrackTranslationOptions(TranslationOptions):
    fields = ('title', 'subtitle')


@register(Testimonial)
class TestimonialTranslationOptions(TranslationOptions):
    fields = ('text', 'author_name', 'author_company')



@register(TeamMember)
class TeamMemberTranslationOptions(TranslationOptions):
    fields = ('name', 'position', 'since')


@register(SiteText)
class SiteTextTranslationOptions(TranslationOptions):
    fields = (
        'about_title',
        'movie_title',
        'sound_title',
        'partners_title',
        'team_title',
        'contact_title',
        'button_about',
        'button_movie',
        'button_sound',
        'button_contact',
        'sound_random',
        'testimonial_text',
        'team_text',
        'contact_textcontact_text',
        'contact_address',
        'contact_form_text',
        'footer_text',
    )