LazyUUIDTaskSet: Add implementation for the and operator

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

diff --git a/tasklib/lazy.py b/tasklib/lazy.py
index c3feed574a81eed69f05c5e34733cb1450173602..891eb5e9e83266fb4bb056b0d4d22b820959d2a0 100644 (file)
--- a/tasklib/lazy.py
+++ b/tasklib/lazy.py
@@ -110,18 +110,37 @@ class LazyUUIDTaskSet(object):
     def __isub__(self, other):
         return self.difference_update(other)
 
     def __isub__(self, other):
         return self.difference_update(other)
 

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

     def __or__(self, other):
         return self.union(other)
 
     def __ior__(self, other):
         return self.update(other)
 
     def __or__(self, other):
         return self.union(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 __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 __le__(self, other):
         return self.issubset(other)