X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/e9ee75194dcb6c60ff6b204f6b6ad135d4859628..edc8896ea467a9d873fcb9f478dcbf2b2638c669:/tasklib/task.py?ds=sidebyside diff --git a/tasklib/task.py b/tasklib/task.py index e8e5dfd..5c05bae 100644 --- a/tasklib/task.py +++ b/tasklib/task.py @@ -532,8 +532,24 @@ class Task(TaskResource): self.warrior.execute_command([self['uuid'], 'delete']) # Refresh the status again, so that we have updated info stored + self.refresh(only_fields=['status', 'start', 'end']) + + def start(self): + if not self.saved: + raise Task.NotSaved("Task needs to be saved before it can be started") + + # Refresh, and raise exception if task is already completed/deleted self.refresh(only_fields=['status']) + if self.completed: + raise Task.CompletedTask("Cannot start a completed task") + elif self.deleted: + raise Task.DeletedTask("Deleted task cannot be started") + + self.warrior.execute_command([self['uuid'], 'start']) + + # Refresh the status again, so that we have updated info stored + self.refresh(only_fields=['status', 'start']) def done(self): if not self.saved: @@ -550,7 +566,7 @@ class Task(TaskResource): self.warrior.execute_command([self['uuid'], 'done']) # Refresh the status again, so that we have updated info stored - self.refresh(only_fields=['status']) + self.refresh(only_fields=['status', 'start', 'end']) def save(self): if self.saved and not self.modified: