veniasilente/veekun_pokedex
1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-07-03 02:55:49 +00:00
Code Issues Projects Releases Wiki Activity

Use identifiers in Pokemon.stat

Browse source
This commit is contained in:
Petr Viktorin 2011-09-02 21:01:53 +03:00
parent 2a73d371f7
commit d8d32a0176
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
pokedex/db/tables.py
View file

@ -1084,19 +1084,19 @@ class Pokemon(TableBase):
def name(self):
return self.default_form.pokemon_name or self.species.name
def stat(self, stat_name):
def stat(self, stat_identifier):
u"""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
if isinstance(stat_identifier, Stat):
stat_identifier = stat_identifier.identifier
for pokemon_stat in self.stats:
if pokemon_stat.stat.name == stat_name:
if pokemon_stat.stat.identifier == stat_identifier:
return pokemon_stat
raise KeyError(u'No stat named %s' % stat_name)
raise KeyError(u'No stat named %s' % stat_identifier)
@property
def better_damage_class(self):
@ -1106,8 +1106,8 @@ class Pokemon(TableBase):
If the attack stats are about equal (within 5), returns None. The
value None, not the damage class called 'None'.
"""
phys = self.stat(u'Attack')
spec = self.stat(u'Special Attack')
phys = self.stat(u'attack')
spec = self.stat(u'special-attack')
diff = phys.base_stat - spec.base_stat