一、os系统模块
Python内置了os系统模块,可以用于执行与操作系统相关的命令,如文件操作、进程管理、环境变量等。其中,os.system()是最为基础和常用的函数,可以帮助我们执行Linux命令。
下面是os.system()的示例代码:
import os os.system('echo "hello world"')
在Linux终端下执行上述代码,结果为:
hello world
可以看到,os.system()可以帮助我们在Python脚本中执行Linux命令,并且可以获得Linux命令的输出结果。但是,它的缺点也非常明显:它无法获得Linux命令的返回值,因此无法判断命令是否执行成功。
二、subprocess模块
subprocess模块是Python中关于子进程的模块,可以帮助我们更加灵活地调用系统命令,并且能够获取到命令执行的返回值。
下面是subprocess模块的示例代码:
import subprocess cmd = 'echo "hello world"' p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) stdout, stderr = p.communicate() status = p.returncode print("stdout: ", stdout) print("stderr: ", stderr) print("status: ", status)
运行上述代码,输出结果为:
stdout: b'hello world\n' stderr: b'' status: 0
可以看到,我们使用subprocess.Popen()函数帮助我们执行了Linux命令,并获取到了Linux的输出结果、错误信息和返回值。其中,Popen()函数支持多种参数,比如:
- shell参数:用于控制是否启用shell
- stdin、stdout、stderr参数:用于控制输入输出的类型和流程
- timeout参数:用于控制执行命令的超时时间
- ...
三、subprocess模块的高级应用
除了上述基础的使用方式,subprocess模块还支持许多高级特性,比如:
- 支持管道操作
- 支持工作目录切换
- 支持环境变量的设置
- 支持文件描述符的传递
- 支持捕获子进程信号
- ...
下面是示例代码,用于展示subprocess模块的高级特性:
import os import subprocess # 管道操作 cmd1 = 'echo "hello" | sed s/hello/hi/g' p1 = subprocess.Popen(cmd1, shell=True) p1.wait() # 工作目录切换 os.chdir('/tmp') cmd2 = 'pwd' p2 = subprocess.Popen(cmd2, shell=True, stdout=subprocess.PIPE) stdout, _ = p2.communicate() print("cwd: ", stdout.strip()) # 环境变量的设置 env = os.environ.copy() env['PATH'] = env['PATH'] + ':/usr/local/bin' cmd3 = 'echo $PATH' p3 = subprocess.Popen(cmd3, shell=True, env=env, stdout=subprocess.PIPE) stdout, _ = p3.communicate() print("PATH: ", stdout.strip()) # 文件描述符传递 cmd4 = 'ls' p4 = subprocess.Popen(cmd4, shell=True, stdout=subprocess.PIPE) with open('/tmp/ls.txt', 'wb') as f: f.write(p4.stdout.read()) # 捕获子进程信号 def on_signal(signum, frame): print("Received signal ", signum) cmd5 = 'sleep 10' p5 = subprocess.Popen(cmd5, shell=True) signal.signal(signal.SIGTERM, on_signal) time.sleep(1) p5.terminate()
总的来说,subprocess模块是Python中一个非常强大的系统模块,参考官方文档使用时请一定要注意各种参数的使用规则,以及与其他模块的配合使用。