Task: Autmoatically detect TaskWarrior instance in from_input

[etc/taskwarrior.git] / tasklib / task.py

diff --git a/tasklib/task.py b/tasklib/task.py
index 0666b0f43a28386a9aa7dfb1ff588e7f17a84f59..50384bfbd2805aa04362b0868d94fd9905937d0c 100644 (file)
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -242,7 +242,7 @@ class SerializingObject(object):
 
     def normalize_uuid(self, value):
         # Enforce sane UUID
 
     def normalize_uuid(self, value):
         # Enforce sane UUID

-        if not isinstance(value, six.text_type) or value == '':
+        if not isinstance(value, six.string_types) or value == '':

             raise ValueError("UUID must be a valid non-empty string, "
                              "not: {}".format(value))
 
             raise ValueError("UUID must be a valid non-empty string, "
                              "not: {}".format(value))
 


@@ -385,7 +385,7 @@ class Task(TaskResource):
         pass
 
     @classmethod
         pass
 
     @classmethod

-    def from_input(cls, input_file=sys.stdin, modify=None):
+    def from_input(cls, input_file=sys.stdin, modify=None, warrior=None):

         """
         Creates a Task object, directly from the stdin, by reading one line.
         If modify=True, two lines are used, first line interpreted as the
         """
         Creates a Task object, directly from the stdin, by reading one line.
         If modify=True, two lines are used, first line interpreted as the


@@ -401,13 +401,18 @@ class Task(TaskResource):
         but defaults to sys.stdin.
         """
 
         but defaults to sys.stdin.
         """
 

-        # TaskWarrior instance is set to None
-        task = cls(None)
-

         # Detect the hook type if not given directly
         name = os.path.basename(sys.argv[0])
         modify = name.startswith('on-modify') if modify is None else modify
 
         # Detect the hook type if not given directly
         name = os.path.basename(sys.argv[0])
         modify = name.startswith('on-modify') if modify is None else modify
 

+        # Create the TaskWarrior instance if none passed
+        if warrior is None:
+            hook_parent_dir = os.path.dirname(os.path.dirname(sys.argv[0]))
+            warrior = TaskWarrior(data_location=hook_parent_dir)
+
+        # TaskWarrior instance is set to None
+        task = cls(warrior)
+

         # Load the data from the input
         task._load_data(json.loads(input_file.readline().strip()))
 
         # Load the data from the input
         task._load_data(json.loads(input_file.readline().strip()))