from rest_framework import serializers

from collection import models


class CollectionSerializer(serializers.ModelSerializer):
    """Collection serializer"""
    class Meta:
        model = models.Collection
        fields = [
            'id',
            'name',
            'block_size',
            'image',
            'on_top'
        ]


class CollectionItemSerializer(serializers.ModelSerializer):
    """CollectionItem serializer"""
    class Meta:
        model = models.CollectionItem
        fields = [
            'collection',
            'item_type',
            'item_ids'
        ]


class GuideSerializer(serializers.ModelSerializer):
    """Guide serializer"""
    class Meta:
        model = models.Guide
        fields = [
            'name',
            'start',
            'end',
            'parent',
            'advertorials',
            'collection'
        ]