pokeapi/pokemon_v2/models.py
2019-12-01 12:38:51 +00:00

1497 lines
32 KiB
Python
Executable file

from django.db import models
#####################
# ABSTRACT MODELS #
#####################
class HasAbility(models.Model):
ability = models.ForeignKey('Ability', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasCharacteristic(models.Model):
characteristic = models.ForeignKey(
'Characteristic', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasContestType(models.Model):
contest_type = models.ForeignKey(
'ContestType', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasContestEffect(models.Model):
contest_effect = models.ForeignKey(
'ContestEffect', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasSuperContestEffect(models.Model):
super_contest_effect = models.ForeignKey(
'SuperContestEffect', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE
)
class Meta:
abstract = True
class HasDescription(models.Model):
description = models.CharField(max_length=1000, default='')
class Meta:
abstract = True
class HasGender(models.Model):
gender = models.ForeignKey(
'Gender', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasEffect(models.Model):
effect = models.CharField(max_length=6000)
class Meta:
abstract = True
class HasEggGroup(models.Model):
egg_group = models.ForeignKey(
'EggGroup', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasEncounterMethod(models.Model):
encounter_method = models.ForeignKey(
'EncounterMethod', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasEncounterCondition(models.Model):
encounter_condition = models.ForeignKey(
'EncounterCondition', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasEvolutionTrigger(models.Model):
evolution_trigger = models.ForeignKey(
'EvolutionTrigger', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasFlavorText(models.Model):
flavor_text = models.CharField(max_length=500)
class Meta:
abstract = True
class HasFlingEffect(models.Model):
item_fling_effect = models.ForeignKey(
'ItemFlingEffect', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasGameIndex(models.Model):
game_index = models.IntegerField()
class Meta:
abstract = True
class HasGeneration(models.Model):
generation = models.ForeignKey('Generation', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasGrowthRate(models.Model):
growth_rate = models.ForeignKey('GrowthRate', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasItem(models.Model):
item = models.ForeignKey('Item', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasItemAttribute(models.Model):
item_attribute = models.ForeignKey(
'ItemAttribute', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasItemCategory(models.Model):
item_category = models.ForeignKey(
'ItemCategory', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasItemPocket(models.Model):
item_pocket = models.ForeignKey(
'ItemPocket', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasLanguage(models.Model):
language = models.ForeignKey(
'Language', blank=True, null=True, related_name="%(class)s_language",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasLocation(models.Model):
location = models.ForeignKey('Location', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasLocationArea(models.Model):
location_area = models.ForeignKey(
'LocationArea', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMetaAilment(models.Model):
move_meta_ailment = models.ForeignKey(
'MoveMetaAilment', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMetaCategory(models.Model):
move_meta_category = models.ForeignKey(
'MoveMetaCategory', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMove(models.Model):
move = models.ForeignKey('Move', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMoveDamageClass(models.Model):
move_damage_class = models.ForeignKey(
'MoveDamageClass', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMoveEffect(models.Model):
move_effect = models.ForeignKey('MoveEffect', blank=True, null=True, on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMoveAttribute(models.Model):
move_attribute = models.ForeignKey(
'MoveAttribute', blank=True, null=True, on_delete=models.CASCADE)
class Meta:
abstract = True
class HasMoveTarget(models.Model):
move_target = models.ForeignKey(
'MoveTarget', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasName(models.Model):
name = models.CharField(max_length=100, db_index=True)
class Meta:
abstract = True
class HasNature(models.Model):
nature = models.ForeignKey('Nature', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasOrder(models.Model):
order = models.IntegerField(blank=True, null=True)
class Meta:
abstract = True
class HasPokeathlonStat(models.Model):
pokeathlon_stat = models.ForeignKey(
'PokeathlonStat', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokedex(models.Model):
pokedex = models.ForeignKey('Pokedex', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokemon(models.Model):
pokemon = models.ForeignKey('Pokemon', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokemonColor(models.Model):
pokemon_color = models.ForeignKey(
'PokemonColor', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokemonForm(models.Model):
pokemon_form = models.ForeignKey(
'PokemonForm', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokemonHabitat(models.Model):
pokemon_habitat = models.ForeignKey(
'PokemonHabitat', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
# HasPokemonMoveMethod
class HasMoveLearnMethod(models.Model):
move_learn_method = models.ForeignKey(
'MoveLearnMethod', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokemonShape(models.Model):
pokemon_shape = models.ForeignKey(
'PokemonShape', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasPokemonSpecies(models.Model):
pokemon_species = models.ForeignKey(
'PokemonSpecies', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasRegion(models.Model):
region = models.ForeignKey('Region', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasShortEffect(models.Model):
short_effect = models.CharField(max_length=300)
class Meta:
abstract = True
class HasStat(models.Model):
stat = models.ForeignKey('Stat', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasType(models.Model):
type = models.ForeignKey('Type', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasTypeEfficacy(models.Model):
damage_type = models.ForeignKey('Type', blank=True, null=True,
related_name="%(class)s_damage_type", on_delete=models.CASCADE)
target_type = models.ForeignKey('Type', blank=True, null=True,
related_name="%(class)s_target_type", on_delete=models.CASCADE)
damage_factor = models.IntegerField()
class Meta:
abstract = True
class HasVersion(models.Model):
version = models.ForeignKey('Version', blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class HasVersionGroup(models.Model):
version_group = models.ForeignKey(
'VersionGroup', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Meta:
abstract = True
class IsDescription(HasLanguage, HasDescription):
class Meta:
abstract = True
class IsName(HasLanguage, HasName):
class Meta:
abstract = True
class IsFlavorText(HasLanguage, HasFlavorText):
class Meta:
abstract = True
####################
# VERSION MODELS #
####################
class Version(HasName, HasVersionGroup):
pass
class VersionName(IsName, HasVersion):
pass
class VersionGroup(HasName, HasGeneration, HasOrder):
pass
class VersionGroupRegion(HasVersionGroup, HasRegion):
pass
# VersionGroupPokemonMoveMethod
class VersionGroupMoveLearnMethod(HasVersionGroup, HasMoveLearnMethod):
pass
#####################
# LANGUAGE MODELS #
#####################
class Language(HasName, HasOrder):
iso639 = models.CharField(max_length=2)
iso3166 = models.CharField(max_length=2)
official = models.BooleanField(default=False)
class LanguageName(IsName):
local_language = models.ForeignKey(
'Language', blank=True, null=True, related_name="locallanguage", on_delete=models.CASCADE)
#######################
# GENERATION MODELS #
#######################
class Generation(HasName):
region = models.OneToOneField('Region', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class GenerationName(IsName, HasGeneration):
pass
###################
# REGION MODELS #
###################
class Region(HasName):
pass
class RegionName(IsName, HasRegion):
pass
####################
# ABILITY MODELS #
####################
class Ability(HasName, HasGeneration):
is_main_series = models.BooleanField(default=False)
class AbilityEffectText(HasLanguage, HasEffect, HasShortEffect, HasAbility):
pass
class AbilityFlavorText(IsFlavorText, HasAbility, HasVersionGroup):
pass
class AbilityName(IsName, HasAbility):
pass
class AbilityChange(HasAbility, HasVersionGroup):
pass
class AbilityChangeEffectText(HasLanguage, HasEffect):
ability_change = models.ForeignKey(
AbilityChange, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
#################
# TYPE MODELS #
#################
class Type(HasName, HasGeneration, HasMoveDamageClass):
pass
class TypeName(IsName, HasType):
pass
class TypeGameIndex(HasType, HasGeneration, HasGameIndex):
pass
class TypeEfficacy(HasTypeEfficacy):
pass
# model for a type's efficacy that was used until a given generation
class TypeEfficacyPast(HasTypeEfficacy, HasGeneration):
pass
#################
# STAT MODELS #
#################
class Stat(HasName, HasMoveDamageClass):
is_battle_only = models.BooleanField(default=False)
game_index = models.IntegerField()
class StatName(IsName, HasStat):
pass
###########################
# CHARACTERISTIC MODELS #
###########################
class Characteristic(HasStat):
gene_mod_5 = models.IntegerField()
class CharacteristicDescription(HasCharacteristic, IsDescription):
pass
######################
# EGG GROUP MODELS #
######################
class EggGroup(HasName):
pass
class EggGroupName(IsName, HasEggGroup):
pass
#################
# ITEM MODELS #
#################
class ItemPocket(HasName):
pass
class ItemPocketName(IsName, HasItemPocket):
pass
class ItemCategory(HasName, HasItemPocket):
pass
class ItemCategoryName(IsName, HasItemCategory):
pass
class ItemFlingEffect(HasName):
pass
class ItemFlingEffectEffectText(HasLanguage, HasEffect, HasFlingEffect):
pass
class Item(HasName, HasItemCategory, HasFlingEffect):
cost = models.IntegerField(blank=True, null=True)
fling_power = models.IntegerField(blank=True, null=True)
class ItemEffectText(HasItem, HasLanguage, HasEffect, HasShortEffect):
pass
class ItemName(HasItem, IsName):
pass
class ItemFlavorText(HasItem, HasVersionGroup, IsFlavorText):
pass
class ItemAttribute(HasName):
pass
class ItemAttributeName(IsName, HasItemAttribute):
pass
class ItemAttributeDescription(IsDescription, HasItemAttribute):
pass
class ItemAttributeMap(HasItem, HasItemAttribute):
pass
class ItemGameIndex(HasItem, HasGeneration, HasGameIndex):
pass
class ItemSprites(HasItem):
sprites = models.CharField(max_length=500)
####################
# CONTEST MODELS #
####################
class ContestType(HasName):
pass
class ContestTypeName(HasContestType, IsName):
flavor = models.CharField(max_length=10)
color = models.CharField(max_length=10)
class ContestEffect(models.Model):
appeal = models.IntegerField()
jam = models.IntegerField()
class ContestEffectEffectText(HasLanguage, HasEffect, HasContestEffect):
pass
class ContestEffectFlavorText(HasLanguage, HasFlavorText, HasContestEffect):
pass
class ContestCombo(models.Model):
first_move = models.ForeignKey('Move', blank=True, null=True,
related_name='first_move', on_delete=models.CASCADE)
second_move = models.ForeignKey('Move', blank=True, null=True,
related_name='second_move', on_delete=models.CASCADE)
##################
# BERRY MODELS #
##################
class BerryFirmness(HasName):
pass
class BerryFirmnessName(IsName):
berry_firmness = models.ForeignKey(
BerryFirmness, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class Berry(HasName, HasItem):
berry_firmness = models.ForeignKey(
BerryFirmness, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
natural_gift_power = models.IntegerField()
natural_gift_type = models.ForeignKey(
Type, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
size = models.IntegerField()
max_harvest = models.IntegerField()
growth_time = models.IntegerField()
soil_dryness = models.IntegerField()
smoothness = models.IntegerField()
# Berry Flavors are a bit of a hack because their relationship
# in terms of flavors to contest types is really awkward the
# way it was handled in the veekun data set. Berry Flavor here
# does not match the csv table. Berry Flavor Map
# is a table fabricated just to suit this project.
class BerryFlavor(HasName):
contest_type = models.OneToOneField(
'ContestType', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class BerryFlavorName(IsName):
berry_flavor = models.ForeignKey(BerryFlavor, blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
class BerryFlavorMap(models.Model):
berry = models.ForeignKey(Berry, blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
berry_flavor = models.ForeignKey(BerryFlavor, blank=True, null=True,
related_name="%(class)s", on_delete=models.CASCADE)
potency = models.IntegerField()
########################
# GROWTH RATE MODELS #
########################
class GrowthRate(HasName):
formula = models.CharField(max_length=500)
class GrowthRateDescription(HasGrowthRate, IsDescription):
pass
###################
# NATURE MODELS #
###################
class Nature(HasName):
decreased_stat = models.ForeignKey(
Stat, blank=True, null=True, related_name='decreased', on_delete=models.CASCADE)
increased_stat = models.ForeignKey(
Stat, blank=True, null=True, related_name='increased', on_delete=models.CASCADE)
hates_flavor = models.ForeignKey(
BerryFlavor, blank=True, null=True, related_name='hates_flavor', on_delete=models.CASCADE)
likes_flavor = models.ForeignKey(
BerryFlavor, blank=True, null=True, related_name='likes_flavor', on_delete=models.CASCADE)
game_index = models.IntegerField()
class NatureName(IsName, HasNature):
pass
class NaturePokeathlonStat(HasNature, HasPokeathlonStat):
max_change = models.IntegerField()
class NatureBattleStylePreference(HasNature):
move_battle_style = models.ForeignKey(
'MoveBattleStyle', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
low_hp_preference = models.IntegerField()
high_hp_preference = models.IntegerField()
#####################
# LOCATION MODELS #
#####################
class Location(HasRegion, HasName):
pass
class LocationName(HasLocation, IsName):
pass
class LocationGameIndex(HasLocation, HasGeneration, HasGameIndex):
pass
class LocationArea(HasLocation, HasGameIndex, HasName):
pass
class LocationAreaName(IsName, HasLocationArea):
pass
class LocationAreaEncounterRate(HasEncounterMethod, HasLocationArea, HasVersion):
rate = models.IntegerField()
######################
# ENCOUNTER MODELS #
######################
class EncounterMethod(HasName, HasOrder):
pass
class EncounterMethodName(HasEncounterMethod, IsName):
pass
class EncounterSlot(HasVersionGroup, HasEncounterMethod):
slot = models.IntegerField(blank=True, null=True)
rarity = models.IntegerField()
class Encounter(HasVersion, HasLocationArea, HasPokemon):
encounter_slot = models.ForeignKey(EncounterSlot, blank=True,
null=True, on_delete=models.CASCADE)
min_level = models.IntegerField()
max_level = models.IntegerField()
class EncounterCondition(HasName):
pass
class EncounterConditionName(HasEncounterCondition, IsName):
pass
class EncounterConditionValue(HasEncounterCondition, HasName):
is_default = models.BooleanField(default=False)
class EncounterConditionValueName(IsName):
encounter_condition_value = models.ForeignKey(
EncounterConditionValue, blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
class EncounterConditionValueMap(models.Model):
encounter = models.ForeignKey(Encounter, blank=True, null=True, on_delete=models.CASCADE)
encounter_condition_value = models.ForeignKey(
EncounterConditionValue, blank=True, null=True, on_delete=models.CASCADE)
#################
# MOVE MODELS #
#################
class Move(
HasName, HasGeneration, HasType, HasMoveDamageClass, HasMoveEffect,
HasMoveTarget, HasContestType, HasContestEffect, HasSuperContestEffect):
power = models.IntegerField(blank=True, null=True)
pp = models.IntegerField(blank=True, null=True)
accuracy = models.IntegerField(blank=True, null=True)
priority = models.IntegerField(blank=True, null=True)
move_effect_chance = models.IntegerField(blank=True, null=True)
class MoveName(HasMove, IsName):
pass
class MoveFlavorText(HasMove, HasVersionGroup, IsFlavorText):
pass
class MoveChange(HasMove, HasVersionGroup, HasType, HasMoveEffect):
power = models.IntegerField(blank=True, null=True)
pp = models.IntegerField(blank=True, null=True)
accuracy = models.IntegerField(blank=True, null=True)
move_effect_chance = models.IntegerField(blank=True, null=True)
##############################
# MOVE DAMAGE CLASS MODELS #
##############################
class MoveDamageClass(HasName):
pass
class MoveDamageClassName(HasMoveDamageClass, IsName):
pass
class MoveDamageClassDescription(HasMoveDamageClass, IsDescription):
pass
##############################
# MOVE BATTLE STYLE MODELS #
##############################
class MoveBattleStyle(HasName):
pass
class MoveBattleStyleName(IsName):
move_battle_style = models.ForeignKey(
MoveBattleStyle, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
########################
# MOVE EFFECT MODELS #
########################
class MoveEffect(models.Model):
pass
class MoveEffectEffectText(HasLanguage, HasMoveEffect, HasEffect, HasShortEffect):
pass
class MoveEffectChange(HasMoveEffect, HasVersionGroup):
pass
class MoveEffectChangeEffectText(HasLanguage, HasEffect):
move_effect_change = models.ForeignKey(
'MoveEffectChange', blank=True, null=True, related_name="%(class)s",
on_delete=models.CASCADE)
################################
# MOVE FLAG/ATTRIBUTE MODELS #
################################
class MoveAttribute(HasName):
pass
class MoveAttributeName(HasMoveAttribute, IsName):
pass
class MoveAttributeDescription(HasMoveAttribute, IsDescription):
pass
class MoveAttributeMap(HasMove, HasMoveAttribute):
pass
########################
# MOVE TARGET MODELS #
########################
class MoveTarget(HasName):
pass
class MoveTargetName(HasMoveTarget, IsName):
pass
class MoveTargetDescription(HasMoveTarget, IsDescription):
pass
######################
# MOVE META MODELS #
######################
class MoveMeta(HasMetaAilment, HasMetaCategory):
move = models.OneToOneField(Move, blank=False, null=False, related_name="%(class)s",
on_delete=models.CASCADE)
min_hits = models.IntegerField(blank=True, null=True)
max_hits = models.IntegerField(blank=True, null=True)
min_turns = models.IntegerField(blank=True, null=True)
max_turns = models.IntegerField(blank=True, null=True)
drain = models.IntegerField(blank=True, null=True)
healing = models.IntegerField(blank=True, null=True)
crit_rate = models.IntegerField(blank=True, null=True)
ailment_chance = models.IntegerField(blank=True, null=True)
flinch_chance = models.IntegerField(blank=True, null=True)
stat_chance = models.IntegerField(blank=True, null=True)
class MoveMetaAilment(HasName):
pass
class MoveMetaAilmentName(HasMetaAilment, IsName):
pass
class MoveMetaCategory(HasName):
pass
class MoveMetaCategoryDescription(HasMetaCategory, IsDescription):
pass
class MoveMetaStatChange(HasMove, HasStat):
change = models.IntegerField()
#######################
# EXPERIENCE MODELS #
#######################
class Experience(HasGrowthRate):
level = models.IntegerField()
experience = models.IntegerField()
###################
# GENDER MODELS #
###################
class Gender(HasName):
pass
####################
# MACHINE MODELS #
####################
class Machine(HasGrowthRate, HasItem):
machine_number = models.IntegerField()
version_group = models.ForeignKey(VersionGroup, blank=True, null=True, on_delete=models.CASCADE)
move = models.ForeignKey(Move, blank=True, null=True, on_delete=models.CASCADE)
#######################
# POKEATHLON MODELS #
#######################
class PokeathlonStat(HasName):
pass
class PokeathlonStatName(IsName, HasPokeathlonStat):
pass
#####################
# PAL PARK MODELS #
#####################
class PalParkArea(HasName):
pass
class PalParkAreaName(IsName):
pal_park_area = models.ForeignKey(
PalParkArea, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class PalPark(HasPokemonSpecies):
pal_park_area = models.ForeignKey(
PalParkArea, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
base_score = models.IntegerField(blank=True, null=True)
rate = models.IntegerField()
##########################
# SUPER CONTEST MODELS #
##########################
class SuperContestEffect(models.Model):
appeal = models.IntegerField()
class SuperContestEffectFlavorText(IsFlavorText, HasSuperContestEffect):
pass
class SuperContestCombo(models.Model):
first_move = models.ForeignKey(Move, blank=True, null=True,
related_name='first', on_delete=models.CASCADE)
second_move = models.ForeignKey(Move, blank=True, null=True,
related_name='second', on_delete=models.CASCADE)
######################
# EVOLUTION MODELS #
######################
class EvolutionChain(models.Model):
baby_trigger_item = models.ForeignKey(Item, blank=True, null=True, on_delete=models.CASCADE)
class EvolutionTrigger(HasName):
pass
class EvolutionTriggerName(HasEvolutionTrigger, IsName):
pass
####################
# POKEDEX MODELS #
####################
class Pokedex(HasName, HasRegion):
is_main_series = models.BooleanField(default=False)
class PokedexName(HasPokedex, IsName):
pass
class PokedexDescription(HasPokedex, IsDescription):
pass
class PokedexVersionGroup(HasPokedex, HasVersionGroup):
pass
####################
# POKEMON MODELS #
####################
class PokemonSpecies(HasName, HasGeneration, HasPokemonColor,
HasPokemonShape, HasGrowthRate, HasOrder):
evolves_from_species = models.ForeignKey(
'self', blank=True, null=True, on_delete=models.CASCADE)
evolution_chain = models.ForeignKey(
EvolutionChain, blank=True, null=True, on_delete=models.CASCADE)
pokemon_habitat = models.ForeignKey(
'PokemonHabitat', blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
gender_rate = models.IntegerField()
capture_rate = models.IntegerField()
base_happiness = models.IntegerField()
is_baby = models.BooleanField(default=False)
hatch_counter = models.IntegerField()
has_gender_differences = models.BooleanField(default=False)
forms_switchable = models.BooleanField(default=False)
class PokemonSpeciesName(IsName, HasPokemonSpecies):
genus = models.CharField(max_length=30)
class PokemonSpeciesDescription(HasPokemonSpecies, IsDescription):
pass
class PokemonSpeciesFlavorText(IsFlavorText, HasPokemonSpecies, HasVersion):
pass
class Pokemon(HasName, HasPokemonSpecies, HasOrder):
height = models.IntegerField()
weight = models.IntegerField()
base_experience = models.IntegerField()
is_default = models.BooleanField(default=False)
class PokemonAbility(HasPokemon, HasAbility):
is_hidden = models.BooleanField(default=False)
slot = models.IntegerField()
class PokemonColor(HasName):
pass
class PokemonColorName(HasPokemonColor, IsName):
pass
class PokemonDexNumber(HasPokemonSpecies, HasPokedex):
pokedex_number = models.IntegerField()
class PokemonEggGroup(HasPokemonSpecies, HasEggGroup):
pass
class PokemonEvolution(HasEvolutionTrigger, HasGender):
evolution_item = models.ForeignKey(
Item, blank=True, null=True, related_name='evolution_item',
on_delete=models.CASCADE)
evolved_species = models.ForeignKey(
PokemonSpecies, related_name="evolved_species", blank=True, null=True,
on_delete=models.CASCADE)
min_level = models.IntegerField(blank=True, null=True)
location = models.ForeignKey(Location, related_name="location",
blank=True, null=True, on_delete=models.CASCADE)
held_item = models.ForeignKey(Item, blank=True, null=True,
related_name='held_item', on_delete=models.CASCADE)
time_of_day = models.CharField(max_length=10, blank=True, null=True)
known_move = models.ForeignKey(Move, blank=True, null=True, on_delete=models.CASCADE)
known_move_type = models.ForeignKey(
Type, related_name="known_move", blank=True, null=True, on_delete=models.CASCADE)
min_happiness = models.IntegerField(blank=True, null=True)
min_beauty = models.IntegerField(blank=True, null=True)
min_affection = models.IntegerField(blank=True, null=True)
relative_physical_stats = models.IntegerField(blank=True, null=True)
party_species = models.ForeignKey(
PokemonSpecies, related_name="party_species", blank=True, null=True,
on_delete=models.CASCADE)
party_type = models.ForeignKey(Type, related_name="party_type",
blank=True, null=True, on_delete=models.CASCADE)
trade_species = models.ForeignKey(
PokemonSpecies, related_name="trade_species", blank=True, null=True,
on_delete=models.CASCADE)
needs_overworld_rain = models.BooleanField(default=False)
turn_upside_down = models.BooleanField(default=False)
class PokemonForm(HasName, HasPokemon, HasOrder):
form_name = models.CharField(max_length=30)
version_group = models.ForeignKey(VersionGroup, blank=True, null=True, on_delete=models.CASCADE)
is_default = models.BooleanField(default=False)
is_battle_only = models.BooleanField(default=False)
is_mega = models.BooleanField(default=False)
form_order = models.IntegerField(blank=True, null=True)
class PokemonFormGeneration(HasPokemonForm, HasGeneration, HasGameIndex):
pass
class PokemonFormName(HasPokemonForm, IsName):
pokemon_name = models.CharField(max_length=30)
class PokemonFormSprites(HasPokemonForm):
sprites = models.CharField(max_length=500)
class PokemonGameIndex(HasPokemon, HasGameIndex, HasVersion):
pass
class PokemonHabitat(HasName):
pass
class PokemonHabitatName(IsName):
pokemon_habitat = models.ForeignKey(
PokemonHabitat, blank=True, null=True, related_name="%(class)s", on_delete=models.CASCADE)
class PokemonItem(HasPokemon, HasVersion, HasItem):
rarity = models.IntegerField()
# PokemonMoveMethod
class MoveLearnMethod(HasName):
pass
# PokemonMoveMethodName
class MoveLearnMethodName(IsName, HasMoveLearnMethod):
pass
class MoveLearnMethodDescription(IsDescription, HasMoveLearnMethod):
pass
class PokemonMove(HasPokemon, HasMoveLearnMethod, HasVersionGroup, HasMove, HasOrder):
level = models.IntegerField()
class PokemonShape(HasName):
pass
class PokemonShapeName(IsName):
awesome_name = models.CharField(max_length=30)
pokemon_shape = models.ForeignKey(
PokemonShape, blank=True, null=True,
related_name="pokemonshapename", on_delete=models.CASCADE
)
class PokemonStat(HasPokemon, HasStat):
base_stat = models.IntegerField()
effort = models.IntegerField()
class PokemonType(HasPokemon, HasType):
slot = models.IntegerField()
class PokemonSprites(HasPokemon):
sprites = models.CharField(max_length=500)