On 01Oct2015 23:58, Cecil Westerhof <ce...@decebal.nl> wrote:
I want to get the first 6 lines of ps output. For this I use:
========================================================================
from subprocess import check_output
ps_command = ('ps', '-eo', 'user,pid,pcpu,pmem,stat,start,time,cmd', '--sort')
message = '\n'.join(check_output(ps_command +
('-%cpu',)).decode("utf-8").splitlines()[0:6])
========================================================================
It works, but does not look very efficient. Is there a better way to
do this?
It depends what you mean by inefficient. I'm presuming you mean that this:
- sucks in all the output of ps instead of just the first 6 lines
- sucks all the output into memory instead of just some of it
- does more in-memory work in splitlines()
- needs ps to run to completion instead of just long enough to print 6 lines
You could just read six lines of output from ps. Something like this
(untested):
lines = []
for lineno, line in \
enumerate(Popen(ps_command + ('-%cpu',),
stdin=NULL, stdout=PIPE).stdout,
1):
lines.append(line.decode("utf-8"))
if lineno >= 6:
break
This works because in a miracle of foresight you can split binary data streams
(the stdout) on line breaks, so you can fetch "binary" lines and decode them
individually.
This approach is important if the output of your command is large or slow, as
it does not need to suck the whole thing into memory or to wait for it to
finish a long procedure. With "ps" these issues are pretty minor; with some
other programs it can be significant, especially if the other program _doesn't_
terminate (consider "tail -f" of an active log file).
Cheers,
Cameron Simpson <c...@zip.com.au>
--
https://mail.python.org/mailman/listinfo/python-list
