pokeapi/pokemon_v2/models.py
Simón Oroño 218864dfbc Make the type of the sprites column in pokemon_v2_pokemonsprites jsonb
This is so Hasura automatically transforms the field into an array
instead of a JSON encoded in a string.

Migration was generated by the `makemigrations` command.
2023-11-26 11:57:16 -04:00

1799 lines
35 KiB
Python

from django.db import models
from django.contrib.postgres.fields import JSONField
#####################
# 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=10)
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=1000)
####################
# 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(blank=True, null=True)
capture_rate = models.IntegerField(blank=True, null=True)
base_happiness = models.IntegerField(blank=True, null=True)
is_baby = models.BooleanField(default=False)
is_legendary = models.BooleanField(default=False)
is_mythical = models.BooleanField(default=False)
hatch_counter = models.IntegerField(blank=True, null=True)
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(blank=True, null=True)
weight = models.IntegerField(blank=True, null=True)
base_experience = models.IntegerField(blank=True, null=True)
is_default = models.BooleanField(default=False)
class PokemonAbility(HasPokemon, HasAbility):
is_hidden = models.BooleanField(default=False)
slot = models.IntegerField()
# model for a Pokemon's abilities that were used until a given generation
class PokemonAbilityPast(HasPokemon, HasAbility, HasGeneration):
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=60)
class PokemonFormSprites(HasPokemonForm):
sprites = models.CharField(max_length=1000)
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 PokemonFormType(HasPokemonForm, HasType):
slot = models.IntegerField()
# model for a Pokemon's types that were used until a given generation
class PokemonTypePast(HasPokemon, HasType, HasGeneration):
slot = models.IntegerField()
class PokemonSprites(HasPokemon):
sprites = JSONField()