From 65e026e29cf14b9c8e314ae35be411c60ea18d92 Mon Sep 17 00:00:00 2001 From: Tomas Babej Date: Thu, 8 Jan 2015 01:56:29 +0100 Subject: [PATCH] Task: Add support for creating Task objects in hook scripts --- tasklib/task.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/tasklib/task.py b/tasklib/task.py index 1ab6347..c34a104 100644 --- a/tasklib/task.py +++ b/tasklib/task.py @@ -5,6 +5,7 @@ import json import logging import os import six +import sys import subprocess DATE_FORMAT = '%Y%m%dT%H%M%SZ' @@ -205,6 +206,33 @@ class Task(TaskResource): """ pass + @classmethod + def from_input(cls, modify=False): + """ + Creates a Task object, directly from the stdin, by reading one line. + If modify=True, two lines are used, first line interpreted as the + original state of the Task object, and second line as its new, + modified value. This is consistent with the TaskWarrior's hook + system. + + Object created by this method should not be saved, deleted + or refreshed, as t could create a infinite loop. For this + reason, TaskWarrior instance is set to None. + """ + + # TaskWarrior instance is set to None + task = cls(None) + + # Load the data from the input + task._load_data(json.loads(sys.stdin.readline().strip())) + + # If this is a on-modify event, we are provided with additional + # line of input, which provides updated data + if modify: + task._update_data(json.loads(sys.stdin.readline().strip())) + + return task + def __init__(self, warrior, **kwargs): self.warrior = warrior @@ -427,6 +455,12 @@ class Task(TaskResource): else: self._load_data(new_data) + def export_data(self): + """ + Exports current data contained in the Task as JSON + """ + + return json.dumps(self._data) class TaskFilter(SerializingObject): """ -- 2.39.2