管道子进程标准输出到变量

时间:2010-12-22 23:45:48

标签: python subprocess pipe python-2.6

我想使用子进程模块在pythong中运行命令,并将输出存储在变量中。但是,我不希望将命令的输出打印到终端。 对于此代码:

def storels():
   a = subprocess.Popen("ls",shell=True)
storels()

我在终端中获取目录列表,而不是将其存储在a中。我也试过了:

 def storels():
       subprocess.Popen("ls > tmp",shell=True)
       a = open("./tmp")
       [Rest of Code]
 storels()

这也会将ls的输出打印到我的终端。我甚至尝试使用有点过时的os.system方法执行此命令,因为在终端中运行ls > tmp根本不会向终端打印ls,而是将其存储在tmp中。但是,同样的事情发生了。

修改

在遵循marcog的建议后,我收到以下错误,但仅在运行更复杂的命令时。 cdrecord --help。 Python吐了出来:

Traceback (most recent call last):
  File "./install.py", line 52, in <module>
    burntrack2("hi")
  File "./install.py", line 46, in burntrack2
    a = subprocess.Popen("cdrecord --help",stdout = subprocess.PIPE)
  File "/usr/lib/python2.6/subprocess.py", line 633, in __init__
    errread, errwrite)
  File "/usr/lib/python2.6/subprocess.py", line 1139, in _execute_child
    raise child_exception
OSError: [Errno 2] No such file or directory

3 个答案:

答案 0 :(得分:119)

要获得ls的输出,请使用stdout=subprocess.PIPE

>>> proc = subprocess.Popen('ls', stdout=subprocess.PIPE)
>>> output = proc.stdout.read()
>>> print output
bar
baz
foo

命令cdrecord --help输出到stderr,所以你需要管道那个indstead。您还应该将命令分解为令牌列表,如下所示,或者替代方法是传递shell=True参数但是这会激活一个完全爆炸的shell,如果不这样做可能会很危险控制命令字符串的内容。

>>> proc = subprocess.Popen(['cdrecord', '--help'], stderr=subprocess.PIPE)
>>> output = proc.stderr.read()
>>> print output
Usage: wodim [options] track1...trackn
Options:
    -version    print version information and exit
    dev=target  SCSI target to use as CD/DVD-Recorder
    gracetime=# set the grace time before starting to write to #.
...

如果你有一个输出到stdout和stderr的命令并且想要合并它们,你可以通过将stderr管道输出到stdout然后捕获stdout来实现。

subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

正如Chris Morgan所述,您应该使用proc.communicate()代替proc.read()

>>> proc = subprocess.Popen(['cdrecord', '--help'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
>>> out, err = proc.communicate()
>>> print 'stdout:', out
stdout: 
>>> print 'stderr:', err
stderr:Usage: wodim [options] track1...trackn
Options:
    -version    print version information and exit
    dev=target  SCSI target to use as CD/DVD-Recorder
    gracetime=# set the grace time before starting to write to #.
...

答案 1 :(得分:29)

如果您使用的是python 2.7或更高版本,最简单的方法是使用subprocess.check_output()命令。这是一个例子:

output = subprocess.check_output('ls')

要重定向stderr,您可以使用以下内容:

output = subprocess.check_output('ls', stderr=subprocess.STDOUT)



如果要将参数传递给命令,可以使用列表或使用调用shell并使用单个字符串。

output = subprocess.check_output(['ls', '-a'])
output = subprocess.check_output('ls -a', shell=True)

答案 2 :(得分:10)

使用a = subprocess.Popen("cdrecord --help",stdout = subprocess.PIPE) ,您需要使用列表或使用shell=True;

其中任何一个都可行。前者更可取。

a = subprocess.Popen(['cdrecord', '--help'], stdout=subprocess.PIPE)

a = subprocess.Popen('cdrecord --help', shell=True, stdout=subprocess.PIPE)

此外,您应该使用Popen.stdout.read(请参阅子流程文档了解原因),而不是使用Popen.stderr.read / .communicate()

proc = subprocess.Popen(['cdrecord', '--help'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = proc.communicate()