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 moves, types, items, and abilities to lookup index. #15

Browse source
This commit is contained in:
Eevee 2009-07-26 23:18:11 -07:00
parent bf5ce11242
commit 76d7bbe0e7
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pokedex/lookup.py
View file

@ -13,13 +13,20 @@ import pokedex.db.tables as tables
# retrieve something from the index # retrieve something from the index
indexed_tables = {} indexed_tables = {}
for cls in [ for cls in [
tables.Ability,
tables.Item,
tables.Move,
tables.Pokemon, tables.Pokemon,
tables.Type,
]: ]:
indexed_tables[cls.__tablename__] = cls indexed_tables[cls.__tablename__] = cls
# Dictionary of extra keys to file types of objects under, e.g. Pokémon can # Dictionary of extra keys to file types of objects under, e.g. Pokémon can
# also be looked up purely by number # also be looked up purely by number
extra_keys = { extra_keys = {
tables.Move: [
lambda row: u"move %d" % row.id,
],
tables.Pokemon: [ tables.Pokemon: [
lambda row: unicode(row.id), lambda row: unicode(row.id),
], ],
@ -69,7 +76,7 @@ def get_index(session):
speller_entries.append(name) speller_entries.append(name)
for extra_key_func in extra_keys[cls]: for extra_key_func in extra_keys.get(cls, []):
extra_key = extra_key_func(row) extra_key = extra_key_func(row)
writer.add_document(name=extra_key, **row_key) writer.add_document(name=extra_key, **row_key)