from import_export.resources import ModelResource

from .models import Category, Product


class CategoryResource(ModelResource):
    class Meta:
        model = Category
        fields = ("id", "name", "image", "enabled", "meta_description")


class ProductResource(ModelResource):
    class Meta:
        model = Product
        fields = (
            "id",
            "name",
            "model_name",
            "stock",
            "price",
            "vat",
            "description",
            "meta_description",
            "length",
            "width",
            "height",
            "weight",
            "category",
            "manufacturer",
        )
