mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
Added moves, types, items, and abilities to lookup index. #15
This commit is contained in:
parent
bf5ce11242
commit
76d7bbe0e7
1 changed files with 8 additions and 1 deletions
|
@ -13,13 +13,20 @@ import pokedex.db.tables as tables
|
|||
# retrieve something from the index
|
||||
indexed_tables = {}
|
||||
for cls in [
|
||||
tables.Ability,
|
||||
tables.Item,
|
||||
tables.Move,
|
||||
tables.Pokemon,
|
||||
tables.Type,
|
||||
]:
|
||||
indexed_tables[cls.__tablename__] = cls
|
||||
|
||||
# Dictionary of extra keys to file types of objects under, e.g. Pokémon can
|
||||
# also be looked up purely by number
|
||||
extra_keys = {
|
||||
tables.Move: [
|
||||
lambda row: u"move %d" % row.id,
|
||||
],
|
||||
tables.Pokemon: [
|
||||
lambda row: unicode(row.id),
|
||||
],
|
||||
|
@ -69,7 +76,7 @@ def get_index(session):
|
|||
|
||||
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)
|
||||
writer.add_document(name=extra_key, **row_key)
|
||||
|
||||
|
|
Loading…
Reference in a new issue