Make plumbing respect the same env vars as the CLI. #180

[zzz-pokedex.git] / pokedex / lookup.py

diff --git a/pokedex/lookup.py b/pokedex/lookup.py
index cf6a15f..a75136b 100644 (file)
--- a/pokedex/lookup.py
+++ b/pokedex/lookup.py
@@ -104,6 +104,9 @@ class PokedexLookup(object):
 
         # Defaults
         if not directory:
+            directory = os.environ.get('POKEDEX_INDEX_DIR', None)
+
+        if not directory:
             directory = pkg_resources.resource_filename('pokedex',
                                                         'data/whoosh-index')
         self.directory = directory