👍 Add custom wradex directory feature

astropenguin · Apr 23, 2019 · ddfc61a · ddfc61a
1 parent 7c8a295
commit ddfc61a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/wradex/constants.py b/wradex/constants.py
@@ -7,12 +7,18 @@
 ]
 
 # standard library
+import os
 from pathlib import Path
 
 # dependent packages
 import wradex
 
 # local constants
-WRADEX_DIR = Path('~', '.wradex').expanduser()
+if 'WRADEX_DIR' in os.environ:
+    path = os.environ.get('WRADEX_DIR')
+    WRADEX_DIR = Path(path).expanduser()
+else:
+    WRADEX_DIR = Path.home() / '.wradex'
+
 WRADEX_CONFIG  = Path(WRADEX_DIR, 'config.yaml')
 _WRADEX_CONFIG = Path(*wradex.__path__, 'data', 'config.yaml')