Stop treewalker escaping from subtree

jgraham · jgraham · commit 37b0fcc452f5 · 2009-10-15T21:27:20.000+02:00
diff --git a/src/html5lib/treewalkers/_base.py b/src/html5lib/treewalkers/_base.py
@@ -154,11 +154,12 @@ def __iter__(self):
                         namespace, name, attributes, hasChildren = details
                         if name not in voidElements:
                             yield self.endTag(namespace, name)
+                    if self.tree is currentNode:
+                        currentNode = None
+                        break
                     nextSibling = self.getNextSibling(currentNode)
                     if nextSibling is not None:
                         currentNode = nextSibling
                         break
-                    if self.tree is currentNode:
-                        currentNode = None
                     else:
                         currentNode = self.getParentNode(currentNode)
