From: Scott <scott@wannabe.house>
Date: Fri, 21 Jan 2022 18:19:34 +0000 (-0800)
Subject: Wait for both streams to end.
X-Git-Url: https://wannabe.guru.org/gitweb/?a=commitdiff_plain;h=a4500713dc910eb6ddc3104fd9a58ce3a39d0e7f;p=python_utils.git

Wait for both streams to end.
---

diff --git a/exec_utils.py b/exec_utils.py
index a52e206..b4e88bd 100644
--- a/exec_utils.py
+++ b/exec_utils.py
@@ -30,12 +30,13 @@ def cmd_showing_output(command: str, ) -> int:
     sel = selectors.DefaultSelector()
     sel.register(p.stdout, selectors.EVENT_READ)
     sel.register(p.stderr, selectors.EVENT_READ)
-    should_exit = False
-    while not should_exit:
+    stream_ends = 0
+    while stream_ends < 2:
         for key, _ in sel.select():
             char = key.fileobj.read(1)
             if not char:
-                should_exit = True
+                stream_ends += 1
+                continue
             if key.fileobj is p.stdout:
                 sys.stdout.buffer.write(char)
                 if char in line_enders: