pokeapi/pokemon_v2/serializers.py

2893 lines
90 KiB
Python
Raw Normal View History

2015-09-06 02:10:13 +00:00
2015-02-22 17:11:40 +00:00
from __future__ import unicode_literals
2015-09-06 01:22:06 +00:00
from rest_framework import serializers
from collections import OrderedDict
2015-09-06 02:10:13 +00:00
2015-02-22 17:11:40 +00:00
"""
PokeAPI v2 serializers in order of dependency
2015-02-22 17:11:40 +00:00
"""
from .models import *
2015-02-22 17:11:40 +00:00
2015-10-06 17:01:32 +00:00
#########################
# 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):
2015-10-06 17:01:32 +00:00
class Meta:
model = Ability
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class BerryFirmnessSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = BerryFirmness
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
2015-11-10 19:20:06 +00:00
class BerryFlavorSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = BerryFlavor
fields = ('name', 'url')
class BerrySummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Berry
fields = ('name', 'url')
class CharacteristicSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Characteristic
fields = ('url',)
2015-10-06 17:01:32 +00:00
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):
2015-10-06 17:01:32 +00:00
class Meta:
model = EggGroup
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class EncounterConditionSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = EncounterCondition
fields = ('name', 'url')
class EncounterConditionValueSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EncounterConditionValue
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class EncounterMethodSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = EncounterMethod
fields = ('name', 'url')
class EvolutionTriggerSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EvolutionTrigger
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class EvolutionChainSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = EvolutionChain
fields = ('url',)
class GenerationSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Generation
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class GenderSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Gender
fields = ('name', 'url')
class GrowthRateSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = GrowthRate
2015-10-06 17:01:32 +00:00
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')
2015-10-23 04:56:01 +00:00
class ItemAttributeSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
2015-10-23 04:56:01 +00:00
model = ItemAttribute
2015-10-06 17:01:32 +00:00
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):
2015-10-06 17:01:32 +00:00
class Meta:
model = Language
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class LocationSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Location
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class LocationAreaSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = LocationArea
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class MachineSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Machine
fields = ('url',)
class MoveBattleStyleSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = MoveBattleStyle
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
2015-10-06 17:01:32 +00:00
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')
2015-10-06 17:01:32 +00:00
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):
2015-10-06 17:01:32 +00:00
class Meta:
model = PokemonForm
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class RegionSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Region
fields = ('name', 'url')
2015-10-06 17:01:32 +00:00
class StatSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Stat
fields = ('name', 'url')
class SuperContestEffectSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = SuperContestEffect
fields = ('url',)
class TypeSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Type
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class VersionSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = Version
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
class VersionGroupSummarySerializer(serializers.HyperlinkedModelSerializer):
2015-10-06 17:01:32 +00:00
class Meta:
model = VersionGroup
2015-10-06 17:01:32 +00:00
fields = ('name', 'url')
#####################
# MAP SERIALIZERS #
#####################
2015-11-10 19:20:06 +00:00
class BerryFlavorMapSerializer(serializers.ModelSerializer):
berry = BerrySummarySerializer()
flavor = BerryFlavorSummarySerializer(source='berry_flavor')
2015-11-10 19:20:06 +00:00
class Meta:
model = BerryFlavorMap
fields = ('potency', 'berry', 'flavor')
2015-11-10 19:20:06 +00:00
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 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 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()
2015-09-11 12:15:45 +00:00
class Meta:
model = ItemAttributeMap
fields = ('version_group', 'move_learn_method')
2015-09-11 12:15:45 +00:00
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',)
2015-10-06 17:01:32 +00:00
2015-09-11 12:15:45 +00:00
################################
# CHARACTERISTIC SERIALIZERS #
################################
2015-09-11 12:15:45 +00:00
class CharacteristicDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
2015-09-11 12:15:45 +00:00
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 <= 30):
values.append(mod)
mod += 5
return values;
#########################
# CONTEST SERIALIZERS #
#########################
class SuperContestEffectFlavorTextSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = SuperContestEffectFlavorText
fields = ('flavor_text', 'language')
2015-09-11 12:15:45 +00:00
class SuperContestEffectDetailSerializer(serializers.ModelSerializer):
2015-09-10 23:02:06 +00:00
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')
2015-09-11 12:15:45 +00:00
2015-10-06 17:01:32 +00:00
########################
# 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
2015-10-06 17:01:32 +00:00
############################
# 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')
2015-10-06 17:01:32 +00:00
########################
# 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 #
##########################
2015-10-06 17:01:32 +00:00
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
2015-10-06 17:01:32 +00:00
class LocationGameIndexSerializer(serializers.ModelSerializer):
generation = GenerationSummarySerializer()
class Meta:
model = LocationGameIndex
fields = ('game_index', 'generation')
2015-10-06 17:01:32 +00:00
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")
2015-10-06 17:01:32 +00:00
class Meta:
model = Location
fields = ('id', 'name', 'region', 'names', 'game_indices', 'areas')
2015-10-06 17:01:32 +00:00
#########################
# ABILITY SERIALIZERS #
#########################
class AbilityEffectTextSerializer(serializers.ModelSerializer):
2015-09-11 12:15:45 +00:00
language = LanguageSummarySerializer()
class Meta:
model = AbilityEffectText
fields = ('effect', 'short_effect', 'language')
class AbilityFlavorTextSerializer(serializers.ModelSerializer):
2015-11-28 14:30:55 +00:00
flavor_text = serializers.CharField()
2015-09-11 12:15:45 +00:00
language = LanguageSummarySerializer()
version_group = VersionGroupSummarySerializer()
class Meta:
model = AbilityFlavorText
2015-11-28 14:30:55 +00:00
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):
2015-09-11 12:15:45 +00:00
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")
2015-11-28 14:30:55 +00:00
# flavor_text_entries = serializers.SerializerMethodField('get_flavor_text')
names = AbilityNameSerializer(many=True, read_only=True, source="abilityname")
generation = GenerationSummarySerializer()
2015-11-28 14:30:55 +00:00
effect_changes = AbilityChangeSerializer(many=True, read_only=True, source="abilitychange")
pokemon = serializers.SerializerMethodField('get_ability_pokemon')
2015-09-11 12:15:45 +00:00
class Meta:
model = Ability
fields = (
'id',
'name',
2015-09-11 12:15:45 +00:00
'is_main_series',
'generation',
'names',
2015-11-28 14:30:55 +00:00
'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
2015-11-28 14:30:55 +00:00
# def get_flavor_text(self, obj):
# version_objects = VersionGroup.objects.all()
# version_data = VersionGroupSummarySerializer(version_objects, many=True, context=self.context).data
# flavor_texts = AbilityFlavorText.objects.filter(ability=obj)
# group_ids = flavor_texts.values('flavor').distinct()
# entries = []
# for group_id in group_ids:
# id = group_id['version_group']
# texts = flavor_texts.filter(version_group=id)
# detail = OrderedDict()
# detail['version_group'] = version_data[id-1]
# detail['entries'] = AbilityFlavorTextSerializer(texts, many=True, context=self.context).data
# entries.append(detail)
# return entries
######################
# 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")
2015-10-18 23:01:33 +00:00
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')
2015-10-18 23:01:33 +00:00
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')
2015-10-23 04:56:01 +00:00
################################
# ITEM ATTRIBUTE SERIALIZERS #
################################
class ItemAttributeNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemAttributeName
fields = ('name', 'language')
2015-10-23 04:56:01 +00:00
class ItemAttributeDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
2015-10-23 04:56:01 +00:00
model = ItemAttributeDescription
fields = ('description', 'language')
2015-10-23 04:56:01 +00:00
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:
2015-10-23 04:56:01 +00:00
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
2015-10-23 04:56:01 +00:00
###################################
# 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 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")
2015-10-23 04:56:01 +00:00
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')
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',
'baby_trigger_for'
)
2015-10-23 04:56:01 +00:00
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
2015-10-06 17:01:32 +00:00
########################
# NATURE SERIALIZERS #
########################
class NatureBattleStylePreferenceSerializer(serializers.ModelSerializer):
move_battle_style = MoveBattleStyleSummarySerializer()
class Meta:
model = NatureBattleStylePreference
fields = ( 'low_hp_preference', 'high_hp_preference', 'move_battle_style',)
2015-10-06 17:01:32 +00:00
class NatureNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
2015-10-06 17:01:32 +00:00
class Meta:
model = NatureName
fields = ('name', 'language')
2015-10-06 17:01:32 +00:00
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")
2015-10-06 17:01:32 +00:00
class Meta:
model = Nature
fields = ('id', 'name', 'decreased_stat', 'increased_stat', 'likes_flavor', 'hates_flavor', 'berries', 'pokeathlon_stat_changes', 'move_battle_style_preferences', 'names')
2015-10-06 17:01:32 +00:00
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
2015-10-06 17:01:32 +00:00
#######################
# BERRY SERIALIZERS #
#######################
2015-10-06 17:01:32 +00:00
class BerryFirmnessNameSerializer(serializers.ModelSerializer):
2015-10-06 17:01:32 +00:00
language = LanguageSummarySerializer()
class Meta:
2015-10-06 17:01:32 +00:00
model = BerryFirmnessName
fields = ('name', 'language')
2015-10-06 17:01:32 +00:00
class BerryFirmnessDetailSerializer(serializers.ModelSerializer):
2015-10-06 17:01:32 +00:00
names = BerryFirmnessNameSerializer(many=True, read_only=True, source="berryfirmnessname")
berries = BerrySummarySerializer(many=True, read_only=True, source="berry")
2015-10-06 17:01:32 +00:00
class Meta:
model = BerryFirmness
fields = ('id', 'name', 'berries', 'names')
2015-11-10 19:20:06 +00:00
class BerryFlavorNameSerializer(serializers.ModelSerializer):
2015-11-10 19:20:06 +00:00
language = LanguageSummarySerializer()
class Meta:
model = BerryFlavorName
fields = ('name', 'language')
2015-11-10 19:20:06 +00:00
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()
2015-10-06 17:01:32 +00:00
firmness = BerryFirmnessSummarySerializer(source="berry_firmness")
2015-11-10 19:20:06 +00:00
flavors = serializers.SerializerMethodField('get_berry_flavors')
class Meta:
model = Berry
2015-10-06 17:01:32 +00:00
fields = (
'id',
'name',
'growth_time',
'max_harvest',
'natural_gift_power',
2015-10-06 17:01:32 +00:00
'size',
'smoothness',
'soil_dryness',
'firmness',
'flavors',
'item',
'natural_gift_type',
2015-10-06 17:01:32 +00:00
)
2015-11-10 19:20:06 +00:00
def get_berry_flavors(self, obj):
2015-11-10 19:20:06 +00:00
flavor_map_objects = BerryFlavorMap.objects.filter(berry=obj)
flavor_maps = BerryFlavorMapSerializer(flavor_map_objects, many=True, context=self.context).data
flavors = []
2015-11-10 19:20:06 +00:00
for map in flavor_maps:
del map['berry']
flavors.append(map)
return flavors
###########################
# EGG GROUP SERIALIZERS #
###########################
2015-10-06 17:01:32 +00:00
class PokemonEggGroupSerializer(serializers.ModelSerializer):
2015-10-06 17:01:32 +00:00
species = PokemonSpeciesSummarySerializer(source="pokemon_species")
egg_group = EggGroupSummarySerializer()
class Meta:
2015-10-06 17:01:32 +00:00
model = PokemonEggGroup
fields = ('species', 'egg_group')
2015-10-06 17:01:32 +00:00
class EggGroupNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
2015-10-06 17:01:32 +00:00
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')
2015-10-06 17:01:32 +00:00
def get_species(self, obj):
2015-10-06 17:01:32 +00:00
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 #
######################
class TypeEfficacySerializer(serializers.ModelSerializer):
class Meta:
model = TypeEfficacy
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")
2015-10-23 04:56:01 +00:00
move_damage_class = MoveDamageClassSummarySerializer()
damage_relations = serializers.SerializerMethodField('get_type_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', 'game_indices', 'generation', 'move_damage_class', 'names', 'pokemon', 'moves')
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'])
if relation['damage_factor'] == 200:
relations['double_damage_to'].append(TypeSummarySerializer(type, context=self.context).data)
elif relation['damage_factor'] == 50:
relations['half_damage_to'].append(TypeSummarySerializer(type, context=self.context).data)
elif relation['damage_factor'] == 0:
relations['no_damage_to'].append(TypeSummarySerializer(type, context=self.context).data)
# 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'])
if relation['damage_factor'] == 200:
relations['double_damage_from'].append(TypeSummarySerializer(type, context=self.context).data)
elif relation['damage_factor'] == 50:
relations['half_damage_from'].append(TypeSummarySerializer(type, context=self.context).data)
elif relation['damage_factor'] == 0:
relations['no_damage_from'].append(TypeSummarySerializer(type, context=self.context).data)
return relations
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 = VersionSummarySerializer()
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 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')
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'
)
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'] == None:
details['normal']['use_before'] = []
details['normal']['use_before'].append(combo['second_move'])
for combo in normal_after_data:
if details['normal']['use_after'] == None:
details['normal']['use_after'] = []
details['normal']['use_after'].append(combo['first_move'])
for combo in super_before_data:
if details['super']['use_before'] == None:
details['super']['use_before'] = []
details['super']['use_before'].append(combo['second_move'])
for combo in super_after_data:
if details['super']['use_after'] == 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 #
##############################
2015-10-06 17:01:32 +00:00
class PokemonFormDetailSerializer(serializers.ModelSerializer):
2015-10-06 17:01:32 +00:00
pokemon = PokemonSummarySerializer()
version_group = VersionGroupSummarySerializer()
class Meta:
model = PokemonForm
fields = ('id', 'name', 'order', 'form_order', 'is_default', 'is_battle_only', 'is_mega', 'form_name', 'pokemon', 'version_group')
#################################
# 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
class PokemonMoveSerializer(serializers.ModelSerializer):
class Meta:
model = PokemonMove
###############################
# 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
2015-10-06 17:01:32 +00:00
##############################
# POKEMON ITEM SERIALIZERS #
##############################
class PokemonItemSerializer(serializers.ModelSerializer):
version = VersionSummarySerializer()
2015-10-06 17:01:32 +00:00
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 PokemonDetailSerializer(serializers.ModelSerializer):
abilities = serializers.SerializerMethodField('get_pokemon_abilities')
game_indices = PokemonGameIndexSerializer(many=True, read_only=True, source="pokemongameindex")
moves = serializers.SerializerMethodField('get_pokemon_moves')
2015-10-06 17:01:32 +00:00
species = PokemonSpeciesSummarySerializer(source="pokemon_species")
stats = PokemonStatSerializer(many=True, read_only=True, source="pokemonstat")
types = serializers.SerializerMethodField('get_pokemon_types')
2015-10-06 17:01:32 +00:00
forms = PokemonFormSummarySerializer(many=True, read_only=True, source="pokemonform")
held_items = serializers.SerializerMethodField('get_pokemon_held_items')
location_area_encounters = serializers.SerializerMethodField('get_encounters')
class Meta:
model = Pokemon
fields = (
2015-10-06 17:01:32 +00:00
'id',
'name',
2015-10-06 17:01:32 +00:00
'base_experience',
'height',
'is_default',
'order',
'weight',
'abilities',
2015-10-06 17:01:32 +00:00
'forms',
'game_indices',
2015-10-06 17:01:32 +00:00
'held_items',
'location_area_encounters',
'moves',
'species',
'stats',
'types',
)
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
2015-10-06 17:01:32 +00:00
pokemon_moves = PokemonMove.objects.filter(pokemon_id=obj).order_by('level')
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)
2015-10-06 17:01:32 +00:00
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]
2015-10-06 17:01:32 +00:00
pokemon_move_details['version_group_details'].append(version_detail)
move_list.append(pokemon_move_details)
return move_list
2015-10-06 17:01:32 +00:00
def get_pokemon_held_items(self, obj):
# Get items related to this pokemon and pull out unique Item IDs
2015-10-06 17:01:32 +00:00
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']
2015-10-06 17:01:32 +00:00
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_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_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(pokemon=obj).order_by('location_area')
encounters_list = []
# break encounters into pokemon groupings
for area in all_encounters.values('location_area').distinct():
location_area_object = LocationArea.objects.get(pk=area['location_area'])
location_area_detail = OrderedDict()
location_area_detail['location_area'] = LocationAreaSummarySerializer(location_area_object, context=self.context).data
location_area_detail['version_details'] = []
area_encounters = all_encounters.filter(location_area=area['location_area']).order_by('version')
# each pokemon has multiple versions it could be encountered in
for ver in area_encounters.values('version').distinct():
version_detail = OrderedDict()
version_detail['max_chance'] = 0
version_detail['encounter_details'] = []
version_detail['version'] = version_data[ver['version'] - 1]
area_data = EncounterDetailSerializer(area_encounters.filter(version=ver['version']), many=True, context=self.context).data
# each version has multiple ways a pokemon can be encountered
for encounter in area_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)
location_area_detail['version_details'].append(version_detail)
encounters_list.append(location_area_detail)
return encounters_list
#################################
# POKEMON SPECIES SERIALIZERS #
#################################
2015-10-06 17:01:32 +00:00
class EvolutionTriggerNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
2015-10-06 17:01:32 +00:00
model = EvolutionTriggerName
fields = ('name', 'language')
class EvolutionTriggerDetailSerializer(serializers.HyperlinkedModelSerializer):
names = EvolutionTriggerNameSerializer(many=True, read_only=True, source="evolutiontriggername")
pokemon_species = serializers.SerializerMethodField('get_species')
2015-10-06 17:01:32 +00:00
class Meta:
model = EvolutionTrigger
fields = ('id', 'name', 'names', 'pokemon_species')
2015-10-06 17:01:32 +00:00
def get_species(self, obj):
2015-10-06 17:01:32 +00:00
evo_objects = PokemonEvolution.objects.filter(evolution_trigger=obj);
species_list = []
2015-10-06 17:01:32 +00:00
for evo in evo_objects:
species = PokemonSpeciesSummarySerializer(evo.evolved_species, context=self.context).data
species_list.append(species)
return species_list
class PokemonSpeciesDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonSpeciesDescription
fields = ('description', 'language')
class PokemonSpeciesNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonSpeciesName
fields = ('name', 'genus', 'language')
2015-10-06 17:01:32 +00:00
class PokemonSpeciesEvolutionSerializer(serializers.ModelSerializer):
"""
This is here purely to help build pokemon evolution chains
"""
class Meta:
model = PokemonSpecies
2015-10-06 17:01:32 +00:00
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")
2015-10-06 17:01:32 +00:00
pokedex_numbers = PokemonDexEntrySerializer(many=True, read_only=True, source="pokemondexnumber")
egg_groups = serializers.SerializerMethodField('get_pokemon_egg_groups')
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 = PokemonSummarySerializer(many=True, read_only=True, source="pokemon")
2015-10-06 17:01:32 +00:00
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',
'hatch_counter',
'has_gender_differences',
'forms_switchable',
'growth_rate',
2015-10-06 17:01:32 +00:00
'pokedex_numbers',
'egg_groups',
'color',
'shape',
'evolves_from_species',
'evolution_chain',
'habitat',
'generation',
'names',
'pal_park_encounters',
'form_descriptions',
'genera',
'varieties'
)
def get_pokemon_names(self, obj):
2015-10-06 17:01:32 +00:00
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
2015-10-06 17:01:32 +00:00
def get_pokemon_egg_groups(self, obj):
2015-10-06 17:01:32 +00:00
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
2015-10-06 17:01:32 +00:00
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()
2015-10-06 17:01:32 +00:00
trigger = EvolutionTriggerSummarySerializer(source="evolution_trigger")
class Meta:
model = PokemonEvolution
fields = (
'item',
'trigger',
'gender',
'held_item',
'known_move',
'known_move_type',
'location',
2015-10-06 17:01:32 +00:00
'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')
2015-10-06 17:01:32 +00:00
summary_data = PokemonSpeciesSummarySerializer(pokemon_objects, many=True, context=self.context).data
ref_data = PokemonSpeciesEvolutionSerializer(pokemon_objects, many=True, context=self.context).data
2015-10-06 17:01:32 +00:00
chain = entry = OrderedDict()
current_evolutions = None
evolution_data = None
previous_entry = None
previous_species = None
for index, species in enumerate(ref_data):
# If evolves from something
if species['evolves_from_species']:
# In case this pokemon is one of multiple evolutions a pokemon can make
if previous_species['id'] == species['evolves_from_species']:
current_evolutions = previous_entry['evolves_to']
entry = OrderedDict()
2015-11-16 20:42:24 +00:00
many=False
try:
evolution_object = PokemonEvolution.objects.get(evolved_species=species['id'])
except PokemonEvolution.MultipleObjectsReturned:
evolution_object = PokemonEvolution.objects.filter(evolved_species=species['id'])
many=True
evolution_data = PokemonEvolutionSerializer(evolution_object, many=many, context=self.context).data
2015-10-06 17:01:32 +00:00
current_evolutions.append(entry)
2015-10-06 17:01:32 +00:00
entry['is_baby'] = species['is_baby']
entry['species'] = summary_data[index]
2015-11-28 14:30:55 +00:00
entry['evolution_details'] = evolution_data or None
2015-10-06 17:01:32 +00:00
entry['evolves_to'] = []
# Keep track of previous entries for complex chaining
previous_entry = entry
previous_species = species
return chain
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.order_by('pokedex_number').filter(pokedex=obj)
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