Merge pull request #95 from khaeru/develop

Check TASKRC environment var, default to ~/.taskrc
ralphbean · Feb 24, 2015 · 17133f2 · 17133f2
2 parents 57939f4 + 03b908b
commit 17133f2
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/taskw/warrior.py b/taskw/warrior.py
@@ -36,6 +36,10 @@
 
 open = lambda fname, mode: codecs.open(fname, mode, "utf-8")
 
+# Location of configuration file: either specified by TASKRC environment
+# variable, or ~/.taskrc (default).
+TASKRC = os.getenv("TASKRC", "~/.taskrc")
+
 
 class TaskWarriorBase(with_metaclass(abc.ABCMeta, object)):
     """ The task warrior
@@ -46,7 +50,7 @@ class TaskWarriorBase(with_metaclass(abc.ABCMeta, object)):
 
     def __init__(
         self,
-        config_filename="~/.taskrc",
+        config_filename=TASKRC,
         config_overrides=None,
         marshal=False
     ):
@@ -159,7 +163,7 @@ def filter_by(self, func):
         return filtered
 
     @classmethod
-    def load_config(cls, config_filename="~/.taskrc", overrides=None):
+    def load_config(cls, config_filename=TASKRC, overrides=None):
         """ Load ~/.taskrc into a python dict
 
         >>> config = TaskWarrior.load_config()
@@ -414,7 +418,7 @@ class TaskWarriorShellout(TaskWarriorBase):
 
     def __init__(
         self,
-        config_filename="~/.taskrc",
+        config_filename=TASKRC,
         config_overrides=None,
         marshal=False,
     ):