Groups | Search | Server Info | Keyboard shortcuts | Login | Register [http] [https] [nntp] [nntps]

Groups > comp.lang.python > #97317

Re: Only getting the first 6 lines

Show all headers | View raw

On 01Oct2015 23:58, Cecil Westerhof <Cecil@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 <cs@zip.com.au>

Back to comp.lang.python | Previous | Next — Previous in thread | Next in thread | Find similar | Unroll thread

Thread

Only getting the first 6 lines Cecil Westerhof <Cecil@decebal.nl> - 2015-10-01 23:58 +0200
  Re: Only getting the first 6 lines Ian Kelly <ian.g.kelly@gmail.com> - 2015-10-01 16:58 -0600
  Re: Only getting the first 6 lines Cameron Simpson <cs@zip.com.au> - 2015-10-02 08:50 +1000
    Re: Only getting the first 6 lines Cecil Westerhof <Cecil@decebal.nl> - 2015-10-02 13:08 +0200
  Re: Only getting the first 6 lines Peter Otten <__peter__@web.de> - 2015-10-02 09:37 +0200
    Re: Only getting the first 6 lines Cecil Westerhof <Cecil@decebal.nl> - 2015-10-02 13:11 +0200

csiph-web