from modeltranslation.decorators import register
from modeltranslation.translator import TranslationOptions

from engine.core.models import (
    Attribute,
    AttributeGroup,
    AttributeValue,
    Brand,
    Category,
    CategoryTag,
    Product,
    ProductTag,
    Promotion,
)


@register(AttributeGroup)
class AttributeGroupOptions(TranslationOptions):
    fields = ("name",)


@register(Attribute)
class AttributeOptions(TranslationOptions):
    fields = ("name",)


@register(AttributeValue)
class AttributeValueOptions(TranslationOptions):
    fields = ("value",)


@register(Brand)
class BrandTranslationOptions(TranslationOptions):
    fields = ("description",)


@register(Category)
class CategoryTranslationOptions(TranslationOptions):
    fields = ("name", "description")


@register(Product)
class ProductTranslationOptions(TranslationOptions):
    fields = ("name", "description")


@register(ProductTag)
class ProductTagOptions(TranslationOptions):
    fields = ("name",)


@register(CategoryTag)
class CategoryTagOptions(TranslationOptions):
    fields = ("name",)


@register(Promotion)
class PromotionOptions(TranslationOptions):
    fields = ("name", "description")