travis: Drop support for Python2.6

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

diff --git a/tasklib/lazy.py b/tasklib/lazy.py
index 543abf039af5da4006478979d2c9033995371e5a..891eb5e9e83266fb4bb056b0d4d22b820959d2a0 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()
         # 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']:
 
     def __eq__(self, other):
         if other['uuid']:


@@ -83,7 +83,7 @@ class LazyUUIDTaskSet(object):
             raise AttributeError
         else:
             self.replace()
             raise AttributeError
         else:
             self.replace()

-            return self.name
+            return getattr(self, name)

 
     def __repr__(self):
         return "LazyUUIDTaskSet([{0}])".format(', '.join(self._uuids))
 
     def __repr__(self):
         return "LazyUUIDTaskSet([{0}])".format(', '.join(self._uuids))


@@ -105,21 +105,98 @@ class LazyUUIDTaskSet(object):
             yield LazyUUIDTask(self._tw, uuid)
 
     def __sub__(self, other):
             yield LazyUUIDTask(self._tw, uuid)
 
     def __sub__(self, other):

-        return LazyUUIDTaskSet(self._tw,
-                               self._uuids - set(t['uuid'] for t in other))
+        return self.difference(other)

 
     def __isub__(self, other):
 
     def __isub__(self, other):

-        self._uuids -= set(t['uuid'] for t in other)
-        return self
+        return self.difference_update(other)

 
 

-    def __or__(self, other):
+    def __rsub__(self, other):

         return LazyUUIDTaskSet(self._tw,
         return LazyUUIDTaskSet(self._tw,

-                               self._uuids | set(t['uuid'] for t in other))
+            set(t['uuid'] for t in other) - self._uuids)
+
+    def __or__(self, other):
+        return self.union(other)

 
     def __ior__(self, other):
 
     def __ior__(self, other):

+        return self.update(other)
+
+    def __ror__(self, other):
+        return self.union(other)
+
+    def __xor__(self, other):
+        return self.symmetric_difference(other)
+
+    def __ixor__(self, other):
+        return self.symmetric_difference_update(other)
+
+    def __rxor__(self, other):
+        return self.symmetric_difference(other)
+
+    def __and__(self, other):
+        return self.intersection(other)
+
+    def __iand__(self, other):
+        return self.intersection_update(other)
+
+    def __rand__(self, other):
+        return self.intersection(other)
+
+    def __le__(self, other):
+        return self.issubset(other)
+
+    def __ge__(self, other):
+        return self.issuperset(other)
+
+    def issubset(self, other):
+        return all([task in other for task in self])
+
+    def issuperset(self, other):
+        return all([task in self for task in other])
+
+    def union(self, other):
+        return LazyUUIDTaskSet(self._tw,
+            self._uuids | set(t['uuid'] for t in other))
+
+    def intersection(self, other):
+        return LazyUUIDTaskSet(self._tw,
+            self._uuids & set(t['uuid'] for t in other))
+
+    def difference(self, other):
+        return LazyUUIDTaskSet(self._tw,
+            self._uuids - set(t['uuid'] for t in other))
+
+    def symmetric_difference(self, other):
+        return LazyUUIDTaskSet(self._tw,
+            self._uuids ^ set(t['uuid'] for t in other))
+
+    def update(self, other):

         self._uuids |= set(t['uuid'] for t in other)
         return self
 
         self._uuids |= set(t['uuid'] for t in other)
         return self
 

+    def intersection_update(self, other):
+        self._uuids &= set(t['uuid'] for t in other)
+        return self
+
+    def difference_update(self, other):
+        self._uuids -= set(t['uuid'] for t in other)
+        return self
+
+    def symmetric_difference_update(self, other):
+        self._uuids ^= set(t['uuid'] for t in other)
+        return self
+
+    def add(self, task):
+        self._uuids.add(task['uuid'])
+
+    def remove(self, task):
+        self._uuids.remove(task['uuid'])
+
+    def pop(self):
+        return self._uuids.pop()
+
+    def clear(self):
+        self._uuids.clear()
+

     def replace(self):
         """
         Performs conversion to the regular TaskQuerySet object, referenced by
     def replace(self):
         """
         Performs conversion to the regular TaskQuerySet object, referenced by