veniasilente/veekun_pokedex
1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00
Code Issues Projects Releases Wiki Activity

Added some Pokémon methods to hide form weirdness. #5

Browse source
This commit is contained in:
Eevee 2009-06-15 20:48:27 -07:00
parent 61f973ab27
commit f9391b409a
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
pokedex/db/tables.py
View file

@ -205,6 +205,26 @@ class Pokemon(TableBase):
has_gen4_fem_sprite = Column(Boolean, nullable=False)
has_gen4_fem_back_sprite = Column(Boolean, nullable=False)
### Stuff to handle alternate Pokémon forms
@property
def national_id(self):
"""Returns the National Pokédex number for this Pokémon. Use this
instead of the id directly; alternate formes may make the id incorrect.
"""
if self.forme_base_pokemon_id:
return self.forme_base_pokemon_id
return self.id
@property
def full_name(self):
"""Returns the name of this Pokémon, including its Forme, if any."""
if self.forme_name:
return "%s %s" % (self.forme_name.capitalize(), self.name)
return self.name
class PokemonAbility(TableBase):
__tablename__ = 'pokemon_abilities'
pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)