X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/710d5afab2bce3b087d4be58b9cb60f6facccd30..f881626c51f4a954f0330ef141b1b3ff944e35df:/tasklib/task.py diff --git a/tasklib/task.py b/tasklib/task.py index faf6604..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__) @@ -165,54 +168,10 @@ class TaskAnnotation(TaskResource): # their data dics are the same return self.task == other.task and self._data == other._data - __repr__ = __unicode__ - - -class LazyUUIDTask(object): - """ - A lazy wrapper around Task object, referenced by UUID. - - - Supports comparison with LazyUUIDTask or Task objects (equality by UUIDs) - - If any attribute other than 'uuid' requested, a lookup in the - backend will be performed and this object will be replaced by a proper - Task object. - """ - - def __init__(self, tw, uuid): - self._tw = tw - self._uuid = uuid - - def __getitem__(self, key): - # LazyUUIDTask does not provide anything else other than 'uuid' - if key is 'uuid': - return self._uuid - else: - self.replace() - return self[key] - - def __getattr__(self, name): - # Getattr is called only if the attribute could not be found using - # normal means - self.replace() - return self.name - - def __eq__(self, other): - if other['uuid']: - # For saved Tasks, just define equality by equality of uuids - return self['uuid'] == other['uuid'] + def __ne__(self, other): + return not self.__eq__(other) - def __hash__(self): - return self['uuid'].__hash__() - - def replace(self): - """ - Performs conversion to the regular Task object, referenced by the - stored UUID. - """ - - replacement = self._tw.tasks.get(uuid=self._uuid) - self.__class__ = replacement.__class__ - self.__dict__ = replacement.__dict__ + __repr__ = __unicode__ class Task(TaskResource): @@ -324,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 @@ -348,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 @@ -551,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.