Handle BrokenPipeError in SSH server.

prompt-toolkit · Aug 23, 2020 · 67d44cc · 67d44cc
1 parent cacbdf8
commit 67d44cc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/prompt_toolkit/contrib/ssh/server.py b/prompt_toolkit/contrib/ssh/server.py
@@ -34,8 +34,11 @@ def __init__(
         # in the SSH channel.
         class Stdout:
             def write(s, data):
-                if self._chan is not None:
-                    self._chan.write(data.replace("\n", "\r\n"))
+                try:
+                    if self._chan is not None:
+                        self._chan.write(data.replace("\n", "\r\n"))
+                except BrokenPipeError:
+                    pass  # Channel not open for sending.
 
             def flush(s):
                 pass