X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/9eaa4f203355eb57f95596408568f72fd723fee4..dbaa652d6c0d37f40d013e0129235b6d50335868:/tasklib/task.py diff --git a/tasklib/task.py b/tasklib/task.py index f0bb3a8..c7c9e6d 100644 --- a/tasklib/task.py +++ b/tasklib/task.py @@ -13,6 +13,9 @@ DATE_FORMAT = '%Y%m%dT%H%M%SZ' REPR_OUTPUT_SIZE = 10 PENDING = 'pending' COMPLETED = 'completed' +DELETED = 'deleted' +WAITING = 'waiting' +RECURRING = 'recurring' logger = logging.getLogger(__name__) @@ -39,7 +42,7 @@ class ReadOnlyDictView(object): return len(self.viewed_dict) def __unicode__(self): - return u'ReadOnlyDictView: {0}'.format(repr(self.viewed_dict)) + return six.u('ReadOnlyDictView: {0}'.format(repr(self.viewed_dict))) __repr__ = __unicode__ @@ -165,6 +168,9 @@ class TaskAnnotation(TaskResource): # their data dics are the same return self.task == other.task and self._data == other._data + def __ne__(self, other): + return not self.__eq__(other) + __repr__ = __unicode__ @@ -277,6 +283,9 @@ class Task(TaskResource): # If the tasks are not saved, compare the actual instances return id(self) == id(other) + def __ne__(self, other): + return not self.__eq__(other) + def __hash__(self): if self['uuid']: # For saved Tasks, just define equality by equality of uuids @@ -301,6 +310,10 @@ class Task(TaskResource): def pending(self): return self['status'] == six.text_type('pending') + @property + def recurring(self): + return self['status'] == six.text_type('recurring') + @property def active(self): return self['start'] is not None @@ -504,6 +517,15 @@ class TaskQuerySet(object): def completed(self): return self.filter(status=COMPLETED) + def deleted(self): + return self.filter(status=DELETED) + + def waiting(self): + return self.filter(status=WAITING) + + def recurring(self): + return self.filter(status=RECURRING) + def filter(self, *args, **kwargs): """ Returns a new TaskQuerySet with the given filters added.