Python标准库 - subproce

编写python脚本时, 经常要执行linux操作系统命令, 如mkdir zzzz. 目前比较推荐的方法是使用subprocess模块.

通过该模块的帮助文档, 可看到其主要提供了4个api, 和相应的使用说明.

Main API

========

call(...): Runs a command, waits for it to complete, then returns

    the return code.

check_call(...): Same as call() but raises CalledProcessError()

    if return code is not 0

check_output(...): Same as check_call() but returns the contents of

    stdout instead of a return code

Popen(...): A class for flexibly executing a command in a new process

结合API的说明来看, 若执行简单的命令, 像前面所说mkdir zzzz, 前三个API都可以, 只是处理细节稍有不同.

In [54]: subprocess.call("mkdir /tmp/zzzz", shell=True)

Out[54]: 0

In [55]: subprocess.call("mkdir /tmp/zzzz", shell=True)

mkdir: cannot create directory `/tmp/zzzz': File exists

Out[55]: 1

In [56]: subprocess.check_call("mkdir /tmp/zzzz", shell=True)

Out[56]: 0

In [57]: subprocess.check_call("mkdir /tmp/zzzz", shell=True)

mkdir: cannot create directory `/tmp/zzzz': File exists

---------------------------------------------------------------------------

CalledProcessError                        Traceback (most recent call last)

<iPython-input-57-2a531671f16e> in <module>()

----> 1 subprocess.check_call("mkdir /tmp/zzzz", shell=True)

/usr/local/python27/lib/python2.7/subprocess.pyc in check_call(*popenargs, **kwargs)

    184         if cmd is None:

    185             cmd = popenargs[0]

--> 186         raise CalledProcessError(retcode, cmd)

    187     return 0

    188

CalledProcessError: Command 'mkdir /tmp/zzzz' returned non-zero exit status 1

In [59]: subprocess.check_output("mkdir /tmp/zzzz", shell=True)

Out[59]: ''

In [60]: subprocess.check_output("mkdir /tmp/zzzz", shell=True)

mkdir: cannot create directory `/tmp/zzzz': File exists

---------------------------------------------------------------------------

CalledProcessError                        Traceback (most recent call last)

<ipython-input-60-4911eea4ecd3> in <module>()

----> 1 subprocess.check_output("mkdir /tmp/zzzz", shell=True)

/usr/local/python27/lib/python2.7/subprocess.pyc in check_output(*popenargs, **kwargs)

    217         if cmd is None:

    218             cmd = popenargs[0]

--> 219         raise CalledProcessError(retcode, cmd, output=output)

    220     return output

    221

CalledProcessError: Command 'mkdir /tmp/zzzz' returned non-zero exit status 1

又若执行复杂的命令或脚本, 需要获取其标准输出, 和标准错误输出, 就要用到Popen接口了. 前三个API, 其实是第四个的简化版, 是将参数直接传给了Popen的构造函数, 不过大部分参数都是保留了默认值, 来看下该构造函数的样子.

def __init__(self, args, bufsize=0, executable=None,

             stdin=None, stdout=None, stderr=None,

             preexec_fn=None, close_fds=False, shell=False,

             cwd=None, env=None, universal_newlines=False,

             startupinfo=None, creationflags=0):

注意参数shell=False, 其含义为, 是否将args(要执行的命令)置于操作系统的shell环境中运行.

下面给出一个代码段, 用到了Popen, 算是对subprocess模块的小结.

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import subprocess

def exec_cmd(cmd):

    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE,

                            stderr=subprocess.PIPE, shell=True)

    stdout, stderr = proc.communicate()

    if proc.returncode != 0:

        return proc.returncode, stderr

    return proc.returncode, stdout

def main():

    cmd = "ls -l"

    # cmd = "ls -lz"

    returncode, stdoutdata = exec_cmd(cmd)

    print str(returncode) + " - " + stdoutdata

if __name__ == '__main__':

    main()