协慌网

登录 贡献 社区

在 Python 中调用外部命令

如何在 Python 脚本中调用外部命令(就像我在 Unix shell 或 Windows 命令提示符下键入它一样)?

答案

查看标准库中的子进程模块

from subprocess import call
call(["ls", "-l"])

子进程系统的优势在于它更灵活(您可以获得 stdout,stderr,“真实” 状态代码,更好的错误处理等)。

官方文档推荐使用替代 os.system()的进程模块:

子流程模块提供了更强大的工具来生成新流程并检索其结果; 使用该模块比使用此函数 [ os.system() ] 更os.system()

子流程文档中的 “ 用子流程模块替换旧函数 ” 部分可能有一些有用的配方。

子进程模块的官方文档:

以下是调用外部程序的方法及其优缺点的摘要:

  1. os.system("some_command with args")将命令和参数传递给系统的 shell。这很好,因为您实际上可以以这种方式一次运行多个命令并设置管道和输入 / 输出重定向。例如:

    os.system("some_command < input_file | another_command > output_file")

    但是,虽然这很方便,但您必须手动处理 shell 字符的转义,例如空格等。另一方面,这也允许您运行只是 shell 命令而不是外部程序的命令。请参阅文档

  2. stream = os.popen("some_command with args")将执行与os.system相同的操作,除了它为您提供了一个类似文件的对象,您可以使用它来访问该进程的标准输入 / 输出。还有 3 种其他的 popen 变体,它们对 i / o 的处理方式略有不同。如果您将所有内容都作为字符串传递,那么您的命令将传递给 shell; 如果你将它们作为列表传递,那么你不必担心逃避任何事情。请参阅文档

  3. subprocess模块的Popen类。这是为了取代os.popen但由于功能如此全面而具有稍微复杂的缺点。例如,你会说:

    print subprocess.Popen("echo Hello World", shell=True, stdout=subprocess.PIPE).stdout.read()

    代替:

    print os.popen("echo Hello World").read()

    但是在一个统一的类而不是 4 个不同的 popen 函数中拥有所有选项是很好的。请参阅文档

  4. subprocess模块的call函数。这基本上就像Popen类一样,并且接受所有相同的参数,但它只是等待命令完成并为您提供返回代码。例如:

    return_code = subprocess.call("echo Hello World", shell=True)

    请参阅文档

  5. 如果您使用的是 Python 3.5 或更高版本,则可以使用新的subprocess.run函数,该函数与上述类似,但更灵活,并在命令完成执行时返回CompletedProcess对象。

  6. os 模块还具有 C 程序中的所有 fork / exec / spawn 函数,但我不建议直接使用它们。

subprocess模块可能应该是你使用的。

最后请注意,对于所有将 shell 执行的最终命令作为字符串传递的方法,您负责转义它。如果您传递的字符串的任何部分无法完全受信任,则会产生严重的安全隐患 。例如,如果用户正在输入字符串的某些 / 任何部分。如果您不确定,请仅将这些方法与常量一起使用。为了给您一些暗示,请考虑以下代码:

print subprocess.Popen("echo %s " % user_input, stdout=PIPE).stdout.read()

并想象用户输入 “我的妈妈不爱我 && rm -rf /”。

我通常使用:

import subprocess

p = subprocess.Popen('ls', shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
for line in p.stdout.readlines():
    print line,
retval = p.wait()

您可以使用管道中的stdout数据自由地执行所需操作。实际上,您可以简单地省略这些参数( stdout=stderr= ),它的行为类似于os.system()