Python | 执行和解析 Linux 命令

Python 是一种流行的编程语言，它在执行和解析 Linux 命令方面非常有用。它可以通过内置的 os 模块和 subprocess 模块轻松执行 shell 命令，并使用 stdout 和 stderr 捕获命令的输出和错误。

执行简单的 Linux 命令

使用 Python 执行简单的 Linux 命令非常容易。您可以使用 os.system() 函数来执行命令。例如，下面的代码将在终端中打印出系统中当前的工作目录：

import os

os.system("pwd")

这将输出类似于以下内容的结果：

/Users/johndoe/Documents

捕获命令的输出

如果您想捕获命令的输出并对其进行分析，则可以使用 subprocess 模块。该模块提供了一个名为 run() 的函数，它可用于执行命令并捕获命令的输出和错误。例如，下面的代码将在终端中打印出当前用户的用户名：

import subprocess

cmd = "whoami"
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)

print(result.stdout.decode().strip())

这将输出类似于以下内容的结果：

johndoe

处理命令的错误

使用 subprocess 模块时，可以使用 returncode 属性检查执行命令时是否发生了错误。如果返回值为 0，则说明命令成功执行，否则可能需要进一步检查命令的输出或错误。

例如，下面的代码尝试移动一个不存在的文件，并检查是否收到相应的错误：

import subprocess

cmd = "mv nonexistent_file.txt new_file.txt"
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)

if result.returncode != 0:
    print("Error:", result.stderr.decode().strip())

输出类似于以下内容：

Error: mv: rename nonexistent_file.txt to new_file.txt: No such file or directory

使用管道和重定向

您可以在 Python 中使用管道和重定向执行复杂的命令。例如，下面的代码将启动一个名为 datagen 的进程，并使用管道和重定向来生成输出文件：

import subprocess

with open('output.txt', 'w') as f:
    cmd = "datagen --size 1000 --format csv"
    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
    f.write(result.stdout.decode().strip())

这将生成一个名为 output.txt 的文件，其中包含 datagen 命令的输出。

结论

在 Python 中执行和解析 Linux 命令非常容易。您可以使用 os 模块执行简单的命令，并使用 subprocess 模块获取命令的输出和错误。可以通过处理 returncode 属性来检查命令是否成功执行，并使用管道和重定向来处理复杂的命令。