ResourceScripts/disks.py at master · Nemezor/ResourceScripts · GitHub

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
import psutil
import subprocess
import sys

partitions = {}

def translate_filesystem_type(fs):
    if fs == "fuseblk":
        return "ntfs"
    elif fs == "hfsplus":
        return "hfs+"
    return fs

data = ""

for partition in psutil.disk_partitions(all=True):
    if (partition.device.startswith("/dev")):
        partitions[partition.device[5:]] = partition

data += "\t\"disks\": {\n"
counters = psutil.disk_io_counters(perdisk=True)
c = 0
for counter in counters:
    io = counters[counter]
    data += "\t\t\"" + counter + "\": {\n"
    data += "\t\t\t\"read_bytes\": \"" + str(io.read_bytes) + "\",\n"
    data += "\t\t\t\"write_bytes\": \"" + str(io.write_bytes) + "\",\n"
    if counter in partitions:
        data += "\t\t\t\"partition\": \"true\",\n"
        disk = partitions[counter]
        usage = psutil.disk_usage(disk.mountpoint)
        data += "\t\t\t\"mount\": \"" + disk.mountpoint + "\",\n"
        data += "\t\t\t\"filesystem\": \"" + translate_filesystem_type(disk.fstype) + "\",\n"
        data += "\t\t\t\"space_total\": \"" + str(usage.total) + "\",\n"
        data += "\t\t\t\"space_used\": \"" + str(usage.used) + "\",\n"
        data += "\t\t\t\"space_free\": \"" + str(usage.free) + "\",\n"
        data += "\t\t\t\"space_percentage\": \"" + str(usage.percent) + "\"\n"
    else:
        data += "\t\t\t\"partition\": \"false\"\n"
    c += 1
    if c == len(counters):
        data += "\t\t}\n"
    else:
        data += "\t\t},\n"
data += "\t}\n"

print(data)