import atexit
import logging
+import selectors
import shlex
import subprocess
import sys
stderr=subprocess.PIPE,
universal_newlines=False,
)
- for char in iter(lambda: p.stdout.read(1), b''):
- sys.stdout.buffer.write(char)
- if char in line_enders:
- sys.stdout.flush()
- p.wait()
+ sel = selectors.DefaultSelector()
+ sel.register(p.stdout, selectors.EVENT_READ)
+ sel.register(p.stderr, selectors.EVENT_READ)
+ while True:
+ for key, _ in sel.select():
+ char = key.fileobj.read(1)
+ if not char:
+ p.wait()
+ return
+ if key.fileobj is p.stdout:
+ sys.stdout.buffer.write(char)
+ if char in line_enders:
+ sys.stdout.flush()
+ else:
+ sys.stderr.buffer.write(char)
+ if char in line_enders:
+ sys.stderr.flush()
def cmd_with_timeout(command: str, timeout_seconds: Optional[float]) -> int: