如何处理python调用的命令的输出

时间:2013-05-17 14:41:17

标签: python python-2.7

我想编写一个python程序,它调用ipcs并使用其输出来删除共享内存段和信号量。 我有一个有效的解决方案,但我觉得必须有更好的方法来做到这一点。 这是我的计划:

import subprocess

def getid(ip):
    ret=''
    while (output[ip]==' '):
        ip=ip+1
    while((output[ip]).isdigit()):
        ret=ret+output[ip]
        ip=ip+1
    return ret


print 'invoking ipcs'
output = subprocess.check_output(['ipcs'])
print output

for i in range (len(output)):
    if (output[i]=='m'):
        r=getid(i+1)
        print r
        if (r):
            op = subprocess.check_output(['ipcrm','-m',r])
            print op
    elif (output[i]=='s'):
        r=getid(i+1)
        print r
        if (r):
            op = subprocess.check_output(['ipcrm','-s',r])
            print op

print 'invoking ipcs'
output = subprocess.check_output(['ipcs'])
print output

特别是,有没有更好的方法来写“getid”?即不是逐个字符地解析它, 我可以逐字符串地解析它吗?

这是输出变量的样子(解析前):

Message Queues:
T     ID               KEY        MODE       OWNER    GROUP

Shared Memory:
T     ID               KEY        MODE       OWNER    GROUP
m 262144                    0 --rw-rw-rw- xyz              None
m 262145                    0 --rw-rw-rw- xyz              None
m 262146                    0 --rw-rw-rw- xyz              None
m 196611                    0 --rw-rw-rw- xyz              None
m 196612                    0 --rw-rw-rw- xyz              None
m 262151                    0 --rw-rw-rw- xyz              None

Semaphores:
T     ID               KEY        MODE       OWNER    GROUP
s 262144                    0 --rw-rw-rw- xyz              None
s 262145                    0 --rw-rw-rw- xyz              None
s 196610                    0 --rw-rw-rw- xyz              None

谢谢!

3 个答案:

答案 0 :(得分:1)

您可以在输出时逐行输出ipcs的输出。然后我会使用.strip().split()来解析每一行,以及类似try except块的内容,以确保该行符合您的条件。将其解析为字符流会使事情变得更复杂,我不推荐它。

import subprocess
proc = subprocess.Popen(['ipcs'],stdout=subprocess.PIPE)
for line in iter(proc.stdout.readline,''):
    line=line.strip().split()
    try:
        r = int(line[1])
    except:
        continue
    if line[0] == "m":
        op = subprocess.check_output(['ipcrm','-m',str(r)])
    elif line[0] == "s":
        op = subprocess.check_output(['ipcrm','-s',str(r)])
    print op
proc.wait()

答案 1 :(得分:1)

实际上没有必要一次迭代输出一个char。

首先,你应该将输出字符串分成行并迭代它们,一次处理一个。这是通过使用splitlines字符串方法完成的(有关详细信息,请参阅the docs)。

您可以使用split()进一步分割空白行,但考虑到输出的规律性,regular expression很适合该法案。基本上,如果第一个字符是ms,则下一个数字位数是您的ID,ms匹配是否决定您的下一个操作。

您可以使用名称来识别您识别的字符组,这样可以更方便地阅读正则表达式,并且可以更方便地处理由groupdict引起的结果。

import re
pattern = re.compile('^((?P<mem>m)|(?P<sem>s))\s+(?P<id>\d+)')

for line in output.splitlines():
    m = pattern.match(line)
    if m:
        groups = m.groupdict()
        _id = groups['id']
        if groups['mem']:
            print 'handling a memory line' 
            pass #  handle memory case
        else:
            print ' handling a semaphore line' 
            pass #  handle semaphore case

答案 2 :(得分:-1)

您可以使用字符串拆分方法根据字符串中的空格拆分字符串。

所以你可以使用,

for line in output:
   if (line.split(" ")[0] == 'm')):
      id = line.split(" ")[2]
      print id