little more work on pokemon evolution chains

This commit is contained in:
Adickes 2015-10-02 23:40:21 -04:00
parent 163e29c527
commit 6d7fd53ece
9 changed files with 568 additions and 107 deletions

View file

@ -42,9 +42,9 @@ MEDIA_ROOT = PROJECT_ROOT.child('media')
MEDIA_URL = '/media/'
STATIC_ROOT = PROJECT_ROOT.child('static')
STATIC_ROOT = PROJECT_ROOT.child('assets')
STATIC_URL = '/static/'
STATIC_URL = '/assets/'
ALLOWED_HOSTS = ['localhost']

View file

@ -48,8 +48,15 @@ from pokemon_v2.views import *
router = routers.DefaultRouter()
router.register(r"ability", AbilityResource)
router.register(r"berry", BerryResource)
router.register(r"egg-group", EggGroupResource)
router.register(r"generation", GenerationResource)
router.register(r"growth-rate", GrowthRateResource)
router.register(r"item", ItemResource)
router.register(r"item-category", ItemCategoryResource)
router.register(r"item-attribute", ItemFlagResource)
router.register(r"item-fling-effect", ItemFlingEffectResource)
router.register(r"item-pocket", ItemPocketResource)
router.register(r"language", LanguageResource)
router.register(r"move", MoveResource)
router.register(r"move-ailment", MoveMetaAilmentResource)
@ -60,7 +67,9 @@ router.register(r"move-target", MoveTargetResource)
router.register(r"nature", NatureResource)
router.register(r"pokedex", PokedexResource)
router.register(r"pokemon", PokemonResource)
router.register(r"pokemon-habitat", PokemonHabitatResource)
router.register(r"pokemon-shape", PokemonShapeResource)
router.register(r"pokemon-species", PokemonSpeciesResource)
router.register(r"pokemon-color", PokemonColorResource)
router.register(r"region", RegionResource)
router.register(r"stat", StatResource)

View file

@ -497,7 +497,8 @@ for index, info in enumerate(data):
if index > 0:
model = ItemFlingEffect (
id = int(info[0])
id = int(info[0]),
name = info[1]
)
model.save()

View file

@ -1,8 +1,8 @@
id
1
2
3
4
5
6
7
id,identifier
1,badly-poison
2,burn
3,berry-effect
4,herb-effect
5,paralyze
6,poison
7,flinch

1 id identifier
2 1 badly-poison
3 2 burn
4 3 berry-effect
5 4 herb-effect
6 5 paralyze
7 6 poison
8 7 flinch

View file

@ -0,0 +1,20 @@
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.db import models, migrations
class Migration(migrations.Migration):
dependencies = [
('pokemon_v2', '0001_squashed_0013_auto_20150926_0913'),
]
operations = [
migrations.AddField(
model_name='itemflingeffect',
name='name',
field=models.CharField(default='name', max_length=100),
preserve_default=False,
),
]

View file

@ -0,0 +1,20 @@
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.db import models, migrations
class Migration(migrations.Migration):
dependencies = [
('pokemon_v2', '0002_itemflingeffect_name'),
]
operations = [
migrations.AlterField(
model_name='pokemonhabitatname',
name='pokemon_habitat',
field=models.ForeignKey(related_name='pokemonhabitatname', blank=True, to='pokemon_v2.PokemonHabitat', null=True),
preserve_default=True,
),
]

View file

@ -584,7 +584,7 @@ class ItemCategoryName(IsName, HasItemCategory):
pass
class ItemFlingEffect(models.Model):
class ItemFlingEffect(HasName):
pass
@ -1287,7 +1287,7 @@ class PokemonHabitat(HasName):
class PokemonHabitatName(IsName):
pokemon_habitat = models.ForeignKey(PokemonHabitat, blank=True, null=True)
pokemon_habitat = models.ForeignKey(PokemonHabitat, blank=True, null=True, related_name="%(class)s")
class PokemonItem(HasPokemon, HasVersion, HasItem):

View file

