py: Combine continuous block of emit steps into with_cleanup emit call.

Because different emitters need to handle with-cleanup in different ways.

1 files changed, 4 insertions, 1 deletions

diff --git a/py/emitbc.c b/py/emitbc.c
index 14298bad4..d871aa4ce 100644
--- a/py/emitbc.c
+++ b/py/emitbc.c
@@ -758,7 +758,10 @@ void mp_emit_bc_setup_with(emit_t *emit, mp_uint_t label) {
     emit_write_bytecode_byte_unsigned_label(emit, MP_BC_SETUP_WITH, label);
 }
 
-void mp_emit_bc_with_cleanup(emit_t *emit) {
+void mp_emit_bc_with_cleanup(emit_t *emit, mp_uint_t label) {
+    mp_emit_bc_pop_block(emit);
+    mp_emit_bc_load_const_tok(emit, MP_TOKEN_KW_NONE);
+    mp_emit_bc_label_assign(emit, label);
     emit_bc_pre(emit, -4);
     emit_write_bytecode_byte(emit, MP_BC_WITH_CLEANUP);
 }