如何在 Python 脚本中调用外部命令(就像我在 Unix shell 或 Windows 命令提示符下键入它一样)?
查看标准库中的子进程模块 :
from subprocess import call
call(["ls", "-l"])
子进程与系统的优势在于它更灵活(您可以获得 stdout,stderr,“真实” 状态代码,更好的错误处理等)。
官方文档推荐使用替代 os.system()的子进程模块:
子流程模块提供了更强大的工具来生成新流程并检索其结果; 使用该模块比使用此函数 [
os.system()
] 更os.system()
。
子流程文档中的 “ 用子流程模块替换旧函数 ” 部分可能有一些有用的配方。
子进程模块的官方文档:
以下是调用外部程序的方法及其优缺点的摘要:
os.system("some_command with args")
将命令和参数传递给系统的 shell。这很好,因为您实际上可以以这种方式一次运行多个命令并设置管道和输入 / 输出重定向。例如:
os.system("some_command < input_file | another_command > output_file")
但是,虽然这很方便,但您必须手动处理 shell 字符的转义,例如空格等。另一方面,这也允许您运行只是 shell 命令而不是外部程序的命令。请参阅文档 。
stream = os.popen("some_command with args")
将执行与os.system
相同的操作,除了它为您提供了一个类似文件的对象,您可以使用它来访问该进程的标准输入 / 输出。还有 3 种其他的 popen 变体,它们对 i / o 的处理方式略有不同。如果您将所有内容都作为字符串传递,那么您的命令将传递给 shell; 如果你将它们作为列表传递,那么你不必担心逃避任何事情。请参阅文档 。
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 函数中拥有所有选项是很好的。请参阅文档 。
subprocess
模块的call
函数。这基本上就像Popen
类一样,并且接受所有相同的参数,但它只是等待命令完成并为您提供返回代码。例如:
return_code = subprocess.call("echo Hello World", shell=True)
请参阅文档 。
如果您使用的是 Python 3.5 或更高版本,则可以使用新的subprocess.run
函数,该函数与上述类似,但更灵活,并在命令完成执行时返回CompletedProcess
对象。
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()
。