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

Lookup now respects "ability" as a lookup type.

Browse source
This commit is contained in:
Eevee 2010-04-30 00:22:03 -07:00
parent 97f96a5695
commit 9ca0f486f6
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pokedex/lookup.py
View file

@ -290,13 +290,15 @@ class PokedexLookup(object):
Returns None for a bogus name.
"""
# Table object
if hasattr(name, '__tablename__'):
return getattr(name, '__tablename__')
elif name in self.indexed_tables:
return name
elif name + 's' in self.indexed_tables:
return name + 's'
else:
# Table name
for table in self.indexed_tables.values():
if name in (table.__tablename__, table.__singlename__):
return table.__tablename__
# Bogus. Be nice and return dummy
return None