在Python中,os.system()
函数用于执行系统命令
- 使用列表传递参数:将命令及其参数作为列表的元素传递,而不是将它们连接成一个字符串。这样可以避免shell注入攻击,并允许shell更有效地处理参数。
import os command = ["ls", "-l"] os.system(command)
- 使用
subprocess
模块:subprocess
模块提供了更多功能和更好的性能。它允许你与命令进行交互,捕获输出,设置环境变量等。使用subprocess.run()
函数执行命令是一个很好的选择。
import subprocess command = ["ls", "-l"] result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) print(result.stdout)
-
尽量避免使用
os.system()
:如果可能的话,尽量使用subprocess
模块或其他Python内置库(如os
、glob
、re
等)完成你的任务。这样可以提高代码的可读性和可维护性,同时减少潜在的安全风险。 -
并行执行命令:如果你需要同时执行多个命令,可以考虑使用
concurrent.futures
模块(特别是ThreadPoolExecutor
或ProcessPoolExecutor
类)来并行执行它们。这样可以充分利用多核处理器的性能,提高整体执行效率。
import concurrent.futures import subprocess commands = [ ["ls", "-l"], ["pwd"], ["uname", "-a"] ] with concurrent.futures.ThreadPoolExecutor() as executor: results = list(executor.map(lambda command: subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True), commands)) for result in results: print(result.stdout)
总之,尽量避免使用os.system()
,并优先考虑使用subprocess
模块或其他Python内置库来执行系统命令。这样可以提高执行效率,增强代码的可读性和可维护性,同时降低潜在的安全风险。