- sel.register(p.stdout, selectors.EVENT_READ) # type: ignore
- sel.register(p.stderr, selectors.EVENT_READ) # type: ignore
- done = False
- while not done:
- for key, _ in sel.select():
- char = key.fileobj.read(1) # type: ignore
- if not char:
- sel.unregister(key.fileobj)
- if len(sel.get_map()) == 0:
- sys.stdout.flush()
- sys.stderr.flush()
- sel.close()
- done = True
- if key.fileobj is p.stdout:
- # sys.stdout.buffer.write(char)
- os.write(sys.stdout.fileno(), char)
- if char in line_enders:
- sys.stdout.flush()
- else:
- # sys.stderr.buffer.write(char)
- os.write(sys.stderr.fileno(), char)
- if char in line_enders:
- sys.stderr.flush()
- p.wait()
+ timer = None
+ if timeout_seconds:
+ import threading
+
+ timer = threading.Timer(timeout_seconds, timer_expired(p))
+ timer.start()
+ try:
+ sel.register(p.stdout, selectors.EVENT_READ) # type: ignore
+ sel.register(p.stderr, selectors.EVENT_READ) # type: ignore
+ done = False
+ while not done:
+ for key, _ in sel.select():
+ char = key.fileobj.read(1) # type: ignore
+ if not char:
+ sel.unregister(key.fileobj)
+ if len(sel.get_map()) == 0:
+ sys.stdout.flush()
+ sys.stderr.flush()
+ sel.close()
+ done = True
+ if key.fileobj is p.stdout:
+ os.write(sys.stdout.fileno(), char)
+ if char in line_enders:
+ sys.stdout.flush()
+ else:
+ os.write(sys.stderr.fileno(), char)
+ if char in line_enders:
+ sys.stderr.flush()
+ p.wait()
+ finally:
+ if timer:
+ timer.cancel()