From: Tomas Babej <tomasbabej@gmail.com>
Date: Fri, 16 Jan 2015 23:24:36 +0000 (+0100)
Subject: Task: Provide automatic hook type detection
X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/commitdiff_plain/303c3185205c2f39b2cd19b7f0adaf089ed1c8f7?ds=inline

Task: Provide automatic hook type detection
---

diff --git a/tasklib/task.py b/tasklib/task.py
index fc56a89..040c15b 100644
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -207,7 +207,7 @@ class Task(TaskResource):
         pass
 
     @classmethod
-    def from_input(cls, input_file=sys.stdin, modify=False):
+    def from_input(cls, input_file=sys.stdin, modify=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
@@ -226,6 +226,10 @@ class Task(TaskResource):
         # 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
+
         # Load the data from the input
         task._load_data(json.loads(input_file.readline().strip()))