add default for HOME and PWD if not in ENV

3YOURMIND · Apr 20, 2019 · 5bb0699 · 5bb0699
1 parent 3e5c5db
commit 5bb0699
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vaultify/config.py b/vaultify/config.py
@@ -7,8 +7,8 @@
 MODULE_BASE_DIR = os.path.dirname(
     os.path.realpath(__file__))
 ETC_DEFAULT_CONFIG = '/etc/default/vaultify.yml'
-USER_CONFIG = '{}/.vaultify.yml'.format(os.environ.get("HOME"))
-LOCAL_CONFIG = '{}/.vaultify.yml'.format(os.environ.get("PWD"))
+USER_CONFIG = '{}/.vaultify.yml'.format(os.environ.get("HOME", '.'))
+LOCAL_CONFIG = '{}/.vaultify.yml'.format(os.environ.get("PWD", '.'))
 
 CFG_DEFAULT_FILES = [
     ETC_DEFAULT_CONFIG,