如何在python中调用外部程序并检索输出和返回代码？

Posted 2023-02-16

【中文标题】如何在python中调用外部程序并检索输出和返回代码？

【英文标题】：How to call an external program in python and retrieve the output and return code?

【发布时间】：2010-10-16 23:00:52

【问题描述】：

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

【问题讨论】：

有一些关于 SO 的现有问题和答案会对您有所帮助：***.com/questions/89228/…

这能回答你的问题吗？ How to execute a program or call a system command?

【参考方案1】：

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

from subprocess import Popen, PIPE

process = Popen(["ls", "-la", "."], stdout=PIPE)
(output, err) = process.communicate()
exit_code = process.wait()

【讨论】：

我已经编辑了上面的答案以反映 Ambroz 的建议，以防有人没有阅读 cmets 并使用以前不正确的代码。

如果由于某种原因这不起作用，您可能需要将 shell=True 添加到参数中（在 Windows 中时？）

看来the above solution可以用简单的调用subprocess.run()替换（需要Python >= 3.5）。

我可以像一行有错误一样获取错误行的行号，并且函数返回我的行号如 1.0？这样我就可以添加标签并突出显示错误。

为了获取错误输出，还应添加“stderr=PIPE”：process = Popen(["ls", "-la", "."], stdout=PIPE, stderr=PIPE)

【参考方案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 中）我在执行带有长输出的命令时遇到问题

【参考方案3】：

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

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

【讨论】：

我只复制粘贴了代码以查看它是否有效，我收到一个错误，out 的类型为 bytes，因此它不能在 write 方法中使用。此外，它会打印字符，但不会停止。

【参考方案4】：

根据 Ambroz Bizjak 之前的评论，这里有一个对我有用的解决方案：

import shlex
from subprocess import Popen, PIPE

cmd = "..."
process = Popen(shlex.split(cmd), stdout=PIPE)
process.communicate()
exit_code = process.wait()

【讨论】：

这是迄今为止最好的答案。

我有一个类似的帖子here 展示了如何从进程中获取三样东西：exitcode、stdout、stderr。

【参考方案5】：

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