Add on-add hook

Author: lauft

on_modify.py

@@ -63,15 +63,13 @@ def extract_tags_from(json_obj):
 
 
 def extract_annotation_from(json_obj):
-
     if 'annotations' not in json_obj:
         return '\'\''
 
     return json_obj['annotations'][0]['description']
 
 
 def main(old, new):
-
     start_or_stop = ''
 
     # Started task.
@@ -104,7 +102,14 @@ def main(old, new):
 
 
 if __name__ == "__main__":
-    old = json.loads(input_stream.readline().decode("utf-8", errors="replace"))
-    new = json.loads(input_stream.readline().decode("utf-8", errors="replace"))
+    # Hook is called with two lines input ('old', 'new') in case of an on-modify event,
+    # but only with one line ('new') in case of an on-add event.
+    # We want to call the hook with an emtpy 'old' ('{}') in the latter case.
+    # Prepending '{}' makes sure this happens if only one line is added...
+    lines = ['{}'] + [line.decode("utf-8", errors="replace") for line in input_stream.readlines()]
+
+    new = json.loads(lines.pop())
+    old = json.loads(lines.pop())
+
     print(json.dumps(new))
     main(old, new)

test/docker/Dockerfile

@@ -55,6 +55,7 @@ WORKDIR /task-on-modify-hook
 # Install on-modify-hook
 RUN mkdir -p ~/.task/hooks
 RUN cp on_modify.py ~/.task/hooks/on-modify.timewarrior
+RUN ln -s on-add.timewarrior on-modify.timewarrior
 RUN chmod +x ~/.task/hooks/on-modify.timewarrior
 
 RUN echo "timew: $( timew --version )" ; \