lazy: Properly override getattr calls

[etc/taskwarrior.git] / tasklib / lazy.py

diff --git a/tasklib/lazy.py b/tasklib/lazy.py
index e85a2047daecc92fbbe715ceb187b26b952d3816..04ccdf5216f7bee3f3b5c83d24b304c3e095bc22 100644 (file)
--- a/tasklib/lazy.py
+++ b/tasklib/lazy.py
@@ -29,7 +29,7 @@ class LazyUUIDTask(object):
         # Getattr is called only if the attribute could not be found using
         # normal means
         self.replace()
-        return self.name
+        return getattr(self, name)
 
     def __eq__(self, other):
         if other['uuid']:
@@ -39,6 +39,9 @@ class LazyUUIDTask(object):
     def __hash__(self):
         return self['uuid'].__hash__()
 
+    def __repr__(self):
+        return "LazyUUIDTask: {0}".format(self._uuid)
+
     @property
     def saved(self):
         """
@@ -80,11 +83,17 @@ class LazyUUIDTaskSet(object):
             raise AttributeError
         else:
             self.replace()
-            return self.name
+            return getattr(self, name)
+
+    def __repr__(self):
+        return "LazyUUIDTaskSet([{0}])".format(', '.join(self._uuids))
 
     def __eq__(self, other):
         return set(t['uuid'] for t in other) == self._uuids
 
+    def __ne__(self, other):
+        return not (self == other)
+
     def __contains__(self, task):
         return task['uuid'] in self._uuids
 
@@ -95,6 +104,22 @@ class LazyUUIDTaskSet(object):
         for uuid in self._uuids:
             yield LazyUUIDTask(self._tw, uuid)
 
+    def __sub__(self, other):
+        return LazyUUIDTaskSet(self._tw,
+                               self._uuids - set(t['uuid'] for t in other))
+
+    def __isub__(self, other):
+        self._uuids -= set(t['uuid'] for t in other)
+        return self
+
+    def __or__(self, other):
+        return LazyUUIDTaskSet(self._tw,
+                               self._uuids | set(t['uuid'] for t in other))
+
+    def __ior__(self, other):
+        self._uuids |= set(t['uuid'] for t in other)
+        return self
+
     def replace(self):
         """
         Performs conversion to the regular TaskQuerySet object, referenced by