Task: Make Task object non-iterable

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

diff --git a/tasklib/task.py b/tasklib/task.py
index 44c6965c1e3fdde7be836e48a4f8befb7f9a38c8..096f656d7d9e903ecebd3b6a2c36f58b568afb79 100644 (file)
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -34,6 +34,14 @@ class TaskResource(object):
         self._original_data = data.copy()
 
     def __getitem__(self, key):
         self._original_data = data.copy()
 
     def __getitem__(self, key):

+        # This is a workaround to make TaskResource non-iterable
+        # over simple index-based iteration
+        try:
+            int(key)
+            raise StopIteration
+        except ValueError:
+            pass
+

         hydrate_func = getattr(self, 'deserialize_{0}'.format(key),
                                lambda x: x)
         return hydrate_func(self._data.get(key))
         hydrate_func = getattr(self, 'deserialize_{0}'.format(key),
                                lambda x: x)
         return hydrate_func(self._data.get(key))