fix: gracefully handle SIGPIPE (#236)

python-poetry · Sep 2, 2022 · ac59662 · ac59662
1 parent aee0dae
commit ac59662
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/cleo/application.py b/cleo/application.py
@@ -325,6 +325,12 @@ def run(
 
             try:
                 exit_code = self._run(io)
+            except BrokenPipeError:
+                # If we are piped to another process, it may close early and send a
+                # SIGPIPE: https://docs.python.org/3/library/signal.html#note-on-sigpipe
+                devnull = os.open(os.devnull, os.O_WRONLY)
+                os.dup2(devnull, sys.stdout.fileno())
+                exit_code = 0
             except Exception as e:
                 if not self._catch_exceptions:
                     raise

diff --git a/cleo/io/outputs/stream_output.py b/cleo/io/outputs/stream_output.py
@@ -77,6 +77,7 @@ def _write(self, message: str, new_line: bool = False) -> None:
             message += "\n"
 
         self._stream.write(message)
+        self._stream.flush()
 
     def _has_color_support(self) -> bool:
         # Follow https://no-color.org/