Iterate directly over result of get_tokens

sebastinas · sebastinas · commit 711845fa9963 · 2022-05-15T19:05:40.000+02:00
diff --git a/bpython/curtsiesfrontend/interpreter.py b/bpython/curtsiesfrontend/interpreter.py
@@ -96,11 +96,10 @@ def format(self, tbtext: str, lexer: Any) -> None:
         # FIXME: lexer should is a Lexer
         traceback_informative_formatter = BPythonFormatter(default_colors)
         traceback_code_formatter = BPythonFormatter({Token: ("d")})
-        tokens = list(lexer.get_tokens(tbtext))
 
         no_format_mode = False
         cur_line = []
-        for token, text in tokens:
+        for token, text in lexer.get_tokens(tbtext):
             if text.endswith("\n"):
                 cur_line.append((token, text))
                 if no_format_mode:
