Tách đầu ra của ps bằng cách sử dụng Python

Question

Trên Linux, lệnh ps aux xuất ra một danh sách các quy trình có nhiều cột cho mỗi chỉ số. ví dụ.

USER  PID %CPU %MEM VSZ RSS TTY  STAT START TIME COMMAND 
... 
postfix 22611 0.0 0.2 54136 2544 ?  S 15:26 0:00 pickup -l -t fifo -u 
apache 22920 0.0 1.5 198340 16588 ?  S 09:58 0:05 /usr/sbin/httpd 

Tôi muốn có thể đọc điều này bằng cách sử dụng Python và tách từng hàng và sau đó mỗi cột để chúng có thể được sử dụng làm giá trị.

Đối với hầu hết các phần, đây không phải là một vấn đề:

ps = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE).communicate()[0] 
processes = ps.split('\n') 

bây giờ tôi có thể lặp qua các quy trình để có được mỗi hàng và chia nó ra bằng dấu cách, ví dụ

sep = re.compile('[\s]+') 
for row in processes: 
    print sep.split(row) 

Tuy nhiên, vấn đề là cột cuối cùng, lệnh, đôi khi có dấu cách. Trong ví dụ trên, bạn có thể thấy trong lệnh

pickup -l -t fifo -u 

mà sẽ được chia ra như

['postfix', '22611', '0.0', '0.2', '54136', '2544', '?', 'S', '15:26', '0:00', 'pickup', '-l', '-t', 'fifo', '-u'] 

nhưng tôi thực sự muốn nó như:

['postfix', '22611', '0.0', '0.2', '54136', '2544', '?', 'S', '15:26', '0:00', 'pickup -l -t fifo -u'] 

Vì vậy, câu hỏi của tôi là, làm thế nào tôi có thể chia ra các cột nhưng khi nói đến cột lệnh, giữ toàn bộ chuỗi như một phần tử danh sách chứ không phải chia nhỏ theo không gian?

Answer 1

Sử dụng thông số thứ hai để split chỉ định số trường tối đa để chia chuỗi thành. Tôi đoán bạn có thể tìm số bằng cách đếm số trường trong dòng đầu tiên, tức là tiêu đề cột.

ps = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE).communicate()[0] 
processes = ps.split('\n') 
# this specifies the number of splits, so the splitted lines 
# will have (nfields+1) elements 
nfields = len(processes[0].split()) - 1 
for row in processes[1:]: 
    print row.split(None, nfields) 

Answer 2

Các maxsplit đối số tùy chọn cho phương thức split có thể giúp bạn:

sep.split.(row, maxsplit=42) 

Answer 3

Tại sao bạn không sử dụng PSI để thay thế? PSI cung cấp thông tin quy trình trên Linux và các biến thể Unix khác.

import psi.process 
for p in psi.process.ProcessTable().values(): … 

Answer 4

Kiểm tra gói python.psutils.

psutil.process_iter trả về trình tạo mà bạn có thể sử dụng để lặp qua tất cả các quy trình. p.cmdline là danh sách các đối số cmdline của từng đối tượng quy trình, được tách riêng theo cách bạn muốn.

Bạn có thể tạo từ điển pids vs (pid,cmdline,path) chỉ với một dòng và sau đó sử dụng nó bất cứ khi nào bạn muốn.

pid_dict = dict([(p.pid, dict([('pid',p.pid), ('cmdline',p.cmdline), ('path',p.path)])) 
       for p in psutil.process_iter()])) 

Answer 5

Dưới đây là một thói quen và sử dụng thoải mái để giúp bạn đi:

def getProcessData(): 
    ps = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE).communicate()[0] 
    processes = ps.split('\n') 
    # this specifies the number of splits, so the splitted lines 
    # will have (nfields+1) elements 
    nfields = len(processes[0].split()) - 1 
    retval = [] 
    for row in processes[1:]: 
     retval.append(row.split(None, nfields)) 
    return retval 

wantpid = int(contents[0]) 
pstats = getProcessData() 
for ps in pstats: 
    if (not len(ps) >= 1): continue 
    if (int(ps[1]) == wantpid): 
     print "process data:" 
     print "USER    PID  %CPU  %MEM  VSZ  RSS  TTY  STAT  START TIME  COMMAND" 
     print "%-10.10s %10.10s %10.10s %10.10s %10.10s %10.10s %10.10s %10.10s %10.10s %s" % (ps[0], ps[1], ps[2], ps[3], ps[4], ps[5], ps[6], ps[7], ps[8], ps[9])