在Linux环境下使用Python编写自动化脚本可以帮助简化重复性任务，提高工作效率。以下是学习和实践Python Linux自动化脚本编写的详细教程。 ### 基础知识准备 1. **熟悉Linux命令行**：了解基本的Linux命令是编写自动化脚本的前提条件。 2. **掌握Python基础语法**：需要对Python语言的基本语法有深入了解，包括变量、数据类型、控制结构、函数等。 ### 常用库和模块 为了处理不同的任务，Python提供了多种标准库和第三方库： - **os模块**：用于与操作系统进行交互，例如创建/删除文件夹、获取环境变量等。 - **subprocess模块**：可以启动新的进程，并连接到它们的输入/输出管道，实现更复杂的系统级操作。 - **shutil模块**：提供了一组高级的文件操作函数，如复制、移动文件或目录。 - **argparse模块**：如果计划让脚本接受命令行参数，则此模块非常有用。 - **logging模块**：对于调试和记录脚本运行状态非常重要[^2]。 ### 编写第一个自动化脚本 假设有一个需求是遍历所有NVMe设备并检查它们关联的CPU节点信息。这可以通过调用系统命令并解析其输出来完成。下面是一个简单的例子： ```python import os import subprocess def get_cpu_node_info(): # 获取当前系统的CPU节点信息 cpu_info = subprocess.check_output(['lscpu']).decode() print(cpu_info) def list_nvme_devices(): # 列出所有的NVMe设备 nvme_list_cmd = "nvme list | sed '1,2d' | awk '{print $1}' | awk -F '/' '{print $NF}'" result = subprocess.check_output(nvme_list_cmd, shell=True).decode().splitlines() return result def check_nvme_busid_and_node(nvme_device): # 为每个NVMe设备查找总线ID并通过PCI工具查询相关节点信息 busid_cmd = f"readlink -f /sys/block/{nvme_device} | cut -d '/' -f 6" busid = subprocess.check_output(busid_cmd, shell=True).decode().strip() print(f"busid = {busid}") pci_info_cmd = f"lspci -s {busid} -vv | grep -i node" pci_info = subprocess.check_output(pci_info_cmd, shell=True).decode() print(pci_info) print("======================================") if __name__ == "__main__": get_cpu_node_info() for device in list_nvme_devices(): check_nvme_busid_and_node(device) ``` 这段代码演示了如何将给定的shell脚本转换成Python版本，其中涉及到了执行shell命令、处理命令输出以及逻辑流程控制。 ### 实践建议 - 开始时尽量从简单的小型项目着手，逐步增加复杂度。 - 使用版本控制系统（如Git）管理你的脚本源码。 - 对于错误处理要谨慎，确保即使遇到异常情况也能妥善应对。 - 文档化你的脚本，包括用途说明、参数描述及示例用法等。 通过不断练习和完善自己的技能集，你可以构建出强大且高效的自动化解决方案。