class CategoryWithImageSerializer(serializers.ModelSerializer):
    image = serializers.ImageField('image')

    class Meta:
        model = Category
        fields = ('name', 'slug', 'image')


class CategoryOverviewSerializer(serializers.ModelSerializer):
    children = CategoryWithCountSerializer(many=True, required=False)

    class Meta:
        model = Category
        fields = ('name', 'slug', 'children')

class CategorySerializer(serializers.ModelSerializer):
    title = serializers.ChoiceField(choices='get_category')

    class Meta:
        model = KBCategory
        fields = ('title', )

    def get_category(self):
        return KBCategory.objects.all()