diff options
Diffstat (limited to 'stmhal/pyexec.c')
-rw-r--r-- | stmhal/pyexec.c | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/stmhal/pyexec.c b/stmhal/pyexec.c index b960198ec7..298e58a5fd 100644 --- a/stmhal/pyexec.c +++ b/stmhal/pyexec.c @@ -204,15 +204,12 @@ friendly_repl_reset: continue; } - if (mp_repl_is_compound_stmt(vstr_str(&line))) { - for (;;) { - vstr_add_char(&line, '\n'); - int len = vstr_len(&line); - int ret = readline(&line, "... "); - if (ret == VCP_CHAR_CTRL_D || vstr_len(&line) == len) { - // done entering compound statement - break; - } + while (mp_repl_continue_with_input(vstr_str(&line))) { + vstr_add_char(&line, '\n'); + int ret = readline(&line, "... "); + if (ret == VCP_CHAR_CTRL_D) { + // stop entering compound statement + break; } } |