add __iter__method to InstanceElement, because it's needed for arrays

2013-07-24 17:21:49 +02:00
parent c8648d9ca5
commit 284f2f1671
3 changed files with 9 additions and 3 deletions
--- a/jedi/evaluate_representation.py
+++ b/jedi/evaluate_representation.py
@@ -220,6 +220,10 @@ class InstanceElement(use_metaclass(cache.CachedMetaClass, pr.Base)):
                if not isinstance(command, unicode) else command
                for command in self.var.get_commands()]

+    def __iter__(self):
+        for el in self.var.__iter__():
+            yield InstanceElement(self.instance, el, self.is_class_var)
+
    def __getattr__(self, name):
        return getattr(self.var, name)