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 | |
parent | 60d1211a481a1f0987b5991537aff62b0d8ba774 (diff) | |
download | cliapp-4299e0691a55566865b0e8a9745ee94327c78bca.tar.gz |
Change: runcmd timeout/callback behaviour
-rw-r--r-- | NEWS | 4 | ||||
-rw-r--r-- | cliapp/runcmd.py | 13 | ||||
-rw-r--r-- | without-tests | 1 |
3 files changed, 14 insertions, 4 deletions
@@ -4,6 +4,10 @@ NEWS for cliapp Version 1.20170823+git, not yet released ---------------------------------------- +* The `runcmd` timeout behaviour has changed a little. Previously, a + timeout meant the command would be terminated. Termination is now + optional: if the `timeout_callback` function returns a true value, + termination happens. Otherwise not. Version 1.20170823, released 2017-08-23 ---------------------------------------- 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 diff --git a/without-tests b/without-tests index d7d5704..b381a46 100644 --- a/without-tests +++ b/without-tests @@ -12,4 +12,5 @@ ./test-plugins/aaa_hello_plugin.py example4.py example5.py +example6.py example_runcmd.py |