py: Detect unmatched tripple quote in repl helper.
diff --git a/py/repl.c b/py/repl.c
index 412ab20..bca1be5 100644
--- a/py/repl.c
+++ b/py/repl.c
@@ -29,10 +29,11 @@
return true;
}
- // also "compound" if unmatched open bracket
+ // also "compound" if unmatched open bracket or triple quote
int n_paren = 0;
int n_brack = 0;
int n_brace = 0;
+ int in_triple_quote = 0;
for (const char *l = line; *l; l++) {
switch (*l) {
case '(': n_paren += 1; break;
@@ -41,9 +42,15 @@
case ']': n_brack -= 1; break;
case '{': n_brace += 1; break;
case '}': n_brace -= 1; break;
+ case '"':
+ if (l[1] == '"' && l[2] == '"') {
+ l += 2;
+ in_triple_quote = 1 - in_triple_quote;
+ }
+ break;
}
}
- return n_paren > 0 || n_brack > 0 || n_brace > 0;
+ return n_paren > 0 || n_brack > 0 || n_brace > 0 || in_triple_quote != 0;
}
#endif // MICROPY_ENABLE_REPL_HELPERS