需求说明：要为系统运维工程师编写一个 Python 脚本，使用 psutil 库收集 100 台 Linux 主机的系统信息（如 IP 地址、主机名、操作系统型号、内存大小、磁盘大小等），并将这些信息存储到 CSV 文件中，以下是一个完整的实现方案。

实现步骤

1. 使用 psutil 收集本地系统信息：
2. 获取主机名、操作系统型号、内存大小、磁盘大小等信息。
3. 使用 paramiko 或 fabric 远程连接到其他主机：
4. 通过 SSH 连接到远程主机，执行命令获取信息。
5. 将收集到的信息存储到 CSV 文件中：
6. 使用 Python 的 csv 模块将数据写入 CSV 文件。

代码实现

1. 安装依赖库

确保安装了以下 Python 库：

pip install psutil paramiko

2. 脚本代码

import psutil
import socket
import platform
import csv
import paramiko

# 定义远程主机信息
REMOTE_HOSTS = [
    {"hostname": "192.168.1.101", "username": "root", "password": "password"},
    {"hostname": "192.168.1.102", "username": "root", "password": "password"},
    # 添加更多主机信息
]

# 收集本地系统信息
def collect_local_info():
    hostname = socket.gethostname()
    os_info = platform.platform()
    memory_info = psutil.virtual_memory()
    disk_info = psutil.disk_partitions()

    # 获取所有磁盘的总大小
    disk_sizes = []
    for partition in disk_info:
        usage = psutil.disk_usage(partition.mountpoint)
        disk_sizes.append(usage.total)

    return {
        "hostname": hostname,
        "os_info": os_info,
        "memory_size": memory_info.total,
        "disk_sizes": disk_sizes,
    }

# 通过 SSH 收集远程主机信息
def collect_remote_info(hostname, username, password):
    try:
        # 创建 SSH 客户端
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(hostname, username=username, password=password, timeout=10)

        # 获取主机名
        stdin, stdout, stderr = ssh.exec_command("hostname")
        remote_hostname = stdout.read().decode().strip()

        # 获取操作系统信息
        stdin, stdout, stderr = ssh.exec_command("cat /etc/os-release")
        os_info = stdout.read().decode().strip()

        # 获取内存大小
        stdin, stdout, stderr = ssh.exec_command("free -b | grep Mem | awk '{print $2}'")
        memory_size = int(stdout.read().decode().strip())

        # 获取磁盘大小
        stdin, stdout, stderr = ssh.exec_command("lsblk -b -o SIZE | grep -v SIZE")
        disk_sizes = [int(size) for size in stdout.read().decode().strip().split()]

        ssh.close()

        return {
            "hostname": remote_hostname,
            "os_info": os_info,
            "memory_size": memory_size,
            "disk_sizes": disk_sizes,
        }
    except Exception as e:
        print(f"Failed to collect info from {hostname}: {e}")
        return None

# 将信息写入 CSV 文件
def write_to_csv(data, filename="system_info.csv"):
    with open(filename, mode="w", newline="") as file:
        writer = csv.writer(file)
        writer.writerow(["Hostname", "OS Info", "Memory Size (bytes)", "Disk Sizes (bytes)"])

        for item in data:
            writer.writerow([
                item["hostname"],
                item["os_info"],
                item["memory_size"],
                ", ".join(map(str, item["disk_sizes"])),
            ])

# 主函数
def main():
    all_info = []

    # 收集本地信息
    local_info = collect_local_info()
    all_info.append(local_info)

    # 收集远程主机信息
    for host in REMOTE_HOSTS:
        remote_info = collect_remote_info(host["hostname"], host["username"], host["password"])
        if remote_info:
            all_info.append(remote_info)

    # 将信息写入 CSV 文件
    write_to_csv(all_info)
    print("System information has been saved to system_info.csv")

if __name__ == "__main__":
    main()

代码说明

1. collect_local_info：
2. 使用 psutil 和 platform 收集本地主机的信息。
3. 获取主机名、操作系统型号、内存大小和磁盘大小。
4. collect_remote_info：
5. 使用 paramiko 通过 SSH 连接到远程主机。
6. 执行命令获取远程主机的信息（如 hostname、cat /etc/os-release、free -b、lsblk 等）。
7. write_to_csv：
8. 将收集到的信息写入 CSV 文件，每行包含主机名、操作系统型号、内存大小和磁盘大小。
9. REMOTE_HOSTS：
10. 定义远程主机的 IP 地址、用户名和密码。可以根据实际情况扩展。
11. 错误处理：
12. 如果 SSH 连接失败或命令执行失败，会捕获异常并跳过该主机。

输出示例

生成的 system_info.csv 文件内容如下：

Hostname,OS Info,Memory Size (bytes),Disk Sizes (bytes)
localhost,Linux-5.15.0-83-generic-x86_64-with-glibc2.31,8589934592,107374182400, 214748364800
remote-host1,NAME="Ubuntu" VERSION="20.04.3 LTS (Focal Fossa)",8589934592,107374182400, 214748364800
remote-host2,NAME="CentOS" VERSION="7 (Core)",8589934592,107374182400, 214748364800

注意事项

1. SSH 连接安全性：
2. 建议使用 SSH 密钥认证代替密码认证，以提高安全性。
3. 并发执行：
4. 如果需要同时收集多台主机的信息，可以使用多线程或多进程（如 concurrent.futures）提高效率。
5. 磁盘信息：
6. lsblk 命令可能需要根据系统环境调整，确保所有主机都支持该命令。
7. 权限问题：
8. 确保脚本在运行时有足够的权限访问远程主机和执行命令。

通过这个脚本，系统运维工程师可以轻松收集多台主机的系统信息，并将其存储到 CSV 文件中，便于后续分析和处理。