veekun_pokedex/pokedex/defaults.py
a_magical_me ffc30bff8f Factor out logic for finding the default db/index. #180
Note: `if not x:` has changed to `if x is not None:`, changing the
semantics slightly.  Shouldn't be a big issue.
2010-05-13 21:45:51 -07:00

45 lines
1.2 KiB
Python

""" pokedex.defaults - logic for finding default paths """
import os
import pkg_resources
def get_default_db_uri_with_origin():
uri = os.environ.get('POKEDEX_DB_ENGINE', None)
origin = 'environment'
if uri is None:
sqlite_path = pkg_resources.resource_filename('pokedex',
'data/pokedex.sqlite')
uri = 'sqlite:///' + sqlite_path
origin = 'default'
return uri, origin
def get_default_index_dir_with_origin():
index_dir = os.environ.get('POKEDEX_INDEX_DIR', None)
origin = 'environment'
if index_dir is None:
index_dir = pkg_resources.resource_filename('pokedex',
'data/whoosh-index')
origin = 'default'
return index_dir, origin
def get_default_csv_dir_with_origin():
csv_dir = pkg_resources.resource_filename('pokedex', 'data/csv')
origin = 'default'
return csv_dir, origin
def get_default_db_uri():
return get_default_db_uri_with_origin()[0]
def get_default_index_dir():
return get_default_index_dir_with_origin()[0]
def get_default_csv_dir():
return get_default_csv_dir_with_origin()[0]