@ -9,7 +9,6 @@ PokeAPI v2 serializers in order of dependency
from .models import *
##########################
# LANGUAGE SERIALIZERS #
##########################
@ -98,9 +97,16 @@ class GenerationDetailSerializer(serializers.ModelSerializer):
fields = ('id', 'name', 'region', 'names')
############################
# GENERATION SERIALIZERS #
############################
#############################
# GROWTH RATE SERIALIZERS #
#############################
class ExperienceSerializer(serializers.ModelSerializer):
class Meta:
model = Experience
fields = ('level', 'experience')
class GrowthRateDescriptionSerializer(serializers.ModelSerializer):
@ -121,10 +127,11 @@ class GrowthRateSummarySerializer(serializers.HyperlinkedModelSerializer):
class GrowthRateDetailSerializer(serializers.ModelSerializer):
descriptions = GrowthRateDescriptionSerializer(many=True, read_only=True, source="growthratedescription")
levels = ExperienceSerializer(many=True, read_only=True, source="experience")
class Meta:
model = GrowthRate
fields = ('id', 'name', 'formula', 'descriptions')
fields = ('id', 'name', 'formula', 'descriptions', 'levels')
#########################
@ -258,6 +265,301 @@ class StatDetailSerializer(serializers.HyperlinkedModelSerializer):
#############################
# ITEM POCKET SERIALIZERS #
#############################
class ItemPocketNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemName
fields = ('name', 'language')
class ItemPocketSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemPocket
fields = ('name', 'url')
class ItemPocketDetailSerializer(serializers.ModelSerializer):
names = ItemPocketNameSerializer(many=True, read_only=True, source="itempocketname")
class Meta:
model = ItemPocket
fields = ('id', 'name', 'names')
###############################
# ITEM CATEGORY SERIALIZERS #
###############################
class ItemCategoryNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemName
fields = ('name', 'language')
class ItemCategorySummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemCategory
fields = ('name', 'url')
class ItemCategoryDetailSerializer(serializers.ModelSerializer):
names = ItemCategoryNameSerializer(many=True, read_only=True, source="itemcategoryname")
pocket = ItemPocketSummarySerializer(source="item_pocket")
class Meta:
model = ItemCategory
fields = ('id', 'name', 'pocket', 'names')
###########################
# ITEM FLAG SERIALIZERS #
###########################
# Calling these Attributes instead of Flags cause I think that just makes more sense
# Probably should change the names of the models as well. Work To Do.
class ItemFlagDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemFlagDescription
fields = ('name', 'description', 'language')
class ItemFlagSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemFlag
fields = ('name', 'url')
class ItemFlagDetailSerializer(serializers.ModelSerializer):
names = serializers.SerializerMethodField('get_flag_names')
descriptions = serializers.SerializerMethodField('get_flag_descriptions')
class Meta:
model = ItemFlag
fields = ('id', 'name', 'names', 'descriptions')
def get_flag_names(self, obj):
item_flag = ItemFlagDescription.objects.filter(item_flag_id=obj)
serializer = ItemFlagDescriptionSerializer(item_flag, many=True, context=self.context)
data = serializer.data
print data
for flag in data:
del flag['description']
return data
def get_flag_descriptions(self, obj):
item_flag = ItemFlagDescription.objects.filter(item_flag_id=obj)
serializer = ItemFlagDescriptionSerializer(item_flag, many=True, context=self.context)
data = serializer.data
for flag in data:
del flag['name']
return data
class ItemFlagMapSerializer(serializers.ModelSerializer):
attribute = ItemFlagSummarySerializer(source='item_flag')
class Meta:
model = ItemFlagMap
fields = ('attribute',)
###########################
# ITEM FLAG SERIALIZERS #
###########################
class ItemFlingEffectDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
description = serializers.CharField(source="effect")
class Meta:
model = ItemFlingEffectDescription
fields = ('description', 'language')
class ItemFlingEffectSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = ItemFlingEffect
fields = ('name', 'url')
class ItemFlingEffectDetailSerializer(serializers.ModelSerializer):
descriptions = ItemCategoryNameSerializer(many=True, read_only=True, source="itemflingeffectdescription")
class Meta:
model = ItemFlingEffect
fields = ('id', 'name', 'descriptions')
#######################
# 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 ItemDescriptionSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemDescription
fields = ('effect', 'short_effect', 'language')
class ItemNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = ItemName
fields = ('name', 'language')
class ItemSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Item
fields = ('name', 'url')
class ItemDetailSerializer(serializers.ModelSerializer):
names = ItemNameSerializer(many=True, read_only=True, source="itemname")
descriptions = ItemDescriptionSerializer(many=True, read_only=True, source="itemdescription")
flavor_text_entries = ItemFlavorTextSerializer(many=True, read_only=True, source="itemflavortext")
category = ItemCategorySummarySerializer(source="item_category")
attributes = ItemFlagMapSerializer(many=True, read_only=True, source="itemflagmap")
fling_effect = ItemFlingEffectSummarySerializer(source="item_fling_effect")
class Meta:
model = Item
fields = (
'id',
'name',
'cost',
'fling_power',
'fling_effect',
'category',
'attributes',
'names',
'descriptions',
'flavor_text_entries'
)
#######################
# BERRY SERIALIZERS #
#######################
class BerrySummarySerializer(serializers.HyperlinkedModelSerializer):
# name = serializers.SerializerMethodField('get_berry_name')
class Meta:
model = Berry
fields = ('url',)
# def get_berry_name(self,obj):
# item = Item.objects.filter(item_id=obj)
# serializer = MoveDamageClassDescriptionSerializer(move_methods, many=True, context=self.context)
# data = serializer.data
# return 'name'
class BerryDetailSerializer(serializers.ModelSerializer):
item = ItemSummarySerializer()
name = serializers.SerializerMethodField('get_berry_name')
class Meta:
model = Berry
def get_berry_name(self,obj):
print self.fields['item'].fields['name']
# print self['item']
# item = Item.objects.filter(item_id=obj)
# serializer = MoveDamageClassDescriptionSerializer(move_methods, many=True, context=self.context)
# data = serializer.data
return 'name'
###########################
# EGG GROUP SERIALIZERS #
###########################
class EggGroupNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = EggGroupName
fields = ('name', 'language')
class EggGroupSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = EggGroup
fields = ('name', 'url')
class EggGroupDetailSerializer(serializers.ModelSerializer):
names = EggGroupNameSerializer(many=True, read_only=True, source="egggroupname")
class Meta:
model = EggGroup
fields = ('id', 'name', 'names')
######################
# TYPE SERIALIZERS #
######################
@ -612,6 +914,35 @@ class PokemonFormSerializer(serializers.ModelSerializer):
#################################
# POKEMON HABITAT SERIALIZERS #
#################################
class PokemonHabitatNameSerializer(serializers.HyperlinkedModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonHabitatName
fields = ('name', 'language')
class PokemonHabitatSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonHabitat
fields = ('name', 'url')
class PokemonHabitatDetailSerializer(serializers.ModelSerializer):
names = PokemonHabitatNameSerializer(many=True, read_only=True, source="pokemonhabitatname")
class Meta:
model = PokemonHabitat
fields = ('id', 'name', 'names')
##############################
# POKEMON MOVE SERIALIZERS #
##############################
@ -716,92 +1047,6 @@ class PokemonShapeDetailSerializer(serializers.ModelSerializer):
#################################
# POKEMON SPECIES SERIALIZERS #
#################################
class PokemonSpeciesNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonSpeciesName
fields = ('name', 'genus', 'language')
class PokemonSpeciesSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonSpecies
fields = ('name', 'url')
class PokemonSpeciesDetailSerializer(serializers.ModelSerializer):
names = serializers.SerializerMethodField('get_pokemon_names')
genera = serializers.SerializerMethodField('get_pokemon_genera')
generation = GenerationSummarySerializer()
growth_rate = GrowthRateSummarySerializer()
pokemon_color = PokemonColorSummarySerializer()
pokemon_shape = PokemonShapeSummarySerializer()
# varieties = PokemonDetailSerializer(many=True, read_only=True, source="pokemon")
# types = TypeSummarySerializer(many=True, read_only=True, source="pokemonspecies")
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',
'pokemon_color',
'pokemon_shape',
'evolves_from_species',
'evolution_chain',
'pokemon_habitat',
'generation',
# 'types',
'names',
'genera',
# 'varieties'
)
def get_pokemon_names(self, obj):
results = PokemonSpeciesName.objects.filter(pokemon_species=obj)
serializer = PokemonSpeciesNameSerializer(results, many=True, context=self.context)
data = 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_types()
##############################
# POKEMON STAT SERIALIZERS #
##############################
@ -856,11 +1101,13 @@ class PokemonDetailSerializer(serializers.ModelSerializer):
types = PokemonTypeSerializer(many=True, read_only=True, source="pokemontype")
# forms = PokemonFormSerializer(many=True, read_only=True, source="pokemonform")
class Meta:
model = Pokemon
fields = (
'id',
'order',
'order',
'pokemon_species',
'name',
'is_default',
'height',
@ -919,15 +1166,116 @@ class PokemonDetailSerializer(serializers.ModelSerializer):
#################################
# POKEMON SPECIES SERIALIZERS #
#################################
class PokemonEvolutionSerializer(serializers.ModelSerializer):
class Meta:
model = PokemonEvolution
class PokemonSpeciesNameSerializer(serializers.ModelSerializer):
language = LanguageSummarySerializer()
class Meta:
model = PokemonSpeciesName
fields = ('name', 'genus', 'language')
class PokemonSpeciesSummarySerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = PokemonSpecies
fields = ('name', 'url')
class PokemonSpeciesDetailSerializer(serializers.ModelSerializer):
names = serializers.SerializerMethodField('get_pokemon_names')
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")
evolution_chain = serializers.SerializerMethodField('get_evolutions')
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',
'color',
'shape',
'evolves_from_species',
'evolution_chain',
'habitat',
'generation',
'names',
'genera',
'varieties'
)
def get_pokemon_names(self, obj):
results = PokemonSpeciesName.objects.filter(pokemon_species=obj)
serializer = PokemonSpeciesNameSerializer(results, many=True, context=self.context)
data = 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_evolutions(self, obj):
chain_id = obj.evolution_chain.id
pokemon_objects = PokemonSpecies.objects.filter(evolution_chain_id=chain_id).order_by('order')
data = PokemonSpeciesSummarySerializer(pokemon_objects, many=True, context=self.context).data
print data
return data
class PokemonDexNumberSerializer(serializers.ModelSerializer):
entry_number = serializers.IntegerField(source="pokedex_number")
pokemon = PokemonSpeciesSummarySerializer(source="pokemon_species")
pokemon_species = PokemonSpeciesSummarySerializer(source="pokemon_species")
class Meta:
model = PokemonDexNumber
fields = ('pokedex', 'entry_number', 'pokemon')
fields = ('pokedex', 'entry_number', 'pokemon_species')

