Merge pull request #460 from TiZCrocodile/support-for-`WITH_CLEANUP_S…

…TART`,`WITH_CLEANUP_FINISH` add support for `WITH_CLEANUP_START`,`WITH_CLEANUP_FINISH`
zrax · Feb 27, 2024 · bb342a9 · bb342a9
2 parents d361ec3 + ab80ec1
commit bb342a9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/ASTree.cpp b/ASTree.cpp
@@ -1885,6 +1885,7 @@ PycRef<ASTNode> BuildFromCode(PycRef<PycCode> code, PycModule* mod)
             }
             break;
         case Pyc::WITH_CLEANUP:
+        case Pyc::WITH_CLEANUP_START:
             {
                 // Stack top should be a None. Ignore it.
                 PycRef<ASTNode> none = stack.top();
@@ -1907,6 +1908,9 @@ PycRef<ASTNode> BuildFromCode(PycRef<PycCode> code, PycModule* mod)
                 }
             }
             break;
+        case Pyc::WITH_CLEANUP_FINISH:
+            /* Ignore this */
+            break;
         case Pyc::SETUP_EXCEPT_A:
             {
                 if (curblock->blktype() == ASTBlock::BLK_CONTAINER) {

diff --git a/tests/xfail/test_with.3.4.pyc b/tests/xfail/test_with.3.4.pyc
diff --git a/tests/xfail/test_with.3.5.pyc b/tests/xfail/test_with.3.5.pyc