LazyUUIDTaskSet: Make the lazy iteration behaviour as defualt

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

diff --git a/tasklib/lazy.py b/tasklib/lazy.py
index 0bd93323762c7f8d265dd6ae455118d27f5b8f80..260acdfb4d5b24702f4553df9e30b56a44225878 100644 (file)
--- a/tasklib/lazy.py
+++ b/tasklib/lazy.py
@@ -66,8 +66,14 @@ class LazyUUIDTaskSet(object):
     def __getattr__(self, name):
         # Getattr is called only if the attribute could not be found using
         # normal means
-        self.replace()
-        return self.name
+
+        if name.startswith('__'):
+            # If some internal method was being search, do not convert
+            # to TaskQuerySet just because of that
+            raise AttributeError
+        else:
+            self.replace()
+            return self.name
 
     def __eq__(self, other):
         return set(t['uuid'] for t in other) == self._uuids
@@ -79,9 +85,8 @@ class LazyUUIDTaskSet(object):
         return len(self._uuids)
 
     def __iter__(self):
-        self.replace()
-        for task in self:
-            yield task
+        for uuid in self._uuids:
+            yield LazyUUIDTask(self._tw, uuid)
 
     def replace(self):
         """