diff options
author | Lars Wirzenius <liw@liw.fi> | 2018-01-21 16:59:12 +0200 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2018-01-21 16:59:12 +0200 |
commit | 4299e0691a55566865b0e8a9745ee94327c78bca (patch) | |
tree | 4a32cb8f37b655e77796ba2905434dfb1dbd3c33 /cliapp | |
parent | 60d1211a481a1f0987b5991537aff62b0d8ba774 (diff) | |
download | cliapp-4299e0691a55566865b0e8a9745ee94327c78bca.tar.gz |
Change: runcmd timeout/callback behaviour
Diffstat (limited to 'cliapp')
-rw-r--r-- | cliapp/runcmd.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/cliapp/runcmd.py b/cliapp/runcmd.py index 7123fed..818bafd 100644 --- a/cliapp/runcmd.py +++ b/cliapp/runcmd.py @@ -174,6 +174,7 @@ def _run_pipeline(procs, feed_stdin, pipe_stdin, pipe_stdout, pipe_stderr, pos = 0 io_size = 1024 latest_output = time.time() + timeout_quit = False timeout = False def set_nonblocking(fd): @@ -200,7 +201,7 @@ def _run_pipeline(procs, feed_stdin, pipe_stdin, pipe_stdout, pipe_stderr, return True # pragma: no cover return False - while not timeout and still_running(): + while not timeout_quit and still_running(): rlist = [] if not stdout_eof and pipe_stdout == subprocess.PIPE: rlist.append(procs[-1].stdout) @@ -255,14 +256,18 @@ def _run_pipeline(procs, feed_stdin, pipe_stdin, pipe_stdout, pipe_stderr, stderr_eof = True timeout = False + if timeout and timeout_callback: # pragma: no cover + result = timeout_callback() + if result: + timeout_quit = True + elif timeout: # pragma: no cover + timeout_quit = True + while not timeout and still_running(): for p in procs: if p.returncode is None: p.wait() - if timeout and timeout_callback: # pragma: no cover - timeout_callback() - errorcodes = [p.returncode for p in procs if p.returncode != 0] or [0] # Ensure that the pipeline doesn't leak file descriptors |