from modeltranslation import translator

from core.apps.common import models


@translator.register(models.Banner)
class BannerTranslation(translator.TranslationOptions):
    fields = [
        'title', 'text' 
    ]
    

@translator.register(models.AboutUs)
class AboutUsTranslation(translator.TranslationOptions):
    fields = [
        'title', 'description', 'full_text'
    ]


@translator.register(models.AboutUsFeature)
class AboutUsFeatureTranslation(translator.TranslationOptions):
    fields = [
        'text'
    ]


@translator.register(models.Service)
class ServiceTranslation(translator.TranslationOptions):
    fields = [
        'title', 'text', 'full_text'
    ]


@translator.register(models.News)
class NewsTranslation(translator.TranslationOptions):
    fields = [
        'title', 'text', 'full_text'
    ]


@translator.register(models.Requisite)
class RequisiteTranslation(translator.TranslationOptions):
    fields = [
        'company_name', 'legal_address'
    ]


@translator.register(models.PrivacyPolicy)
class RequisiteTranslation(translator.TranslationOptions):
    fields = [
        'title', 'text'
    ]


@translator.register(models.UserTerms)
class UserTermsTranslation(translator.TranslationOptions):
    fields = [
        'text'
    ]