Python

概述

因为工作需要，使用python3写了一个检查磁盘监控状态的脚本，支持有阵列以及nvme的磁盘检查。

安装工具

使用之前需要确保你的系统环境下已安装smartctl工具，如果没有则需要安装

1
2
3
apt -y install smartmontools
# or
yum -y install smartmontools

代码示例

脚本主要通过SMART进行磁盘状态的检查.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import subprocess
import re

def run_cmd(cmd):
    try:
        result = subprocess.run(cmd, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
        return result.stdout
    except subprocess.CalledProcessError as e:
        return e.output

def check_nvme(device):
    output = run_cmd(f"smartctl -H {device}")
    if "PASSED" in output:
        return "✅ 健康"
    elif "FAILED" in output:
        return "❌ 故障"
    else:
        return "❔ 无法判断"

def check_megaraid(device):
    for i in range(16):  # 最多尝试 megaraid 0~15
        output = run_cmd(f"smartctl -d megaraid,{i} -H {device}")
        if "PASSED" in output:
            return f"✅ 健康 (megaraid,{i})"
        elif "FAILED" in output:
            return f"❌ 故障 (megaraid,{i})"
        elif "Invalid" not in output and "SCSI Status" not in output:
            # 输出不同于完全无效的 megaraid 索引，说明可能是目标磁盘
            continue
    return "❔ 无法判断 (megaraid)"

def check_sdx(device):
    output = run_cmd(f"smartctl -H {device}")
    if "PASSED" in output:
        return "✅ 健康"
    elif "FAILED" in output:
        return "❌ 故障"
    elif "please try adding '-d megaraid" in output:
        return check_megaraid(device)
    else:
        return "❔ 无法判断"

def list_block_devices():
    lsblk_out = run_cmd("lsblk -dn -o NAME,TYPE")
    devices = []
    for line in lsblk_out.strip().splitlines():
        name, dtype = line.strip().split()
        if dtype == 'disk':
            if name.startswith("nvme"):
                devices.append(f"/dev/{name}")
            elif name.startswith("sd"):
                devices.append(f"/dev/{name}")
    return devices

def main():
    print("=== SMART 健康检测（自动适配） ===")
    for dev in list_block_devices():
        if dev.startswith("/dev/nvme"):
            status = check_nvme(dev)
        else:
            status = check_sdx(dev)
        print(f"{dev}: {status}")

if __name__ == "__main__":
    main()

执行效果