Class BerryFlavorDetailSerializer

pokemon_v2/serializers.py:2127–2178

class BerryFlavorDetailSerializer(serializers.ModelSerializer):
    names = BerryFlavorNameSerializer(
        many=True, read_only=True, source="berryflavorname"
    )
    contest_type = ContestTypeSummarySerializer()
    berries = serializers.SerializerMethodField("get_berries_with_flavor")

    class Meta:
        model = BerryFlavor
        fields = ("id", "name", "berries", "contest_type", "names")

    @extend_schema_field(
        field={
            "type": "array",
            "items": {
                "type": "object",
                "required": ["potency", "berry"],
                "properties": {
                    "potency": {"type": "integer", "examples": [10]},
                    "berry": {
                        "type": "object",
                        "require": ["name", "url"],
                        "properties": {
                            "name": {
                                "type": "string",
                                "description": "The name of the berry",
                                "examples": ["rowap"],
                            },
                            "url": {
                                "type": "string",
                                "format": "uri",
                                "description": "The URL to get more information about the berry",
                                "examples": ["https://pokeapi.co/api/v2/berry/64/"],
                            },
                        },
                    },
                },
            },
        }
    )
    def get_berries_with_flavor(self, obj):
        flavor_map_objects = BerryFlavorMap.objects.filter(
            berry_flavor=obj, potency__gt=0
        ).order_by("potency")
        flavor_maps = BerryFlavorMapSerializer(
            flavor_map_objects, many=True, context=self.context
        ).data

        for map in flavor_maps:
            del map["flavor"]

        return flavor_maps


class BerryDetailSerializer(serializers.ModelSerializer):