pokeapi/pokemon_v2/serializers.py
2023-11-08 10:44:55 +01:00

3474 lines
107 KiB
Python

from collections import OrderedDict
import json
from django.urls import reverse
from rest_framework import serializers
# pylint: disable=redefined-builtin
# PokeAPI v2 serializers in order of dependency
from .models import *
#########################
# SUMMARY SERIALIZERS #
#########################
# Summary serializers are just for list and reference behavior
# Putting summary serializers up top so there are no conflicts
# with reference accross models due to script running order
class AbilitySummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Ability
fields = ("name", "url")
class BerryFirmnessSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = BerryFirmness
fields = ("name", "url")
class BerryFlavorSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = BerryFlavor
fields = ("name", "url")
class BerrySummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Berry
fields = ("name", "url")
class CharacteristicSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Characteristic
fields = ("url",)
class ContestEffectSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ContestEffect
fields = ("url",)
class ContestTypeSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ContestType
fields = ("name", "url")
class EggGroupSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EggGroup
fields = ("name", "url")
class EncounterConditionSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EncounterCondition
fields = ("name", "url")
class EncounterConditionValueSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EncounterConditionValue
fields = ("name", "url")
class EncounterMethodSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EncounterMethod
fields = ("name", "url")
class EvolutionTriggerSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EvolutionTrigger
fields = ("name", "url")
class EvolutionChainSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EvolutionChain
fields = ("url",)
class GenerationSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Generation
fields = ("name", "url")
class GenderSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Gender
fields = ("name", "url")
class GrowthRateSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = GrowthRate
fields = ("name", "url")
class ItemPocketSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemPocket
fields = ("name", "url")
class ItemCategorySummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemCategory
fields = ("name", "url")
class ItemAttributeSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemAttribute
fields = ("name", "url")
class ItemFlingEffectSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemFlingEffect
fields = ("name", "url")
class ItemSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Item
fields = ("name", "url")
class LanguageSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Language
fields = ("name", "url")
class LocationSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Location
fields = ("name", "url")
class LocationAreaSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = LocationArea
fields = ("name", "url")
class MachineSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Machine
fields = ("url",)
class MoveBattleStyleSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = MoveBattleStyle
fields = ("name", "url")
class MoveDamageClassSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = MoveDamageClass
fields = ("name", "url")
class MoveMetaAilmentSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = MoveMetaAilment
fields = ("name", "url")
class MoveMetaCategorySummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = MoveMetaCategory
fields = ("name", "url")
class MoveTargetSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = MoveTarget
fields = ("name", "url")
class MoveSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Move
fields = ("name", "url")
class MoveLearnMethodSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = MoveLearnMethod
fields = ("name", "url")
class NatureSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Nature
fields = ("name", "url")
class PalParkAreaSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PalParkArea
fields = ("name", "url")
class PokeathlonStatSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokeathlonStat
fields = ("name", "url")
class PokedexSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Pokedex
fields = ("name", "url")
class PokemonColorSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonColor
fields = ("name", "url")
class PokemonHabitatSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonHabitat
fields = ("name", "url")
class PokemonShapeSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonShape
fields = ("name", "url")
class PokemonSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Pokemon
fields = ("name", "url")
class PokemonSpeciesSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonSpecies
fields = ("name", "url")
class PokemonFormSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonForm
fields = ("name", "url")
class RegionSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Region
fields = ("name", "url")
class StatSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Stat
fields = ("name", "url")
class SuperContestEffectSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = SuperContestEffect
fields = ("url",)
class TypeSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Type
fields = ("name", "url")
class VersionSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Version
fields = ("name", "url")
class VersionGroupSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = VersionGroup
fields = ("name", "url")
#####################
# MAP SERIALIZERS #
#####################
class BerryFlavorMapSerializer(serializers.ModelSerializer):
berry = BerrySummarySerializer()
flavor = BerryFlavorSummarySerializer(source="berry_flavor")
class Meta:
model = BerryFlavorMap
fields = ("potency", "berry", "flavor")
class ItemAttributeMapSerializer(serializers.ModelSerializer):
item = ItemSummarySerializer()
attribute = ItemAttributeSummarySerializer(source="item_attribute")
class Meta:
model = ItemAttributeMap
fields = (
"item",
"attribute",
)
class MoveMetaStatChangeSerializer(serializers.ModelSerializer):
stat = StatSummarySerializer()
move = MoveSummarySerializer()
class Meta:
model = MoveMetaStatChange
fields = ("change", "move", "stat")
class NaturePokeathlonStatSerializer(serializers.ModelSerializer):
pokeathlon_stat = PokeathlonStatSummarySerializer()
nature = NatureSummarySerializer()
class Meta:
model = NaturePokeathlonStat
fields = ("max_change", "nature", "pokeathlon_stat")
class PokemonAbilitySerializer(serializers.ModelSerializer):
pokemon = PokemonSummarySerializer()
ability = AbilitySummarySerializer()
class Meta:
model = PokemonAbility
fields = ("is_hidden", "slot", "ability", "pokemon")
class PokemonAbilityPastSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
ability = AbilitySummarySerializer()
class Meta:
model = PokemonAbilityPast
fields = ("is_hidden", "pokemon", "generation", "slot", "ability")
class PokemonDexEntrySerializer(serializers.ModelSerializer):
entry_number = serializers.IntegerField(source="pokedex_number")
pokedex = PokedexSummarySerializer()
class Meta:
model = PokemonDexNumber
fields = ("entry_number", "pokedex")
class PokemonTypeSerializer(serializers.ModelSerializer):
pokemon = PokemonSummarySerializer()
type = TypeSummarySerializer()
class Meta:
model = PokemonType
fields = ("slot", "pokemon", "type")
class PokemonFormTypeSerializer(serializers.ModelSerializer):
pokemon_form = PokemonFormSummarySerializer()
type = TypeSummarySerializer()
class Meta:
model = PokemonFormType
fields = ("slot", "pokemon_form", "type")
class PokemonTypePastSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
type = TypeSummarySerializer()
class Meta:
model = PokemonTypePast
fields = ("pokemon", "generation", "slot", "type")
class PokedexVersionGroupSerializer(serializers.ModelSerializer):
pokedex = PokedexSummarySerializer()
version_group = VersionGroupSummarySerializer()
class Meta:
model = PokedexVersionGroup
fields = ("pokedex", "version_group")
class VersionGroupMoveLearnMethodSerializer(serializers.ModelSerializer):
version_group = VersionGroupSummarySerializer()
move_learn_method = MoveLearnMethodSummarySerializer()
class Meta:
model = ItemAttributeMap
fields = ("version_group", "move_learn_method")
class VersionGroupRegionSerializer(serializers.ModelSerializer):
version_group = VersionGroupSummarySerializer()
region = RegionSummarySerializer()
class Meta:
model = ItemAttributeMap
fields = (
"version_group",
"region",
)
class EncounterConditionValueMapSerializer(serializers.ModelSerializer):
condition_value = EncounterConditionValueSummarySerializer(
source="encounter_condition_value"
)
class Meta:
model = EncounterConditionValueMap
fields = ("condition_value",)
################################
# CHARACTERISTIC SERIALIZERS #
################################
class CharacteristicDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = CharacteristicDescription
fields = ("description", "language")
class CharacteristicDetailSerializer(serializers.ModelSerializer):
descriptions = CharacteristicDescriptionSerializer(
many=True, read_only=True, source="characteristicdescription"
)
highest_stat = StatSummarySerializer(source="stat")
gene_modulo = serializers.IntegerField(source="gene_mod_5")
possible_values = serializers.SerializerMethodField("get_values")
class Meta:
model = Characteristic
fields = (
"id",
"gene_modulo",
"possible_values",
"highest_stat",
"descriptions",
)
def get_values(self, obj):
mod = obj.gene_mod_5
values = []
while mod <= 31:
values.append(mod)
mod += 5
return values
#########################
# CONTEST SERIALIZERS #
#########################
class SuperContestEffectFlavorTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = SuperContestEffectFlavorText
fields = ("flavor_text", "language")
class SuperContestEffectDetailSerializer(serializers.ModelSerializer):
flavor_text_entries = SuperContestEffectFlavorTextSerializer(
many=True, read_only=True, source="supercontesteffectflavortext"
)
moves = MoveSummarySerializer(many=True, read_only=True, source="move")
class Meta:
model = SuperContestEffect
fields = ("id", "appeal", "flavor_text_entries", "moves")
class ContestEffectEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ContestEffectEffectText
fields = ("effect", "language")
class ContestEffectFlavorTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ContestEffectFlavorText
fields = ("flavor_text", "language")
class ContestEffectDetailSerializer(serializers.ModelSerializer):
effect_entries = ContestEffectEffectTextSerializer(
many=True, read_only=True, source="contesteffecteffecttext"
)
flavor_text_entries = ContestEffectFlavorTextSerializer(
many=True, read_only=True, source="contesteffectflavortext"
)
class Meta:
model = ContestEffect
fields = ("id", "appeal", "jam", "effect_entries", "flavor_text_entries")
class ContestTypeNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ContestTypeName
fields = ("name", "color", "language")
class ContestTypeDetailSerializer(serializers.ModelSerializer):
names = ContestTypeNameSerializer(
many=True, read_only=True, source="contesttypename"
)
berry_flavor = BerryFlavorSummarySerializer(read_only=True, source="berryflavor")
class Meta:
model = ContestType
fields = ("id", "name", "berry_flavor", "names")
class SuperContestComboSerializer(serializers.ModelSerializer):
first_move = MoveSummarySerializer()
second_move = MoveSummarySerializer()
class Meta:
model = SuperContestCombo
fields = ("first_move", "second_move")
class ContestComboSerializer(serializers.ModelSerializer):
first_move = MoveSummarySerializer()
second_move = MoveSummarySerializer()
class Meta:
model = ContestCombo
fields = ("first_move", "second_move")
########################
# REGION SERIALIZERS #
########################
class RegionNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = RegionName
fields = ("name", "language")
class RegionDetailSerializer(serializers.ModelSerializer):
names = RegionNameSerializer(many=True, read_only=True, source="regionname")
locations = LocationSummarySerializer(many=True, read_only=True, source="location")
version_groups = serializers.SerializerMethodField("get_region_version_groups")
pokedexes = PokedexSummarySerializer(many=True, read_only=True, source="pokedex")
main_generation = GenerationSummarySerializer(read_only=True, source="generation")
class Meta:
model = Region
fields = (
"id",
"name",
"locations",
"main_generation",
"names",
"pokedexes",
"version_groups",
)
def get_region_version_groups(self, obj):
vg_regions = VersionGroupRegion.objects.filter(region=obj)
data = VersionGroupRegionSerializer(
vg_regions, many=True, context=self.context
).data
groups = []
for group in data:
groups.append(group["version_group"])
return groups
############################
# GENERATION SERIALIZERS #
############################
class GenerationNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = GenerationName
fields = ("name", "language")
class GenerationDetailSerializer(serializers.ModelSerializer):
main_region = RegionSummarySerializer(source="region")
names = GenerationNameSerializer(many=True, read_only=True, source="generationname")
abilities = AbilitySummarySerializer(many=True, read_only=True, source="ability")
moves = MoveSummarySerializer(many=True, read_only=True, source="move")
pokemon_species = PokemonSpeciesSummarySerializer(
many=True, read_only=True, source="pokemonspecies"
)
types = TypeSummarySerializer(many=True, read_only=True, source="type")
version_groups = VersionGroupSummarySerializer(
many=True, read_only=True, source="versiongroup"
)
class Meta:
model = Generation
fields = (
"id",
"name",
"abilities",
"main_region",
"moves",
"names",
"pokemon_species",
"types",
"version_groups",
)
########################
# GENDER SERIALIZERS #
########################
class GenderDetailSerializer(serializers.ModelSerializer):
pokemon_species_details = serializers.SerializerMethodField("get_species")
required_for_evolution = serializers.SerializerMethodField("get_required")
class Meta:
model = Gender
fields = ("id", "name", "pokemon_species_details", "required_for_evolution")
def get_species(self, obj):
species_objects = []
if obj.name == "female":
species_objects = PokemonSpecies.objects.filter(gender_rate__gt=0)
elif obj.name == "male":
species_objects = PokemonSpecies.objects.filter(gender_rate__range=[0, 7])
elif obj.name == "genderless":
species_objects = PokemonSpecies.objects.filter(gender_rate=-1)
details = []
for species in species_objects:
detail = OrderedDict()
detail["rate"] = species.gender_rate
detail["pokemon_species"] = PokemonSpeciesSummarySerializer(
species, context=self.context
).data
details.append(detail)
return details
def get_required(self, obj):
evo_objects = PokemonEvolution.objects.filter(gender=obj)
species_list = []
for evo in evo_objects:
species = PokemonSpeciesSummarySerializer(
evo.evolved_species, context=self.context
).data
species_list.append(species)
return species_list
#############################
# GROWTH RATE SERIALIZERS #
#############################
class ExperienceSerializer(serializers.ModelSerializer):
class Meta:
model = Experience
fields = ("level", "experience")
class GrowthRateDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = GrowthRateDescription
fields = ("description", "language")
class GrowthRateDetailSerializer(serializers.ModelSerializer):
descriptions = GrowthRateDescriptionSerializer(
many=True, read_only=True, source="growthratedescription"
)
levels = ExperienceSerializer(many=True, read_only=True, source="experience")
pokemon_species = PokemonSpeciesSummarySerializer(
many=True, read_only=True, source="pokemonspecies"
)
class Meta:
model = GrowthRate
fields = ("id", "name", "formula", "descriptions", "levels", "pokemon_species")
##########################
# LANGUAGE SERIALIZERS #
##########################
class LanguageNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer(source="local_language")
class Meta:
model = LanguageName
fields = ("name", "language")
class LanguageDetailSerializer(serializers.ModelSerializer):
names = LanguageNameSerializer(
many=True, read_only=True, source="languagename_language"
)
class Meta:
model = Language
fields = ("id", "name", "official", "iso639", "iso3166", "names")
########################################
# LOCATION AND ENCOUNTER SERIALIZERS #
########################################
class EncounterConditionNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = EncounterConditionName
fields = ("name", "language")
class EncounterConditionDetailSerializer(serializers.ModelSerializer):
names = EncounterConditionNameSerializer(
many=True, read_only=True, source="encounterconditionname"
)
values = EncounterConditionValueSummarySerializer(
many=True, read_only=True, source="encounterconditionvalue"
)
class Meta:
model = EncounterCondition
fields = ("id", "name", "values", "names")
class EncounterConditionValueNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = EncounterConditionValueName
fields = ("name", "language")
class EncounterConditionValueDetailSerializer(serializers.ModelSerializer):
condition = EncounterConditionSummarySerializer(source="encounter_condition")
names = EncounterConditionValueNameSerializer(
many=True, read_only=True, source="encounterconditionvaluename"
)
class Meta:
model = EncounterConditionValue
fields = ("id", "name", "condition", "names")
class EncounterMethodNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = EncounterMethodName
fields = ("name", "language")
class EncounterMethodDetailSerializer(serializers.ModelSerializer):
names = EncounterMethodNameSerializer(
many=True, read_only=True, source="encountermethodname"
)
class Meta:
model = EncounterMethod
fields = ("id", "name", "order", "names")
class EncounterSlotSerializer(serializers.ModelSerializer):
encounter_method = EncounterMethodSummarySerializer()
chance = serializers.IntegerField(source="rarity")
class Meta:
model = EncounterSlot
fields = ("id", "slot", "chance", "encounter_method", "version_group")
class EncounterDetailSerializer(serializers.ModelSerializer):
version = VersionSummarySerializer()
location_area = LocationAreaSummarySerializer()
pokemon = PokemonSummarySerializer()
condition_values = serializers.SerializerMethodField("get_encounter_conditions")
class Meta:
model = Encounter
fields = (
"min_level",
"max_level",
"version",
"encounter_slot",
"pokemon",
"location_area",
"condition_values",
)
def get_encounter_conditions(self, obj):
condition_values = EncounterConditionValueMap.objects.filter(encounter=obj)
data = EncounterConditionValueMapSerializer(
condition_values, many=True, context=self.context
).data
values = []
for map in data:
values.append(map["condition_value"])
return values
class LocationAreaEncounterRateSerializer(serializers.ModelSerializer):
encounter_method = EncounterMethodSummarySerializer()
version = VersionSummarySerializer()
class Meta:
model = LocationAreaEncounterRate
fields = ("rate", "encounter_method", "version")
class LocationAreaNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = LocationAreaName
fields = ("name", "language")
class LocationAreaDetailSerializer(serializers.ModelSerializer):
location = LocationSummarySerializer()
encounter_method_rates = serializers.SerializerMethodField("get_method_rates")
pokemon_encounters = serializers.SerializerMethodField("get_encounters")
names = LocationAreaNameSerializer(
many=True, read_only=True, source="locationareaname"
)
class Meta:
model = LocationArea
fields = (
"id",
"name",
"game_index",
"encounter_method_rates",
"location",
"names",
"pokemon_encounters",
)
def get_method_rates(self, obj):
# Get encounters related to this area and pull out unique encounter methods
encounter_rates = LocationAreaEncounterRate.objects.filter(
location_area=obj
).order_by("encounter_method_id")
method_ids = encounter_rates.values("encounter_method_id").distinct()
encounter_rate_list = []
for id in method_ids:
encounter_rate_details = OrderedDict()
# Get each Unique Item by ID
encounter_method_object = EncounterMethod.objects.get(
pk=id["encounter_method_id"]
)
encounter_method_data = EncounterMethodSummarySerializer(
encounter_method_object, context=self.context
).data
encounter_rate_details["encounter_method"] = encounter_method_data
# Get Versions associated with each unique item
area_encounter_objects = encounter_rates.filter(
encounter_method_id=id["encounter_method_id"]
)
serializer = LocationAreaEncounterRateSerializer(
area_encounter_objects, many=True, context=self.context
)
encounter_rate_details["version_details"] = []
for area_encounter in serializer.data:
version_detail = OrderedDict()
version_detail["rate"] = area_encounter["rate"]
version_detail["version"] = area_encounter["version"]
encounter_rate_details["version_details"].append(version_detail)
encounter_rate_list.append(encounter_rate_details)
return encounter_rate_list
def get_encounters(self, obj):
# get versions for later use
version_objects = Version.objects.all()
version_data = VersionSummarySerializer(
version_objects, many=True, context=self.context
).data
# all encounters associated with location area
all_encounters = Encounter.objects.filter(location_area=obj).order_by("pokemon")
encounters_list = []
# break encounters into pokemon groupings
for poke in all_encounters.values("pokemon").distinct():
pokemon_object = Pokemon.objects.get(pk=poke["pokemon"])
pokemon_detail = OrderedDict()
pokemon_detail["pokemon"] = PokemonSummarySerializer(
pokemon_object, context=self.context
).data
pokemon_detail["version_details"] = []
poke_encounters = all_encounters.filter(pokemon=poke["pokemon"]).order_by(
"version"
)
# each pokemon has multiple versions it could be encountered in
for ver in poke_encounters.values("version").distinct():
version_detail = OrderedDict()
version_detail["version"] = version_data[ver["version"] - 1]
version_detail["max_chance"] = 0
version_detail["encounter_details"] = []
poke_data = EncounterDetailSerializer(
poke_encounters.filter(version=ver["version"]),
many=True,
context=self.context,
).data
# each version has multiple ways a pokemon can be encountered
for encounter in poke_data:
slot = EncounterSlot.objects.get(pk=encounter["encounter_slot"])
slot_data = EncounterSlotSerializer(slot, context=self.context).data
del encounter["pokemon"]
del encounter["encounter_slot"]
del encounter["location_area"]
del encounter["version"]
encounter["chance"] = slot_data["chance"]
version_detail["max_chance"] += slot_data["chance"]
encounter["method"] = slot_data["encounter_method"]
version_detail["encounter_details"].append(encounter)
pokemon_detail["version_details"].append(version_detail)
encounters_list.append(pokemon_detail)
return encounters_list
class LocationGameIndexSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
class Meta:
model = LocationGameIndex
fields = ("game_index", "generation")
class LocationNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = LocationName
fields = ("name", "language")
class LocationDetailSerializer(serializers.ModelSerializer):
region = RegionSummarySerializer()
names = LocationNameSerializer(many=True, read_only=True, source="locationname")
game_indices = LocationGameIndexSerializer(
many=True, read_only=True, source="locationgameindex"
)
areas = LocationAreaSummarySerializer(
many=True, read_only=True, source="locationarea"
)
class Meta:
model = Location
fields = ("id", "name", "region", "names", "game_indices", "areas")
#########################
# ABILITY SERIALIZERS #
#########################
class AbilityEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = AbilityEffectText
fields = ("effect", "short_effect", "language")
class AbilityFlavorTextSerializer(serializers.ModelSerializer):
flavor_text = serializers.CharField()
language = LanguageSummarySerializer()
version_group = VersionGroupSummarySerializer()
class Meta:
model = AbilityFlavorText
fields = ("flavor_text", "language", "version_group")
class AbilityChangeEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = AbilityChangeEffectText
fields = (
"effect",
"language",
)
class AbilityChangeSerializer(serializers.ModelSerializer):
version_group = VersionGroupSummarySerializer()
effect_entries = AbilityChangeEffectTextSerializer(
many=True, read_only=True, source="abilitychangeeffecttext"
)
class Meta:
model = AbilityChange
fields = ("version_group", "effect_entries")
class AbilityNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = AbilityName
fields = ("name", "language")
class AbilityDetailSerializer(serializers.ModelSerializer):
effect_entries = AbilityEffectTextSerializer(
many=True, read_only=True, source="abilityeffecttext"
)
flavor_text_entries = AbilityFlavorTextSerializer(
many=True, read_only=True, source="abilityflavortext"
)
names = AbilityNameSerializer(many=True, read_only=True, source="abilityname")
generation = GenerationSummarySerializer()
effect_changes = AbilityChangeSerializer(
many=True, read_only=True, source="abilitychange"
)
pokemon = serializers.SerializerMethodField("get_ability_pokemon")
class Meta:
model = Ability
fields = (
"id",
"name",
"is_main_series",
"generation",
"names",
"effect_entries",
"effect_changes",
"flavor_text_entries",
"pokemon",
)
def get_ability_pokemon(self, obj):
pokemon_ability_objects = PokemonAbility.objects.filter(ability=obj)
data = PokemonAbilitySerializer(
pokemon_ability_objects, many=True, context=self.context
).data
pokemon = []
for poke in data:
del poke["ability"]
pokemon.append(poke)
return pokemon
######################
# STAT SERIALIZERS #
######################
class StatNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = StatName
fields = ("name", "language")
class StatDetailSerializer(serializers.ModelSerializer):
names = StatNameSerializer(many=True, read_only=True, source="statname")
move_damage_class = MoveDamageClassSummarySerializer()
characteristics = CharacteristicSummarySerializer(
many=True, read_only=True, source="characteristic"
)
affecting_moves = serializers.SerializerMethodField("get_moves_that_affect")
affecting_natures = serializers.SerializerMethodField("get_natures_that_affect")
class Meta:
model = Stat
fields = (
"id",
"name",
"game_index",
"is_battle_only",
"affecting_moves",
"affecting_natures",
"characteristics",
"move_damage_class",
"names",
)
def get_moves_that_affect(self, obj):
stat_change_objects = MoveMetaStatChange.objects.filter(stat=obj)
stat_changes = MoveMetaStatChangeSerializer(
stat_change_objects, many=True, context=self.context
).data
changes = OrderedDict([("increase", []), ("decrease", [])])
for change in stat_changes:
del change["stat"]
if change["change"] > 0:
changes["increase"].append(change)
else:
changes["decrease"].append(change)
return changes
def get_natures_that_affect(self, obj):
increase_objects = Nature.objects.filter(increased_stat=obj)
increases = NatureSummarySerializer(
increase_objects, many=True, context=self.context
).data
decrease_objects = Nature.objects.filter(decreased_stat=obj)
decreases = NatureSummarySerializer(
decrease_objects, many=True, context=self.context
).data
return OrderedDict([("increase", increases), ("decrease", decreases)])
#############################
# ITEM POCKET SERIALIZERS #
#############################
class ItemPocketNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemName
fields = ("name", "language")
class ItemPocketDetailSerializer(serializers.ModelSerializer):
names = ItemPocketNameSerializer(many=True, read_only=True, source="itempocketname")
categories = ItemCategorySummarySerializer(
many=True, read_only=True, source="itemcategory"
)
class Meta:
model = ItemPocket
fields = ("id", "name", "categories", "categories", "names")
###############################
# ITEM CATEGORY SERIALIZERS #
###############################
class ItemCategoryNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemName
fields = ("name", "language")
class ItemCategoryDetailSerializer(serializers.ModelSerializer):
names = ItemCategoryNameSerializer(
many=True, read_only=True, source="itemcategoryname"
)
pocket = ItemPocketSummarySerializer(source="item_pocket")
items = ItemSummarySerializer(many=True, read_only=True, source="item")
class Meta:
model = ItemCategory
fields = ("id", "name", "items", "names", "pocket")
################################
# ITEM ATTRIBUTE SERIALIZERS #
################################
class ItemAttributeNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemAttributeName
fields = ("name", "language")
class ItemAttributeDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemAttributeDescription
fields = ("description", "language")
class ItemAttributeDetailSerializer(serializers.ModelSerializer):
names = ItemAttributeNameSerializer(
many=True, read_only=True, source="itemattributename"
)
descriptions = ItemAttributeDescriptionSerializer(
many=True, read_only=True, source="itemattributedescription"
)
items = serializers.SerializerMethodField("get_attribute_items")
class Meta:
model = ItemAttribute
fields = ("id", "name", "descriptions", "items", "names")
def get_attribute_items(self, obj):
item_map_objects = ItemAttributeMap.objects.filter(item_attribute=obj)
items = []
for map in item_map_objects:
item_obj = Item.objects.get(pk=map.item.id)
item = ItemSummarySerializer(item_obj, context=self.context).data
items.append(item)
return items
###################################
# ITEM FLING EFFECT SERIALIZERS #
###################################
class ItemFlingEffectEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemFlingEffectEffectText
fields = ("effect", "language")
class ItemFlingEffectDetailSerializer(serializers.ModelSerializer):
effect_entries = ItemFlingEffectEffectTextSerializer(
many=True, read_only=True, source="itemflingeffecteffecttext"
)
items = ItemSummarySerializer(many=True, read_only=True, source="item")
class Meta:
model = ItemFlingEffect
fields = ("id", "name", "effect_entries", "items")
#######################
# ITEM SERIALIZERS #
#######################
class ItemFlavorTextSerializer(serializers.ModelSerializer):
text = serializers.CharField(source="flavor_text")
language = LanguageSummarySerializer()
version_group = VersionGroupSummarySerializer()
class Meta:
model = ItemFlavorText
fields = ("text", "version_group", "language")
class ItemEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemEffectText
fields = ("effect", "short_effect", "language")
class ItemGameIndexSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
class Meta:
model = ItemGameIndex
fields = ("game_index", "generation")
class ItemNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemName
fields = ("name", "language")
class ItemSpritesSerializer(serializers.ModelSerializer):
class Meta:
model = ItemSprites
fields = ("sprites",)
class ItemDetailSerializer(serializers.ModelSerializer):
names = ItemNameSerializer(many=True, read_only=True, source="itemname")
game_indices = ItemGameIndexSerializer(
many=True, read_only=True, source="itemgameindex"
)
effect_entries = ItemEffectTextSerializer(
many=True, read_only=True, source="itemeffecttext"
)
flavor_text_entries = ItemFlavorTextSerializer(
many=True, read_only=True, source="itemflavortext"
)
category = ItemCategorySummarySerializer(source="item_category")
attributes = serializers.SerializerMethodField("get_item_attributes")
fling_effect = ItemFlingEffectSummarySerializer(source="item_fling_effect")
held_by_pokemon = serializers.SerializerMethodField(source="get_held_by_pokemon")
baby_trigger_for = serializers.SerializerMethodField(source="get_baby_trigger_for")
sprites = serializers.SerializerMethodField("get_item_sprites")
machines = serializers.SerializerMethodField("get_item_machines")
class Meta:
model = Item
fields = (
"id",
"name",
"cost",
"fling_power",
"fling_effect",
"attributes",
"category",
"effect_entries",
"flavor_text_entries",
"game_indices",
"names",
"held_by_pokemon",
"sprites",
"baby_trigger_for",
"machines",
)
def get_item_machines(self, obj):
machine_objects = Machine.objects.filter(item=obj)
machines = []
for machine_object in machine_objects:
machine_data = MachineSummarySerializer(
machine_object, context=self.context
).data
version_group_data = VersionGroupSummarySerializer(
machine_object.version_group, context=self.context
).data
machines.append(
{"machine": machine_data, "version_group": version_group_data}
)
return machines
def get_item_sprites(self, obj):
sprites_object = ItemSprites.objects.get(item_id=obj)
sprites_data = ItemSpritesSerializer(sprites_object, context=self.context).data
sprites_data = json.loads(sprites_data["sprites"])
return sprites_data
def get_item_attributes(self, obj):
item_attribute_maps = ItemAttributeMap.objects.filter(item=obj)
serializer = ItemAttributeMapSerializer(
item_attribute_maps, many=True, context=self.context
)
data = serializer.data
attributes = []
for map in data:
attribute = OrderedDict()
attribute["name"] = map["attribute"]["name"]
attribute["url"] = map["attribute"]["url"]
attributes.append(attribute)
return attributes
def get_held_by_pokemon(self, obj):
pokemon_items = PokemonItem.objects.filter(item=obj).order_by("pokemon_id")
pokemon_ids = pokemon_items.values("pokemon_id").distinct()
pokemon_list = []
for id in pokemon_ids:
item_pokemon_details = OrderedDict()
# Get each Unique Item by ID
pokemon_object = Pokemon.objects.get(pk=id["pokemon_id"])
pokemon_data = PokemonSummarySerializer(
pokemon_object, context=self.context
).data
item_pokemon_details["pokemon"] = pokemon_data
# Get Versions associated with each unique item
pokemon_item_objects = pokemon_items.filter(pokemon_id=id["pokemon_id"])
serializer = PokemonItemSerializer(
pokemon_item_objects, many=True, context=self.context
)
item_pokemon_details["version_details"] = []
for pokemon in serializer.data:
version_detail = OrderedDict()
version_detail["rarity"] = pokemon["rarity"]
version_detail["version"] = pokemon["version"]
item_pokemon_details["version_details"].append(version_detail)
pokemon_list.append(item_pokemon_details)
return pokemon_list
def get_baby_trigger_for(self, obj):
try:
chain_object = EvolutionChain.objects.get(baby_trigger_item=obj)
data = EvolutionChainSummarySerializer(
chain_object, context=self.context
).data
except EvolutionChain.DoesNotExist:
data = None
return data
########################
# NATURE SERIALIZERS #
########################
class NatureBattleStylePreferenceSerializer(serializers.ModelSerializer):
move_battle_style = MoveBattleStyleSummarySerializer()
class Meta:
model = NatureBattleStylePreference
fields = (
"low_hp_preference",
"high_hp_preference",
"move_battle_style",
)
class NatureNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = NatureName
fields = ("name", "language")
class NatureDetailSerializer(serializers.ModelSerializer):
names = NatureNameSerializer(many=True, read_only=True, source="naturename")
decreased_stat = StatSummarySerializer()
increased_stat = StatSummarySerializer()
likes_flavor = BerryFlavorSummarySerializer()
hates_flavor = BerryFlavorSummarySerializer()
berries = BerrySummarySerializer(many=True, read_only=True, source="berry")
pokeathlon_stat_changes = serializers.SerializerMethodField("get_pokeathlon_stats")
move_battle_style_preferences = NatureBattleStylePreferenceSerializer(
many=True, read_only=True, source="naturebattlestylepreference"
)
class Meta:
model = Nature
fields = (
"id",
"name",
"decreased_stat",
"increased_stat",
"likes_flavor",
"hates_flavor",
"berries",
"pokeathlon_stat_changes",
"move_battle_style_preferences",
"names",
)
def get_pokeathlon_stats(self, obj):
pokeathlon_stat_objects = NaturePokeathlonStat.objects.filter(nature=obj)
pokeathlon_stats = NaturePokeathlonStatSerializer(
pokeathlon_stat_objects, many=True, context=self.context
).data
for stat in pokeathlon_stats:
del stat["nature"]
return pokeathlon_stats
#######################
# BERRY SERIALIZERS #
#######################
class BerryFirmnessNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = BerryFirmnessName
fields = ("name", "language")
class BerryFirmnessDetailSerializer(serializers.ModelSerializer):
names = BerryFirmnessNameSerializer(
many=True, read_only=True, source="berryfirmnessname"
)
berries = BerrySummarySerializer(many=True, read_only=True, source="berry")
class Meta:
model = BerryFirmness
fields = ("id", "name", "berries", "names")
class BerryFlavorNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = BerryFlavorName
fields = ("name", "language")
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")
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):
item = ItemSummarySerializer()
natural_gift_type = TypeSummarySerializer()
firmness = BerryFirmnessSummarySerializer(source="berry_firmness")
flavors = serializers.SerializerMethodField("get_berry_flavors")
class Meta:
model = Berry
fields = (
"id",
"name",
"growth_time",
"max_harvest",
"natural_gift_power",
"size",
"smoothness",
"soil_dryness",
"firmness",
"flavors",
"item",
"natural_gift_type",
)
def get_berry_flavors(self, obj):
flavor_map_objects = BerryFlavorMap.objects.filter(berry=obj)
flavor_maps = BerryFlavorMapSerializer(
flavor_map_objects, many=True, context=self.context
).data
flavors = []
for map in flavor_maps:
del map["berry"]
flavors.append(map)
return flavors
###########################
# EGG GROUP SERIALIZERS #
###########################
class PokemonEggGroupSerializer(serializers.ModelSerializer):
species = PokemonSpeciesSummarySerializer(source="pokemon_species")
egg_group = EggGroupSummarySerializer()
class Meta:
model = PokemonEggGroup
fields = ("species", "egg_group")
class EggGroupNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = EggGroupName
fields = ("name", "language")
class EggGroupDetailSerializer(serializers.ModelSerializer):
names = EggGroupNameSerializer(many=True, read_only=True, source="egggroupname")
pokemon_species = serializers.SerializerMethodField("get_species")
class Meta:
model = EggGroup
fields = ("id", "name", "names", "pokemon_species")
def get_species(self, obj):
results = PokemonEggGroup.objects.filter(egg_group=obj)
data = PokemonEggGroupSerializer(results, many=True, context=self.context).data
associated_species = []
for species in data:
associated_species.append(species["species"])
return associated_species
######################
# TYPE SERIALIZERS #
######################
# https://stackoverflow.com/a/45987450/3482533
class TypeEfficacySerializer(serializers.ModelSerializer):
class Meta:
model = TypeEfficacy
fields = "__all__"
class TypeEfficacyPastSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
class Meta:
model = TypeEfficacyPast
fields = ("target_type", "damage_type", "damage_factor", "generation")
class TypeGameIndexSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
class Meta:
model = TypeGameIndex
fields = ("game_index", "generation")
class TypeNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = TypeName
fields = ("name", "language")
class TypeDetailSerializer(serializers.ModelSerializer):
"""
Serializer for the Type resource
"""
generation = GenerationSummarySerializer()
names = AbilityNameSerializer(many=True, read_only=True, source="typename")
game_indices = TypeGameIndexSerializer(
many=True, read_only=True, source="typegameindex"
)
move_damage_class = MoveDamageClassSummarySerializer()
damage_relations = serializers.SerializerMethodField("get_type_relationships")
past_damage_relations = serializers.SerializerMethodField(
"get_type_past_relationships"
)
pokemon = serializers.SerializerMethodField("get_type_pokemon")
moves = MoveSummarySerializer(many=True, read_only=True, source="move")
class Meta:
model = Type
fields = (
"id",
"name",
"damage_relations",
"past_damage_relations",
"game_indices",
"generation",
"move_damage_class",
"names",
"pokemon",
"moves",
)
# adds an entry for the given type with the given damage
# factor in the given direction to the set of relations
def add_type_entry(self, relations, type, damage_factor, direction="_damage_to"):
if damage_factor == 200:
relations["double" + direction].append(
TypeSummarySerializer(type, context=self.context).data
)
elif damage_factor == 50:
relations["half" + direction].append(
TypeSummarySerializer(type, context=self.context).data
)
elif damage_factor == 0:
relations["no" + direction].append(
TypeSummarySerializer(type, context=self.context).data
)
def get_type_relationships(self, obj):
relations = OrderedDict()
relations["no_damage_to"] = []
relations["half_damage_to"] = []
relations["double_damage_to"] = []
relations["no_damage_from"] = []
relations["half_damage_from"] = []
relations["double_damage_from"] = []
# Damage To
results = TypeEfficacy.objects.filter(damage_type=obj)
serializer = TypeEfficacySerializer(results, many=True, context=self.context)
for relation in serializer.data:
type = Type.objects.get(pk=relation["target_type"])
damage_factor = relation["damage_factor"]
self.add_type_entry(relations, type, damage_factor, direction="_damage_to")
# Damage From
results = TypeEfficacy.objects.filter(target_type=obj)
serializer = TypeEfficacySerializer(results, many=True, context=self.context)
for relation in serializer.data:
type = Type.objects.get(pk=relation["damage_type"])
damage_factor = relation["damage_factor"]
self.add_type_entry(
relations, type, damage_factor, direction="_damage_from"
)
return relations
# takes a list of past type relations by generation and
# returns a list of lists where each list has the entries
# for a single generation
def group_relations_by_generation(self, serializer_data):
data_by_gen = []
current_generation = ""
generation_data = []
for relation in serializer_data:
gen_name = relation["generation"]["name"]
if gen_name != current_generation:
# first item for this generation so create its list
current_generation = gen_name
generation_data = [relation]
data_by_gen.append(generation_data)
else:
# add to this generation's list
generation_data.append(relation)
return data_by_gen
# removes the entry for the given type in
# the given direction from the set of relations
def remove_type_entry(self, relations, type, direction="_damage_to"):
for k in ["double", "half", "no"]:
rel_list = relations[k + direction]
for i, o in enumerate(rel_list):
if o["name"] == type.name:
del rel_list[i]
return
# returns past type relationships for the given type object
def get_type_past_relationships(self, obj):
# collect data from DB
damage_type_results = list(TypeEfficacyPast.objects.filter(damage_type=obj))
target_type_results = list(TypeEfficacyPast.objects.filter(target_type=obj))
serializer = TypeEfficacyPastSerializer(
damage_type_results + target_type_results, many=True, context=self.context
)
# group data by generation
data_by_gen = self.group_relations_by_generation(serializer.data)
# process each generation's data in turn
final_data = []
past_relations = {}
for gen_data in data_by_gen:
# create past relations object for this generation
past_relations = OrderedDict()
# set generation
past_relations["generation"] = gen_data[0]["generation"]
# use current damage relations object
past_relations["damage_relations"] = self.get_type_relationships(obj)
relations = past_relations["damage_relations"]
current_gen = Generation.objects.get(name=gen_data[0]["generation"]["name"])
# remove types not yet introduced
# e.g. Poison has no effect on Steel, but Steel was not present in generation I
# so it should be absent from the list
relations["no_damage_to"] = self.remove_newer_types(
relations["no_damage_to"], current_gen
)
relations["half_damage_to"] = self.remove_newer_types(
relations["half_damage_to"], current_gen
)
relations["double_damage_to"] = self.remove_newer_types(
relations["double_damage_to"], current_gen
)
relations["no_damage_from"] = self.remove_newer_types(
relations["no_damage_from"], current_gen
)
relations["half_damage_from"] = self.remove_newer_types(
relations["half_damage_from"], current_gen
)
relations["double_damage_from"] = self.remove_newer_types(
relations["double_damage_from"], current_gen
)
# populate offensive relations
results = list(filter(lambda x: x["damage_type"] == obj.id, gen_data))
for relation in results:
type = Type.objects.get(pk=relation["target_type"])
# remove conflicting entry if it exists
self.remove_type_entry(relations, type, direction="_damage_to")
# add entry
damage_factor = relation["damage_factor"]
self.add_type_entry(
relations, type, damage_factor, direction="_damage_to"
)
del relation["generation"]
# populate defensive relations
results = list(filter(lambda x: x["target_type"] == obj.id, gen_data))
for relation in results:
type = Type.objects.get(pk=relation["damage_type"])
# remove conflicting entry if it exists
self.remove_type_entry(relations, type, direction="_damage_from")
# add entry
damage_factor = relation["damage_factor"]
self.add_type_entry(
relations, type, damage_factor, direction="_damage_from"
)
del relation["generation"]
# add to final list
final_data.append(past_relations)
return final_data
def remove_newer_types(self, relations, current_gen):
return list(filter(lambda x: self.type_is_present(x, current_gen), relations))
def type_is_present(self, type, current_gen):
type_obj = Type.objects.get(name=type["name"])
gen_introduced = Generation.objects.get(pk=type_obj.generation.id)
return gen_introduced.id <= current_gen.id
def get_type_pokemon(self, obj):
poke_type_objects = PokemonType.objects.filter(type=obj)
poke_types = PokemonTypeSerializer(
poke_type_objects, many=True, context=self.context
).data
for poke_type in poke_types:
del poke_type["type"]
return poke_types
#########################
# MACHINE SERIALIZERS #
#########################
class MachineDetailSerializer(serializers.ModelSerializer):
item = ItemSummarySerializer()
version_group = VersionGroupSummarySerializer()
move = MoveSummarySerializer()
class Meta:
model = Machine
fields = ("id", "item", "version_group", "move")
###################################
# MOVE BATTLE STYLE SERIALIZERS #
###################################
class MoveBattleStyleNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveBattleStyleName
fields = ("name", "language")
class MoveBattleStyleDetailSerializer(serializers.ModelSerializer):
names = MoveBattleStyleNameSerializer(
many=True, read_only=True, source="movebattlestylename"
)
class Meta:
model = MoveBattleStyle
fields = ("id", "name", "names")
###################################
# MOVE DAMAGE CLASS SERIALIZERS #
###################################
class MoveDamageClassNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveDamageClassName
fields = ("name", "language")
class MoveDamageClassDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveDamageClassDescription
fields = ("description", "language")
class MoveDamageClassDetailSerializer(serializers.ModelSerializer):
names = MoveDamageClassNameSerializer(
many=True, read_only=True, source="movedamageclassname"
)
descriptions = MoveDamageClassDescriptionSerializer(
many=True, read_only=True, source="movedamageclassdescription"
)
moves = MoveSummarySerializer(many=True, read_only=True, source="move")
class Meta:
model = MoveDamageClass
fields = (
"id",
"name",
"descriptions",
"moves",
"names",
)
###########################
# MOVE META SERIALIZERS #
###########################
class MoveMetaAilmentNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveMetaAilmentName
fields = ("name", "language")
class MoveMetaAilmentDetailSerializer(serializers.ModelSerializer):
names = MoveMetaAilmentNameSerializer(
many=True, read_only=True, source="movemetaailmentname"
)
moves = serializers.SerializerMethodField("get_ailment_moves")
class Meta:
model = MoveMetaAilment
fields = ("id", "name", "moves", "names")
def get_ailment_moves(self, obj):
move_meta_objects = MoveMeta.objects.filter(move_meta_ailment=obj)
moves = []
for meta in move_meta_objects:
move_obj = Move.objects.get(pk=meta.move.id)
data = MoveSummarySerializer(move_obj, context=self.context).data
moves.append(data)
return moves
class MoveMetaCategoryDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveMetaCategoryDescription
fields = ("description", "language")
class MoveMetaCategoryDetailSerializer(serializers.ModelSerializer):
descriptions = MoveMetaCategoryDescriptionSerializer(
many=True, read_only=True, source="movemetacategorydescription"
)
moves = serializers.SerializerMethodField("get_category_moves")
class Meta:
model = MoveMetaCategory
fields = ("id", "name", "descriptions", "moves")
def get_category_moves(self, obj):
move_meta_objects = MoveMeta.objects.filter(move_meta_category=obj)
moves = []
for meta in move_meta_objects:
move_obj = Move.objects.get(pk=meta.move.id)
data = MoveSummarySerializer(move_obj, context=self.context).data
moves.append(data)
return moves
class MoveMetaSerializer(serializers.ModelSerializer):
ailment = MoveMetaAilmentSummarySerializer(source="move_meta_ailment")
category = MoveMetaCategorySummarySerializer(source="move_meta_category")
class Meta:
model = MoveMeta
fields = (
"ailment",
"category",
"min_hits",
"max_hits",
"min_turns",
"max_turns",
"drain",
"healing",
"crit_rate",
"ailment_chance",
"flinch_chance",
"stat_chance",
)
#############################
# MOVE TARGET SERIALIZERS #
#############################
class MoveTargetNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveTargetName
fields = ("name", "language")
class MoveTargetDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveTargetDescription
fields = ("description", "language")
class MoveTargetDetailSerializer(serializers.ModelSerializer):
names = MoveTargetNameSerializer(many=True, read_only=True, source="movetargetname")
descriptions = MoveTargetDescriptionSerializer(
many=True, read_only=True, source="movetargetdescription"
)
moves = MoveSummarySerializer(many=True, read_only=True, source="move")
class Meta:
model = MoveTarget
fields = ("id", "name", "descriptions", "moves", "names")
######################
# MOVE SERIALIZERS #
######################
class MoveNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = AbilityName
fields = ("name", "language")
class MoveChangeSerializer(serializers.ModelSerializer):
version_group = VersionGroupSummarySerializer()
type = TypeSummarySerializer()
effect_entries = serializers.SerializerMethodField("get_effects")
effect_chance = serializers.IntegerField(source="move_effect_chance")
class Meta:
model = MoveChange
fields = (
"accuracy",
"power",
"pp",
"effect_chance",
"effect_entries",
"type",
"version_group",
)
def get_effects(self, obj):
effect_texts = MoveEffectEffectText.objects.filter(move_effect=obj.move_effect)
data = MoveEffectEffectTextSerializer(
effect_texts, many=True, context=self.context
).data
return data
class MoveEffectEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveEffectEffectText
fields = ("effect", "short_effect", "language")
class MoveEffectChangeEffectTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveEffectChangeEffectText
fields = ("effect", "language")
class MoveEffectChangeSerializer(serializers.ModelSerializer):
version_group = VersionGroupSummarySerializer()
effect_entries = MoveEffectChangeEffectTextSerializer(
many=True, read_only=True, source="moveeffectchangeeffecttext"
)
class Meta:
model = MoveEffectChange
fields = ("version_group", "effect_entries")
class MoveFlavorTextSerializer(serializers.ModelSerializer):
flavor_text = serializers.CharField()
language = LanguageSummarySerializer()
version_group = VersionGroupSummarySerializer()
class Meta:
model = MoveFlavorText
fields = ("flavor_text", "language", "version_group")
class MoveDetailSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
type = TypeSummarySerializer()
target = MoveTargetSummarySerializer(source="move_target")
contest_type = ContestTypeSummarySerializer()
contest_effect = ContestEffectSummarySerializer()
damage_class = MoveDamageClassSummarySerializer(source="move_damage_class")
meta = MoveMetaSerializer(read_only=True, source="movemeta")
names = MoveNameSerializer(many=True, read_only=True, source="movename")
effect_entries = serializers.SerializerMethodField("get_effect_text")
effect_chance = serializers.IntegerField(source="move_effect_chance")
contest_combos = serializers.SerializerMethodField("get_combos")
stat_changes = serializers.SerializerMethodField("get_move_stat_change")
super_contest_effect = SuperContestEffectSummarySerializer()
past_values = MoveChangeSerializer(many=True, read_only=True, source="movechange")
effect_changes = serializers.SerializerMethodField("get_effect_change_text")
machines = serializers.SerializerMethodField("get_move_machines")
flavor_text_entries = MoveFlavorTextSerializer(
many=True, read_only=True, source="moveflavortext"
)
learned_by_pokemon = serializers.SerializerMethodField()
class Meta:
model = Move
fields = (
"id",
"name",
"accuracy",
"effect_chance",
"pp",
"priority",
"power",
"contest_combos",
"contest_type",
"contest_effect",
"damage_class",
"effect_entries",
"effect_changes",
"generation",
"meta",
"names",
"past_values",
"stat_changes",
"super_contest_effect",
"target",
"type",
"machines",
"flavor_text_entries",
"learned_by_pokemon",
)
def get_learned_by_pokemon(self, obj):
pokemon_moves = PokemonMove.objects.filter(move_id=obj).order_by("pokemon_id")
pokemon_list = []
pokemon_ids = pokemon_moves.values("pokemon_id").distinct()
for id in pokemon_ids:
pokemon_object = Pokemon.objects.get(pk=id["pokemon_id"])
pokemon_data = PokemonSummarySerializer(
pokemon_object, context=self.context
).data
pokemon_list.append(pokemon_data)
return pokemon_list
def get_move_machines(self, obj):
machine_objects = Machine.objects.filter(move=obj)
machines = []
for machine_object in machine_objects:
machine_data = MachineSummarySerializer(
machine_object, context=self.context
).data
version_group_data = VersionGroupSummarySerializer(
machine_object.version_group, context=self.context
).data
machines.append(
{"machine": machine_data, "version_group": version_group_data}
)
return machines
def get_combos(self, obj):
normal_before_objects = ContestCombo.objects.filter(first_move=obj)
normal_before_data = ContestComboSerializer(
normal_before_objects, many=True, context=self.context
).data
normal_after_objects = ContestCombo.objects.filter(second_move=obj)
normal_after_data = ContestComboSerializer(
normal_after_objects, many=True, context=self.context
).data
super_before_objects = SuperContestCombo.objects.filter(first_move=obj)
super_before_data = SuperContestComboSerializer(
super_before_objects, many=True, context=self.context
).data
super_after_objects = SuperContestCombo.objects.filter(second_move=obj)
super_after_data = SuperContestComboSerializer(
super_after_objects, many=True, context=self.context
).data
details = None
if (
normal_before_data
or normal_after_data
or super_before_data
or super_after_data
):
details = OrderedDict()
details["normal"] = OrderedDict()
details["normal"]["use_before"] = None
details["normal"]["use_after"] = None
details["super"] = OrderedDict()
details["super"]["use_before"] = None
details["super"]["use_after"] = None
for combo in normal_before_data:
if details["normal"]["use_before"] is None:
details["normal"]["use_before"] = []
details["normal"]["use_before"].append(combo["second_move"])
for combo in normal_after_data:
if details["normal"]["use_after"] is None:
details["normal"]["use_after"] = []
details["normal"]["use_after"].append(combo["first_move"])
for combo in super_before_data:
if details["super"]["use_before"] is None:
details["super"]["use_before"] = []
details["super"]["use_before"].append(combo["second_move"])
for combo in super_after_data:
if details["super"]["use_after"] is None:
details["super"]["use_after"] = []
details["super"]["use_after"].append(combo["first_move"])
return details
def get_effect_text(self, obj):
effect_texts = MoveEffectEffectText.objects.filter(move_effect=obj.move_effect)
data = MoveEffectEffectTextSerializer(
effect_texts, many=True, context=self.context
).data
return data
def get_effect_change_text(self, obj):
effect_changes = MoveEffectChange.objects.filter(move_effect=obj.move_effect)
data = MoveEffectChangeSerializer(
effect_changes, many=True, context=self.context
).data
return data
def get_move_stat_change(self, obj):
stat_change_objects = MoveMetaStatChange.objects.filter(move=obj)
stat_changes = MoveMetaStatChangeSerializer(
stat_change_objects, many=True, context=self.context
).data
for change in stat_changes:
del change["move"]
return stat_changes
##########################
# PAL PARK SERIALIZERS #
##########################
class PalParkSerializer(serializers.ModelSerializer):
area = PalParkAreaSummarySerializer(read_only=True, source="pal_park_area")
pokemon_species = PokemonSpeciesSummarySerializer()
class Meta:
model = PalPark
fields = ("base_score", "rate", "area", "pokemon_species")
class PalParkAreaNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PalParkAreaName
fields = ("name", "language")
class PalParkAreaDetailSerializer(serializers.ModelSerializer):
names = PalParkAreaNameSerializer(
many=True, read_only=True, source="palparkareaname"
)
pokemon_encounters = serializers.SerializerMethodField("get_encounters")
class Meta:
model = PalParkArea
fields = ("id", "name", "names", "pokemon_encounters")
def get_encounters(self, obj):
pal_park_objects = PalPark.objects.filter(pal_park_area=obj)
parks = PalParkSerializer(
pal_park_objects, many=True, context=self.context
).data
encounters = []
for encounter in parks:
del encounter["area"]
encounters.append(encounter)
return encounters
###############################
# POKEMON COLOR SERIALIZERS #
###############################
class PokemonColorNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonColorName
fields = ("name", "language")
class PokemonColorDetailSerializer(serializers.ModelSerializer):
names = PokemonColorNameSerializer(
many=True, read_only=True, source="pokemoncolorname"
)
pokemon_species = PokemonSpeciesSummarySerializer(
many=True, read_only=True, source="pokemonspecies"
)
class Meta:
model = PokemonColor
fields = ("id", "name", "names", "pokemon_species")
##############################
# POKEMON FORM SERIALIZERS #
##############################
class PokemonFormSpritesSerializer(serializers.ModelSerializer):
class Meta:
model = PokemonFormSprites
fields = ("sprites",)
class PokemonFormNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonFormName
fields = ("name", "pokemon_name", "language")
class PokemonFormDetailSerializer(serializers.ModelSerializer):
pokemon = PokemonSummarySerializer()
version_group = VersionGroupSummarySerializer()
sprites = serializers.SerializerMethodField("get_pokemon_form_sprites")
form_names = serializers.SerializerMethodField("get_pokemon_form_names")
names = serializers.SerializerMethodField("get_pokemon_form_pokemon_names")
types = serializers.SerializerMethodField("get_pokemon_form_types")
class Meta:
model = PokemonForm
fields = (
"id",
"name",
"order",
"form_order",
"is_default",
"is_battle_only",
"is_mega",
"form_name",
"pokemon",
"sprites",
"version_group",
"form_names",
"names",
"types",
)
def get_pokemon_form_names(self, obj):
form_results = PokemonFormName.objects.filter(
pokemon_form=obj, name__regex=".+"
)
form_serializer = PokemonFormNameSerializer(
form_results, many=True, context=self.context
)
data = form_serializer.data
for name in data:
del name["pokemon_name"]
return data
def get_pokemon_form_pokemon_names(self, obj):
form_results = PokemonFormName.objects.filter(
pokemon_form=obj, pokemon_name__regex=".+"
)
form_serializer = PokemonFormNameSerializer(
form_results, many=True, context=self.context
)
data = form_serializer.data
for name in data:
name["name"] = name["pokemon_name"]
del name["pokemon_name"]
return data
def get_pokemon_form_sprites(self, obj):
sprites_object = PokemonFormSprites.objects.get(pokemon_form_id=obj)
sprites_data = PokemonFormSpritesSerializer(
sprites_object, context=self.context
).data
sprites_data = json.loads(sprites_data["sprites"])
return sprites_data
def get_pokemon_form_types(self, obj):
form_type_objects = PokemonFormType.objects.filter(pokemon_form=obj)
form_types = PokemonFormTypeSerializer(
form_type_objects, many=True, context=self.context
).data
for form_type in form_types:
del form_type["pokemon_form"]
# defer to parent Pokemon's types if no form-specific types
if form_types == []:
pokemon_object = Pokemon.objects.get(id=obj.pokemon_id)
pokemon_type_objects = PokemonType.objects.filter(pokemon=pokemon_object)
form_types = PokemonTypeSerializer(
pokemon_type_objects, many=True, context=self.context
).data
for form_type in form_types:
del form_type["pokemon"]
return form_types
#################################
# POKEMON HABITAT SERIALIZERS #
#################################
class PokemonHabitatNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonHabitatName
fields = ("name", "language")
class PokemonHabitatDetailSerializer(serializers.ModelSerializer):
names = PokemonHabitatNameSerializer(
many=True, read_only=True, source="pokemonhabitatname"
)
pokemon_species = PokemonSpeciesSummarySerializer(
many=True, read_only=True, source="pokemonspecies"
)
class Meta:
model = PokemonHabitat
fields = ("id", "name", "names", "pokemon_species")
##############################
# POKEMON MOVE SERIALIZERS #
##############################
class MoveLearnMethodNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveLearnMethodName
fields = ("name", "language")
class MoveLearnMethodDescriptionSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = MoveLearnMethodDescription
fields = ("description", "language")
class MoveLearnMethodDetailSerializer(serializers.ModelSerializer):
names = MoveLearnMethodNameSerializer(
many=True, read_only=True, source="movelearnmethodname"
)
descriptions = MoveLearnMethodDescriptionSerializer(
many=True, read_only=True, source="movelearnmethoddescription"
)
version_groups = serializers.SerializerMethodField("get_method_version_groups")
class Meta:
model = MoveLearnMethod
fields = ("id", "name", "names", "descriptions", "version_groups")
def get_method_version_groups(self, obj):
version_group_objects = VersionGroupMoveLearnMethod.objects.filter(
move_learn_method=obj
)
version_group_data = VersionGroupMoveLearnMethodSerializer(
version_group_objects, many=True, context=self.context
).data
groups = []
for vg in version_group_data:
groups.append(vg["version_group"])
return groups
# https://stackoverflow.com/a/45987450/3482533
class PokemonMoveSerializer(serializers.ModelSerializer):
class Meta:
model = PokemonMove
fields = "__all__"
###############################
# POKEMON SHAPE SERIALIZERS #
###############################
class PokemonShapeNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonShapeName
fields = ("name", "awesome_name", "language")
class PokemonShapeDetailSerializer(serializers.ModelSerializer):
names = serializers.SerializerMethodField("get_shape_names")
awesome_names = serializers.SerializerMethodField("get_shape_awesome_names")
pokemon_species = PokemonSpeciesSummarySerializer(
many=True, read_only=True, source="pokemonspecies"
)
class Meta:
model = PokemonShape
fields = ("id", "name", "awesome_names", "names", "pokemon_species")
def get_shape_names(self, obj):
results = PokemonShapeName.objects.filter(pokemon_shape_id=obj)
serializer = PokemonShapeNameSerializer(
results, many=True, context=self.context
)
data = serializer.data
for entry in data:
del entry["awesome_name"]
return data
def get_shape_awesome_names(self, obj):
results = PokemonShapeName.objects.filter(pokemon_shape_id=obj)
serializer = PokemonShapeNameSerializer(
results, many=True, context=self.context
)
data = serializer.data
for entry in data:
del entry["name"]
return data
##############################
# POKEMON ITEM SERIALIZERS #
##############################
class PokemonItemSerializer(serializers.ModelSerializer):
version = VersionSummarySerializer()
item = ItemSummarySerializer()
class Meta:
model = PokemonItem
fields = ("rarity", "item", "version")
##############################
# POKEMON STAT SERIALIZERS #
##############################
class PokemonStatSerializer(serializers.ModelSerializer):
stat = StatSummarySerializer()
class Meta:
model = PokemonStat
fields = ("base_stat", "effort", "stat")
#########################
# POKEMON SERIALIZERS #
#########################
class PokemonGameIndexSerializer(serializers.ModelSerializer):
version = VersionSummarySerializer()
class Meta:
model = PokemonGameIndex
fields = ("game_index", "version")
class PokemonSpritesSerializer(serializers.ModelSerializer):
class Meta:
model = PokemonSprites
fields = ("sprites",)
class PokemonDetailSerializer(serializers.ModelSerializer):
abilities = serializers.SerializerMethodField("get_pokemon_abilities")
past_abilities = serializers.SerializerMethodField("get_past_pokemon_abilities")
game_indices = PokemonGameIndexSerializer(
many=True, read_only=True, source="pokemongameindex"
)
moves = serializers.SerializerMethodField("get_pokemon_moves")
species = PokemonSpeciesSummarySerializer(source="pokemon_species")
stats = PokemonStatSerializer(many=True, read_only=True, source="pokemonstat")
types = serializers.SerializerMethodField("get_pokemon_types")
past_types = serializers.SerializerMethodField("get_past_pokemon_types")
forms = PokemonFormSummarySerializer(
many=True, read_only=True, source="pokemonform"
)
held_items = serializers.SerializerMethodField("get_pokemon_held_items")
location_area_encounters = serializers.SerializerMethodField("get_encounters")
sprites = serializers.SerializerMethodField("get_pokemon_sprites")
class Meta:
model = Pokemon
fields = (
"id",
"name",
"base_experience",
"height",
"is_default",
"order",
"weight",
"abilities",
"past_abilities",
"forms",
"game_indices",
"held_items",
"location_area_encounters",
"moves",
"species",
"sprites",
"stats",
"types",
"past_types",
)
def get_pokemon_sprites(self, obj):
sprites_object = PokemonSprites.objects.get(pokemon_id=obj)
sprites_data = PokemonSpritesSerializer(
sprites_object, context=self.context
).data
return json.loads(sprites_data["sprites"])
def get_pokemon_moves(self, obj):
version_objects = VersionGroup.objects.all()
version_data = VersionGroupSummarySerializer(
version_objects, many=True, context=self.context
).data
method_objects = MoveLearnMethod.objects.all()
method_data = MoveLearnMethodSummarySerializer(
method_objects, many=True, context=self.context
).data
# Get moves related to this pokemon and pull out unique Move IDs.
# Note that it's important to order by the same column we're using to
# determine if the entries are unique. Otherwise distinct() will
# return apparent duplicates.
pokemon_moves = PokemonMove.objects.filter(pokemon_id=obj).order_by("move_id")
move_ids = pokemon_moves.values("move_id").distinct()
move_list = []
for id in move_ids:
pokemon_move_details = OrderedDict()
# Get each Unique Move by ID
move_object = Move.objects.get(pk=id["move_id"])
move_data = MoveSummarySerializer(move_object, context=self.context).data
pokemon_move_details["move"] = move_data
# Get Versions and Move Methods associated with each unique move
pokemon_move_objects = pokemon_moves.filter(move_id=id["move_id"])
serializer = PokemonMoveSerializer(
pokemon_move_objects, many=True, context=self.context
)
pokemon_move_details["version_group_details"] = []
for move in serializer.data:
version_detail = OrderedDict()
version_detail["level_learned_at"] = move["level"]
version_detail["version_group"] = version_data[
move["version_group"] - 1
]
version_detail["move_learn_method"] = method_data[
move["move_learn_method"] - 1
]
pokemon_move_details["version_group_details"].append(version_detail)
move_list.append(pokemon_move_details)
return move_list
def get_pokemon_held_items(self, obj):
# Get items related to this pokemon and pull out unique Item IDs
pokemon_items = PokemonItem.objects.filter(pokemon_id=obj).order_by("item_id")
item_ids = pokemon_items.values("item_id").distinct()
item_list = []
for id in item_ids:
pokemon_item_details = OrderedDict()
# Get each Unique Item by ID
item_object = Item.objects.get(pk=id["item_id"])
item_data = ItemSummarySerializer(item_object, context=self.context).data
pokemon_item_details["item"] = item_data
# Get Versions associated with each unique item
pokemon_item_objects = pokemon_items.filter(item_id=id["item_id"])
serializer = PokemonItemSerializer(
pokemon_item_objects, many=True, context=self.context
)
pokemon_item_details["version_details"] = []
for item in serializer.data:
version_detail = OrderedDict()
version_detail["rarity"] = item["rarity"]
version_detail["version"] = item["version"]
pokemon_item_details["version_details"].append(version_detail)
item_list.append(pokemon_item_details)
return item_list
def get_pokemon_abilities(self, obj):
pokemon_ability_objects = PokemonAbility.objects.filter(pokemon=obj)
data = PokemonAbilitySerializer(
pokemon_ability_objects, many=True, context=self.context
).data
abilities = []
for ability in data:
del ability["pokemon"]
abilities.append(ability)
return abilities
def get_past_pokemon_abilities(self, obj):
pokemon_past_ability_objects = PokemonAbilityPast.objects.filter(pokemon=obj)
pokemon_past_abilities = PokemonAbilityPastSerializer(
pokemon_past_ability_objects, many=True, context=self.context
).data
# post-process to the form we want
current_generation = ""
past_obj = {}
final_data = []
for pokemon_past_ability in pokemon_past_abilities:
del pokemon_past_ability["pokemon"]
generation = pokemon_past_ability["generation"]["name"]
if generation != current_generation:
current_generation = generation
past_obj = {}
# create past abilities object for this generation
past_obj["generation"] = pokemon_past_ability["generation"]
del pokemon_past_ability["generation"]
# create abilities array
past_obj["abilities"] = [pokemon_past_ability]
# add to past abilities array
final_data.append(past_obj)
else:
# add to existing array for this generation
del pokemon_past_ability["generation"]
past_obj["abilities"].append(pokemon_past_ability)
return final_data
def get_pokemon_types(self, obj):
poke_type_objects = PokemonType.objects.filter(pokemon=obj)
poke_types = PokemonTypeSerializer(
poke_type_objects, many=True, context=self.context
).data
for poke_type in poke_types:
del poke_type["pokemon"]
return poke_types
def get_past_pokemon_types(self, obj):
poke_past_type_objects = PokemonTypePast.objects.filter(pokemon=obj)
poke_past_types = PokemonTypePastSerializer(
poke_past_type_objects, many=True, context=self.context
).data
# post-process to the form we want
current_generation = ""
past_obj = {}
final_data = []
for poke_past_type in poke_past_types:
del poke_past_type["pokemon"]
generation = poke_past_type["generation"]["name"]
if generation != current_generation:
current_generation = generation
past_obj = {}
# create past types object for this generation
past_obj["generation"] = poke_past_type["generation"]
del poke_past_type["generation"]
# create types array
past_obj["types"] = [poke_past_type]
# add to past types array
final_data.append(past_obj)
else:
# add to existing array for this generation
del poke_past_type["generation"]
past_obj["types"].append(poke_past_type)
return final_data
def get_encounters(self, obj):
return reverse("pokemon_encounters", kwargs={"pokemon_id": obj.pk})
#################################
# POKEMON SPECIES SERIALIZERS #
#################################
class EvolutionTriggerNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = EvolutionTriggerName
fields = ("name", "language")
class EvolutionTriggerDetailSerializer(serializers.HyperlinkedModelSerializer):
names = EvolutionTriggerNameSerializer(
many=True, read_only=True, source="evolutiontriggername"
)
pokemon_species = serializers.SerializerMethodField("get_species")
class Meta:
model = EvolutionTrigger
fields = ("id", "name", "names", "pokemon_species")
def get_species(self, obj):
evo_objects = PokemonEvolution.objects.filter(evolution_trigger=obj)
species_list = []
species_names = set()
for evo in evo_objects:
species = PokemonSpeciesSummarySerializer(
evo.evolved_species, context=self.context
).data
if species["name"] not in species_names:
species_list.append(species)
species_names.add(species["name"])
return species_list
class PokemonSpeciesDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonSpeciesDescription
fields = ("description", "language")
class PokemonSpeciesFlavorTextSerializer(serializers.ModelSerializer):
flavor_text = serializers.CharField()
language = LanguageSummarySerializer()
version = VersionSummarySerializer()
class Meta:
model = PokemonSpeciesFlavorText
fields = ("flavor_text", "language", "version")
class PokemonSpeciesNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonSpeciesName
fields = ("name", "genus", "language")
class PokemonSpeciesEvolutionSerializer(serializers.ModelSerializer):
"""
This is here purely to help build pokemon evolution chains
"""
class Meta:
model = PokemonSpecies
fields = ("name", "id", "evolves_from_species", "is_baby")
class PokemonSpeciesDetailSerializer(serializers.ModelSerializer):
names = serializers.SerializerMethodField("get_pokemon_names")
form_descriptions = PokemonSpeciesDescriptionSerializer(
many=True, read_only=True, source="pokemonspeciesdescription"
)
pokedex_numbers = PokemonDexEntrySerializer(
many=True, read_only=True, source="pokemondexnumber"
)
egg_groups = serializers.SerializerMethodField("get_pokemon_egg_groups")
flavor_text_entries = PokemonSpeciesFlavorTextSerializer(
many=True, read_only=True, source="pokemonspeciesflavortext"
)
genera = serializers.SerializerMethodField("get_pokemon_genera")
generation = GenerationSummarySerializer()
growth_rate = GrowthRateSummarySerializer()
color = PokemonColorSummarySerializer(source="pokemon_color")
habitat = PokemonHabitatSummarySerializer(source="pokemon_habitat")
shape = PokemonShapeSummarySerializer(source="pokemon_shape")
evolves_from_species = PokemonSpeciesSummarySerializer()
varieties = serializers.SerializerMethodField("get_pokemon_varieties")
evolution_chain = EvolutionChainSummarySerializer()
pal_park_encounters = serializers.SerializerMethodField("get_encounters")
class Meta:
model = PokemonSpecies
fields = (
"id",
"name",
"order",
"gender_rate",
"capture_rate",
"base_happiness",
"is_baby",
"is_legendary",
"is_mythical",
"hatch_counter",
"has_gender_differences",
"forms_switchable",
"growth_rate",
"pokedex_numbers",
"egg_groups",
"color",
"shape",
"evolves_from_species",
"evolution_chain",
"habitat",
"generation",
"names",
"pal_park_encounters",
"form_descriptions",
"flavor_text_entries",
"genera",
"varieties",
)
def get_pokemon_names(self, obj):
species_results = PokemonSpeciesName.objects.filter(pokemon_species=obj)
species_serializer = PokemonSpeciesNameSerializer(
species_results, many=True, context=self.context
)
data = species_serializer.data
for name in data:
del name["genus"]
return data
def get_pokemon_genera(self, obj):
results = PokemonSpeciesName.objects.filter(pokemon_species=obj)
serializer = PokemonSpeciesNameSerializer(
results, many=True, context=self.context
)
data = serializer.data
genera = []
for entry in data:
if entry["genus"]:
del entry["name"]
genera.append(entry)
return genera
def get_pokemon_egg_groups(self, obj):
results = PokemonEggGroup.objects.filter(pokemon_species=obj)
data = PokemonEggGroupSerializer(results, many=True, context=self.context).data
groups = []
for group in data:
groups.append(group["egg_group"])
return groups
def get_pokemon_varieties(self, obj):
results = Pokemon.objects.filter(pokemon_species=obj)
summary_data = PokemonSummarySerializer(
results, many=True, context=self.context
).data
detail_data = PokemonDetailSerializer(
results, many=True, context=self.context
).data
varieties = []
for index, pokemon in enumerate(detail_data):
entry = OrderedDict()
entry["is_default"] = pokemon["is_default"]
entry["pokemon"] = summary_data[index]
varieties.append(entry)
return varieties
def get_encounters(self, obj):
pal_park_objects = PalPark.objects.filter(pokemon_species=obj)
parks = PalParkSerializer(
pal_park_objects, many=True, context=self.context
).data
encounters = []
for encounter in parks:
del encounter["pokemon_species"]
encounters.append(encounter)
return encounters
class PokemonEvolutionSerializer(serializers.ModelSerializer):
item = ItemSummarySerializer(source="evolution_item")
held_item = ItemSummarySerializer()
known_move = MoveSummarySerializer()
known_move_type = TypeSummarySerializer()
party_species = PokemonSpeciesSummarySerializer()
party_type = TypeSummarySerializer()
trade_species = PokemonSpeciesSummarySerializer()
location = LocationSummarySerializer()
trigger = EvolutionTriggerSummarySerializer(source="evolution_trigger")
class Meta:
model = PokemonEvolution
fields = (
"item",
"trigger",
"gender",
"held_item",
"known_move",
"known_move_type",
"location",
"min_level",
"min_happiness",
"min_beauty",
"min_affection",
"needs_overworld_rain",
"party_species",
"party_type",
"relative_physical_stats",
"time_of_day",
"trade_species",
"turn_upside_down",
)
class EvolutionChainDetailSerializer(serializers.ModelSerializer):
baby_trigger_item = ItemSummarySerializer()
chain = serializers.SerializerMethodField("build_chain")
class Meta:
model = EvolutionChain
fields = ("id", "baby_trigger_item", "chain")
def build_chain(self, obj):
chain_id = obj.id
pokemon_objects = PokemonSpecies.objects.filter(
evolution_chain_id=chain_id
).order_by("order")
summary_data = PokemonSpeciesSummarySerializer(
pokemon_objects, many=True, context=self.context
).data
ref_data = PokemonSpeciesEvolutionSerializer(
pokemon_objects, many=True, context=self.context
).data
# convert evolution data list to tree
evolution_tree = self.build_evolution_tree(ref_data)
# serialize chain recursively from tree
chain = self.build_chain_link_entry(evolution_tree, summary_data)
return chain
# converts a list of Pokemon species evolution data into a tree representing the evolution chain
def build_evolution_tree(self, species_evolution_data):
evolution_tree = OrderedDict()
evolution_tree["species"] = species_evolution_data[0]
evolution_tree["children"] = []
for species in species_evolution_data[1:]:
chain_link = OrderedDict()
chain_link["species"] = species
chain_link["children"] = []
evolves_from_species_id = chain_link["species"]["evolves_from_species"]
# find parent link by DFS
parent_link = evolution_tree
search_stack = [parent_link]
while len(search_stack) > 0:
l = search_stack.pop()
if l["species"]["id"] == evolves_from_species_id:
parent_link = l
break
# "left" to "right" requires reversing the list of children
search_stack += reversed(l["children"])
parent_link["children"].append(chain_link)
return evolution_tree
# serializes an evolution chain link recursively
# chain_link is a tree representing an evolution chain
def build_chain_link_entry(self, chain_link, summary_data):
entry = OrderedDict()
evolution_data = None
species = chain_link["species"]
if species["evolves_from_species"]:
evolution_object = PokemonEvolution.objects.filter(
evolved_species=species["id"]
)
evolution_data = PokemonEvolutionSerializer(
evolution_object, many=True, context=self.context
).data
entry["is_baby"] = species["is_baby"]
species_summary = next(x for x in summary_data if x["name"] == species["name"])
entry["species"] = species_summary
entry["evolution_details"] = evolution_data or []
evolves_to = [
self.build_chain_link_entry(c, summary_data) for c in chain_link["children"]
]
entry["evolves_to"] = evolves_to
return entry
class PokemonDexNumberSerializer(serializers.ModelSerializer):
entry_number = serializers.IntegerField(source="pokedex_number")
pokemon_species = PokemonSpeciesSummarySerializer()
class Meta:
model = PokemonDexNumber
fields = ("pokedex", "entry_number", "pokemon_species")
############################
# POKEATHLON SERIALIZERS #
############################
class PokeathlonStatNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokeathlonStatName
fields = ("name", "language")
class PokeathlonStatDetailSerializer(serializers.HyperlinkedModelSerializer):
names = PokeathlonStatNameSerializer(
many=True, read_only=True, source="pokeathlonstatname"
)
affecting_natures = serializers.SerializerMethodField("get_natures_that_affect")
class Meta:
model = PokeathlonStat
fields = ("id", "name", "affecting_natures", "names")
def get_natures_that_affect(self, obj):
stat_change_objects = NaturePokeathlonStat.objects.filter(pokeathlon_stat=obj)
stat_changes = NaturePokeathlonStatSerializer(
stat_change_objects, many=True, context=self.context
).data
changes = OrderedDict([("increase", []), ("decrease", [])])
for change in stat_changes:
del change["pokeathlon_stat"]
if change["max_change"] > 0:
changes["increase"].append(change)
else:
changes["decrease"].append(change)
return changes
#########################
# POKEDEX SERIALIZERS #
#########################
class PokedexNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokedexName
fields = ("name", "language")
class PokedexDescriptionSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokedexDescription
fields = ("description", "language")
class PokedexDetailSerializer(serializers.ModelSerializer):
region = RegionSummarySerializer()
names = PokedexNameSerializer(many=True, read_only=True, source="pokedexname")
descriptions = PokedexDescriptionSerializer(
many=True, read_only=True, source="pokedexdescription"
)
pokemon_entries = serializers.SerializerMethodField("get_pokedex_entries")
version_groups = serializers.SerializerMethodField("get_pokedex_version_groups")
class Meta:
model = Pokedex
fields = (
"id",
"name",
"is_main_series",
"descriptions",
"names",
"pokemon_entries",
"region",
"version_groups",
)
def get_pokedex_entries(self, obj):
results = PokemonDexNumber.objects.filter(pokedex=obj).order_by(
"pokedex_number"
)
serializer = PokemonDexNumberSerializer(
results, many=True, context=self.context
)
data = serializer.data
for entry in data:
del entry["pokedex"]
return data
def get_pokedex_version_groups(self, obj):
dex_group_objects = PokedexVersionGroup.objects.filter(pokedex=obj)
dex_groups = PokedexVersionGroupSerializer(
dex_group_objects, many=True, context=self.context
).data
results = []
for dex_group in dex_groups:
results.append(dex_group["version_group"])
return results
#########################
# VERSION SERIALIZERS #
#########################
class VersionNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = VersionName
fields = ("name", "language")
class VersionDetailSerializer(serializers.ModelSerializer):
"""
Should have a link to Version Group info but the Circular
dependency and compilation order fight eachother and I'm
not sure how to add anything other than a hyperlink
"""
names = VersionNameSerializer(many=True, read_only=True, source="versionname")
version_group = VersionGroupSummarySerializer()
class Meta:
model = Version
fields = ("id", "name", "names", "version_group")
class VersionGroupDetailSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
versions = VersionSummarySerializer(many=True, read_only=True, source="version")
regions = serializers.SerializerMethodField("get_version_group_regions")
move_learn_methods = serializers.SerializerMethodField("get_learn_methods")
pokedexes = serializers.SerializerMethodField("get_version_groups_pokedexes")
class Meta:
model = VersionGroup
fields = (
"id",
"name",
"order",
"generation",
"move_learn_methods",
"pokedexes",
"regions",
"versions",
)
def get_version_group_regions(self, obj):
vg_regions = VersionGroupRegion.objects.filter(version_group=obj)
data = VersionGroupRegionSerializer(
vg_regions, many=True, context=self.context
).data
regions = []
for region in data:
regions.append(region["region"])
return regions
def get_learn_methods(self, obj):
learn_method_objects = VersionGroupMoveLearnMethod.objects.filter(
version_group=obj
)
learn_method_data = VersionGroupMoveLearnMethodSerializer(
learn_method_objects, many=True, context=self.context
).data
methods = []
for method in learn_method_data:
methods.append(method["move_learn_method"])
return methods
def get_version_groups_pokedexes(self, obj):
dex_group_objects = PokedexVersionGroup.objects.filter(version_group=obj)
dex_groups = PokedexVersionGroupSerializer(
dex_group_objects, many=True, context=self.context
).data
results = []
for dex_group in dex_groups:
results.append(dex_group["pokedex"])
return results