Fix for exiting for loops.

2011-10-16 21:22:08 -07:00
parent 0a54f1a91d
commit 03b96592a2
1 changed files with 7 additions and 0 deletions
--- a/ASTree.cpp
+++ b/ASTree.cpp
@@ -1254,6 +1254,13 @@ PycRef<ASTNode> BuildFromCode(PycRef<PycCode> code, PycModule* mod)
                        curblock = blocks.top();
                    }
                }
+
+                if (curblock->blktype() == ASTBlock::BLK_FOR
+                        && curblock->end() == pos) {
+                    blocks.pop();
+                    blocks.top()->append(curblock.cast<ASTNode>());
+                    curblock = blocks.top();
+                }
            }
            break;
        case Pyc::POP_EXCEPT: