DiskSpaceChecking/FindingDiskSpace.py at main · nash1983/DiskSpaceChecking · 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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import shutil
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

# this code is to check server diskspace and send coloured email to monitoring team. If disk space is lessthan 5% it triggers red colour mail

def colored_mail():
    msg = MIMEMultipart('alternative')
    msg['Subject'] = f"{compName} Disk Usage Statistics \n\n"
#     Add from address & To address
    msg['From'] = "fromAddress"
    msg['To'] = "toAddress"

    text = (f"Warning: <br>\n"
            f" ***********************************************  <br>\n"
            f"{path} - Less than 5% disk space is remaining  <br> \n"
            f"*********************************************** <br>\n"
            f"Total Disk size is: {disk_total1} GB <br>\n"
            f"Total Free space is: {disk_free1} GB <br>\n"
            f"Total Used space is: {disk_used1} GB <br>\n")

    html = """\
    <html>
      <head></head>
      <body>
        <p style="color: red;"> {text}
        </p>

      </body>
    </html>
    """.format(**locals())

    part1 = MIMEText(text, 'plain')
    part2 = MIMEText(html, 'html')

    msg.attach(part1)  # text must be the first one
    msg.attach(part2)  # html must be the last one

    s = smtplib.SMTP('relay*****', 25)
    s.sendmail(msg['From'], msg['To'], msg.as_string())
    s.quit()


def sending_mail():
    with smtplib.SMTP('relay****', 25) as smtp:
        body2 = (
            f"*********************************************** \n"
            f"{path} - Less than 10% disk space is remaining  \n "
            f"*********************************************** \n"
            f"Total Disk size is: {disk_total1} GB \n"
            f"Total Free space is: {disk_free1} GB \n"
            f"Total Used space is: {disk_used1} GB\n")
#         Add from address and to address
        smtp.sendmail(from_addr='fromAddress',
                      to_addrs='toAddress',
                      msg=f"Subject:{compName} Disk Usage Statistics \n\n {body2}")

# add server names
workstations = ['ComputerList']

for compName in workstations:
    path1 = '\\\\' + compName + '\\C$'
    path2 = "\\\\" + compName + "\\E$"
    path3 = "\\\\" + compName + "\\F$"
    path4 = "\\\\" + compName + "\\G$"
    path5 = "\\\\" + compName + "\\H$"
    path6 = "\\\\" + compName + "\\I$"
    path7 = "\\\\" + compName + "\\J$"
    path8 = "\\\\" + compName + "\\K$"
    path9 = "\\\\" + compName + "\\L$"
    path10 = "\\\\" + compName + "\\M$"
    path11 = "\\\\" + compName + "\\N$"
    path12 = "\\\\" + compName + "\\O$"
    path13 = "\\\\" + compName + "\\P$"
    path14 = "\\\\" + compName + "\\R$"
    path15 = "\\\\" + compName + "\\T$"
    path16 = "\\\\" + compName + "\\U$"
    path17 = "\\\\" + compName + "\\V$"
    path18 = "\\\\" + compName + "\\W$"
    path19 = "\\\\" + compName + "\\Y$"
    path20 = "\\\\" + compName + "\\Z$"

    paths = [path1, path2, path3, path4, path5, path5, path6, path7, path8, path9, path10, path11, path12, path13,
             path14, path15, path16, path17, path18, path19, path20]

    for path in paths:
        try:
            check_drive = shutil.disk_usage(path)
            disk_free = float(check_drive.free / 2 ** 30)
            disk_total = float(check_drive.total / 2 ** 30)
            disk_used = float(check_drive.used / 2 ** 30)
            disk_free1 = float("{0:.2f}".format(disk_free))
            disk_total1 = float("{0:.2f}".format(disk_total))
            disk_used1 = float("{0:.2f}".format(disk_used))
            percent = lambda part, whole: float(whole) / 100 * float(part)
            percent1 = percent(5, disk_total)
            percent2 = percent(10, disk_total)
            if disk_free < percent1:
                colored_mail()
            elif disk_free < percent2:
                sending_mail()
        except:
            pass