Python如何获取命令行输出

Python是一种非常强大的编程语言，它提供了许多方法来获取命令行输出。无论是通过执行外部命令还是通过系统状态还是其他方式，Python都能够轻松地捕捉和处理命令行输出。

一、使用subprocess模块执行外部命令

subprocess模块是Python标准库中的一个强大工具，它允许您执行和处理外部命令。通过使用subprocess模块，您可以轻松地捕捉并处理命令行输出。

import subprocess

# 执行外部命令，获取标准输出和错误输出
cmd = "ls -l"
result = subprocess.run(cmd.split(), capture_output=True, text=True)

# 打印标准输出和错误输出
print(result.stdout)
print(result.stderr)

上面的代码演示了如何使用subprocess模块执行外部命令，并将输出捕获到变量中。通过参数capture_output=True和text=True，可以将标准输出和错误输出以字符串的形式获取。

二、使用os模块执行外部命令

除了subprocess模块外，Python的os模块也提供了执行外部命令的功能。通过使用os模块的函数os.popen，您可以执行外部命令并获取标准输出。

import os

# 执行外部命令，获取标准输出
cmd = "ls -l"
result = os.popen(cmd).read()

# 打印标准输出
print(result)

上面的代码演示了如何使用os模块执行外部命令，并将标准输出以字符串形式获取。

三、使用sys模块获取命令行输出

Python的sys模块提供了许多与解释器交互的功能。通过使用sys模块的sys.stdout，您可以获取当前的标准输出，并将其重定向到其他地方。

import sys

# 重定向标准输出到文件
with open("output.txt", "w") as f:
    sys.stdout = f
    print("Hello, World!")
    sys.stdout = sys.__stdout__

# 读取输出文件内容
with open("output.txt", "r") as f:
    result = f.read()

# 打印输出文件内容
print(result)

上面的代码演示了如何使用sys模块重定向标准输出到文件，并将输出文件内容读取到变量中。通过修改sys.stdout，可以将标准输出重定向到其他地方，如文件。

四、使用tee命令获取标准输出

在某些情况下，您可能想要同时获取命令行输出并将其显示在屏幕上。可以使用Linux系统的tee命令来实现这一目的。

import subprocess

# 执行外部命令，使用tee命令获取标准输出并显示在屏幕上
cmd = "ls -l | tee output.txt"
result = subprocess.run(cmd, shell=True)

上面的代码演示了如何使用subprocess模块执行带有tee命令的外部命令，并将标准输出同时显示在屏幕上和保存到文件中。

五、使用logging模块记录命令行输出

如果您希望将命令行输出记录到日志文件中，可以使用Python的logging模块。

import logging
import subprocess

# 创建logger对象
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)

# 创建文件处理器
file_handler = logging.FileHandler("output.log")
file_handler.setLevel(logging.INFO)

# 创建格式化器
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
file_handler.setFormatter(formatter)

# 将文件处理器添加到logger对象中
logger.addHandler(file_handler)

# 执行外部命令，捕获标准输出并写入日志文件
cmd = "ls -l"
result = subprocess.run(cmd.split(), capture_output=True, text=True)

# 将标准输出写入日志文件
logger.info(result.stdout)

上面的代码演示了如何使用logging模块将命令行输出写入日志文件中。通过设置logger对象和文件处理器的级别，可以控制日志的记录级别。

小结

Python提供了多种获取命令行输出的方法，包括使用subprocess模块执行外部命令、使用os模块执行外部命令、使用sys模块获取命令行输出、使用tee命令获取标准输出以及使用logging模块记录命令行输出等。根据具体的需求，您可以选择合适的方法来获取和处理命令行输出。