From d063b959a15254009379809bb60cc22b8f4b6af2 Mon Sep 17 00:00:00 2001 From: Tomas Babej Date: Thu, 15 Jan 2015 22:02:17 +0100 Subject: [PATCH 1/1] Task: Do not hardcode sys.stdin as input source for hooks --- tasklib/task.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tasklib/task.py b/tasklib/task.py index 0a74f6c..f21ea43 100644 --- a/tasklib/task.py +++ b/tasklib/task.py @@ -207,7 +207,7 @@ class Task(TaskResource): pass @classmethod - def from_input(cls, modify=False): + def from_input(cls, input_file=sys.stdin, 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 @@ -218,18 +218,21 @@ class Task(TaskResource): 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. + + Input_file argument can be used to specify the input file, + but defaults to sys.stdin. """ # TaskWarrior instance is set to None task = cls(None) # Load the data from the input - task._load_data(json.loads(sys.stdin.readline().strip())) + task._load_data(json.loads(input_file.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())) + task._update_data(json.loads(input_file.readline().strip())) return task -- 2.39.5