【Python】在代码中执行终端命令并获取输出和运行状态

2023-09-11 08:27

短信预约 -IT技能 免费直播动态提醒

文章目录

0 前言
1 os库
- 1.1 os.system
- 1.2 os.popen
2 subprocess库
- 2.1 subprocess.run
- 2.2 subprocess.Popen
3 参考链接

0 前言

在Python编程过程中，我们可能会遇到需要在终端命令行执行某个命令并获取其输出的操作，我们首先想到可能就是C语言中的system("pause");语句，确实，python当中也有类似的命令，同时，为了满足进一步的需求，比如判断指令是否执行完毕或者是得到执行输出的内容。

1 os库

1.1 os.system

这个指令仅在终端运行该指令，不能获取命令行返回的信息，但是它会在输出终端显示命令执行的输出。

# test.pyprint("Hello")# run.pyimport oscmd = "python test.py"os.system(cmd)# output:Hello

这个操作是会堵塞程序执行

1.2 os.popen

os.popen不仅可以执行命令，还可以返回执行结果

# test.pyprint("Hello")# run.pyimport oscmd = "python test.py"out = os.popen(cmd)print(out.readlines())# output:['Hello\n']

不过，这个库需要注意的是，如果命令执行输出有中文掺杂，最好不要使用这个命令，它的编码可能会存在一定问题。

2 subprocess库

subprocess库主要功能是输入输出流，在VS Code中打开它的说明，如下图所示。

在这里插入图片描述

从图中可以看出这个库的主要两个函数接口是run和Popen，往下翻还可以找到它其他的一些比较老的函数。这里介绍这两种主要函数的使用。

2.1 subprocess.run

根据上面的subprocess的说明，可以知道，run函数可以在命令行运行一个指令，并等待它运行完毕，并返回一个“已结束进程的实例”。这个函数的参数比较多，这里简单介绍几个。

在这里插入图片描述

args :  需要执行的命令行（cmd命令行，不是shell）stdout :  命令执行完的输出，可以设置为一个文件对象，或者是它自带的类PIPEstderr :  命令行执行完的报错信息，参数设置和stdout一样encoding :  编码设置，因为它默认的输出都是bytes，十六进制的，加上这个可以转换成字符串格式，如果不加上这个也可以在获取输出时decode一下，变成字符串。shell :  这个似乎不是指定在powershell中执行，但一般会加上 shell = True

下面是一个简单的实例

# test.pyprint("Hello")# run.pyimport subprocesscmd = "python test.py"b = subprocess.run(cmd, shell=True, stdout=subprocess.PIPE).stdoutprint(b.decode('gbk'))

需要注意，这里要使用gbk的编码方式，这个可能是和subprocess这个库有关，它内部默认使用的终端是cmd终端，编码方式有些特殊。如果使用默认的utf-8解码会出现报错 ‘utf-8’ codec can’t decode byte 0xb9 in position 0: invalid start byte ，具体原理和解决方式可以参考这篇博客。