59

如何使用 python 脚本调用外部程序并检索输出和返回代码?

4

5 回答 5

82

查看subprocess模块:下面是一个简单的示例...

from subprocess import Popen, PIPE

process = Popen(["ls", "-la", "."], stdout=PIPE)
(output, err) = process.communicate()
exit_code = process.wait()
于 2009-04-01T19:28:30.133 回答
18

在 Ambroz Bizjak 之前的评论之后,这里有一个对我有用的解决方案:

import shlex
from subprocess import Popen, PIPE

cmd = "..."
process = Popen(shlex.split(cmd), stdout=PIPE)
process.communicate()
exit_code = process.wait()
于 2013-03-03T20:00:32.397 回答
4

经过一些研究,我有以下代码对我来说效果很好。它基本上实时打印标准输出和标准错误。希望它可以帮助其他需要它的人。

stdout_result = 1
stderr_result = 1


def stdout_thread(pipe):
    global stdout_result
    while True:
        out = pipe.stdout.read(1)
        stdout_result = pipe.poll()
        if out == '' and stdout_result is not None:
            break

        if out != '':
            sys.stdout.write(out)
            sys.stdout.flush()


def stderr_thread(pipe):
    global stderr_result
    while True:
        err = pipe.stderr.read(1)
        stderr_result = pipe.poll()
        if err == '' and stderr_result is not None:
            break

        if err != '':
            sys.stdout.write(err)
            sys.stdout.flush()


def exec_command(command, cwd=None):
    if cwd is not None:
        print '[' + ' '.join(command) + '] in ' + cwd
    else:
        print '[' + ' '.join(command) + ']'

    p = subprocess.Popen(
        command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd
    )

    out_thread = threading.Thread(name='stdout_thread', target=stdout_thread, args=(p,))
    err_thread = threading.Thread(name='stderr_thread', target=stderr_thread, args=(p,))

    err_thread.start()
    out_thread.start()

    out_thread.join()
    err_thread.join()

    return stdout_result + stderr_result
于 2014-03-14T02:57:53.417 回答
3

在此处查看子流程模块:http: //docs.python.org/library/subprocess.html#module-subprocess。它应该可以完成您需要的工作。

于 2009-04-01T19:30:23.763 回答
2

我开发了一个小库(py-execute),它允许您执行外部程序、检索输出和 retcode,同时在控制台中实时获取输出:

>>> from py_execute.process_executor import execute
>>> ret = execute('echo "Hello"')
Hello
>>> ret
(0, 'Hello\n')

您可以避免通过模拟 user_io 打印到控制台:

>>> from mock import Mock
>>> execute('echo "Hello"', ui=Mock())
(0, 'Hello\n')

我写它是因为使用普通 Popen(在 Python 2.7 中)我在执行带有长输出的命令时遇到了问题

于 2014-08-11T08:19:53.573 回答