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 Pokemon.stat() accessor.

Browse source
This commit is contained in:
Eevee 2010-04-15 20:05:03 -07:00
parent e73eca020c
commit 4ae8bb5f21
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pokedex/db/tables.py
View file

@ -350,6 +350,22 @@ class Pokemon(TableBase):
return self
### Not forms!
def stat(self, stat_name):
"""Returns a PokemonStat record for the given stat name (or Stat row
object). Uses the normal has-many machinery, so all the stats are
effectively cached.
"""
if isinstance(stat_name, Stat):
stat_name = stat_name.name
for pokemon_stat in self.stats:
if pokemon_stat.stat.name == stat_name:
return pokemon_stat
return None
class PokemonAbility(TableBase):
__tablename__ = 'pokemon_abilities'
pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False)