Factor out logic for finding the default db/index. #180

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

diff --git a/pokedex/defaults.py b/pokedex/defaults.py
new file mode 100644 (file)
index 0000000..7321291
--- /dev/null
+++ b/pokedex/defaults.py
@@ -0,0 +1,45 @@
+""" 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]
+
+