Fix WASI test failures

Author: ambv

Lib/_colorize.py

@@ -213,8 +213,13 @@ def can_colorize(*, file: IO[str] | IO[bytes] | None = None) -> bool:
 theme_no_color = default_theme.no_colors()
 
 
-def get_theme(*, tty_file: IO[str] | IO[bytes] | None = None) -> Theme:
-    if can_colorize(file=tty_file):
+def get_theme(
+    *,
+    tty_file: IO[str] | IO[bytes] | None = None,
+    force_color: bool = False,
+    force_no_color: bool = False,
+) -> Theme:
+    if force_color or (not force_no_color and can_colorize(file=tty_file)):
         return _theme
     return theme_no_color
 

Lib/_pyrepl/reader.py

@@ -492,11 +492,7 @@ def get_prompt(self, lineno: int, cursor_on_line: bool) -> str:
 
         if self.can_colorize:
             t = THEME()
-            prompt = (
-                f"{t.prompt}"
-                f"{prompt}"
-                f"{t.reset}"
-            )
+            prompt = f"{t.prompt}{prompt}{t.reset}"
         return prompt
 
     def push_input_trans(self, itrans: input.KeymapTranslator) -> None:

Lib/traceback.py

@@ -13,7 +13,6 @@
 import _colorize
 
 from contextlib import suppress
-from _colorize import get_theme, theme_no_color
 
 __all__ = ['extract_stack', 'extract_tb', 'format_exception',
            'format_exception_only', 'format_list', 'format_stack',
@@ -187,7 +186,10 @@ def format_exception_only(exc, /, value=_sentinel, *, show_group=False, **kwargs
 def _format_final_exc_line(etype, value, *, insert_final_newline=True, colorize=False):
     valuestr = _safe_string(value, 'exception')
     end_char = "\n" if insert_final_newline else ""
-    theme = (theme_no_color if not colorize else get_theme()).traceback
+    if colorize:
+        theme = _colorize.get_theme(force_color=True).traceback
+    else:
+        theme = _colorize.get_theme(force_no_color=True).traceback
     if value is None or not valuestr:
         line = f"{theme.type}{etype}{theme.reset}{end_char}"
     else:
@@ -534,8 +536,10 @@ def format_frame_summary(self, frame_summary, **kwargs):
         filename = frame_summary.filename
         if frame_summary.filename.startswith("<stdin>-"):
             filename = "<stdin>"
-
-        theme = (theme_no_color if not colorize else get_theme()).traceback
+        if colorize:
+            theme = _colorize.get_theme(force_color=True).traceback
+        else:
+            theme = _colorize.get_theme(force_no_color=True).traceback
         row.append(
             '  File {}"{}"{}, line {}{}{}, in {}{}{}\n'.format(
                 theme.filename,
@@ -1373,7 +1377,10 @@ def _format_syntax_error(self, stype, **kwargs):
         """Format SyntaxError exceptions (internal helper)."""
         # Show exactly where the problem was found.
         colorize = kwargs.get("colorize", False)
-        theme = (theme_no_color if not colorize else get_theme()).traceback
+        if colorize:
+            theme = _colorize.get_theme(force_color=True).traceback
+        else:
+            theme = _colorize.get_theme(force_no_color=True).traceback
         filename_suffix = ''
         if self.lineno is not None:
             yield '  File {}"{}"{}, line {}{}{}\n'.format(