Merge pull request #13073 from mgorny/rich-pipe-handling

pypa · Nov 9, 2024 · fe0925b · fe0925b · The-Lincoln · Nov 10, 2024
2 parents 4204359 + 099ae97
commit fe0925b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/news/13072.trivial.rst b/news/13072.trivial.rst
diff --git a/src/pip/_internal/utils/logging.py b/src/pip/_internal/utils/logging.py
@@ -137,12 +137,19 @@ def __rich_console__(
             yield Segment("\n")
 
 
+class PipConsole(Console):
+    def on_broken_pipe(self) -> None:
+        # Reraise the original exception, rich 13.8.0+ exits by default
+        # instead, preventing our handler from firing.
+        raise BrokenPipeError() from None
+
+
 class RichPipStreamHandler(RichHandler):
     KEYWORDS: ClassVar[Optional[List[str]]] = []
 
     def __init__(self, stream: Optional[TextIO], no_color: bool) -> None:
         super().__init__(
-            console=Console(file=stream, no_color=no_color, soft_wrap=True),
+            console=PipConsole(file=stream, no_color=no_color, soft_wrap=True),
             show_time=False,
             show_level=False,
             show_path=False,