View file

@ -66,6 +66,20 @@ class AbilityResource(PokeapiCommonViewset):
list_serializer_class = AbilitySummarySerializer
class BerryResource(PokeapiCommonViewset):
queryset = Berry.objects.all()
serializer_class = BerryDetailSerializer
list_serializer_class = BerrySummarySerializer
class EggGroupResource(PokeapiCommonViewset):
queryset = EggGroup.objects.all()
serializer_class = EggGroupDetailSerializer
list_serializer_class = EggGroupSummarySerializer
class GenerationResource(PokeapiCommonViewset):
queryset = Generation.objects.all()
@ -80,6 +94,41 @@ class GrowthRateResource(PokeapiCommonViewset):
list_serializer_class = GrowthRateSummarySerializer
class ItemResource(PokeapiCommonViewset):
queryset = Item.objects.all()
serializer_class = ItemDetailSerializer
list_serializer_class = ItemSummarySerializer
class ItemCategoryResource(PokeapiCommonViewset):
queryset = ItemCategory.objects.all()
serializer_class = ItemCategoryDetailSerializer
list_serializer_class = ItemCategorySummarySerializer
class ItemFlagResource(PokeapiCommonViewset):
queryset = ItemFlag.objects.all()
serializer_class = ItemFlagDetailSerializer
list_serializer_class = ItemFlagSummarySerializer
class ItemFlingEffectResource(PokeapiCommonViewset):
queryset = ItemFlingEffect.objects.all()
serializer_class = ItemFlingEffectDetailSerializer
list_serializer_class = ItemFlingEffectSummarySerializer
class ItemPocketResource(PokeapiCommonViewset):
queryset = ItemPocket.objects.all()
serializer_class = ItemPocketDetailSerializer
list_serializer_class = ItemPocketSummarySerializer
class LanguageResource(PokeapiCommonViewset):
queryset = Language.objects.all()
@ -148,6 +197,13 @@ class PokemonColorResource(PokeapiCommonViewset):
list_serializer_class = PokemonColorSummarySerializer
class PokemonHabitatResource(PokeapiCommonViewset):
queryset = PokemonHabitat.objects.all()
serializer_class = PokemonHabitatDetailSerializer
list_serializer_class = PokemonHabitatSummarySerializer
class PokemonShapeResource(PokeapiCommonViewset):
queryset = PokemonShape.objects.all()
@ -162,6 +218,13 @@ class PokemonResource(PokeapiCommonViewset):
list_serializer_class =PokemonSummarySerializer
class PokemonSpeciesResource(PokeapiCommonViewset):
queryset = PokemonSpecies.objects.all()
serializer_class = PokemonSpeciesDetailSerializer
list_serializer_class = PokemonSpeciesSummarySerializer
class RegionResource(PokeapiCommonViewset):
queryset = Region.objects.all()