-
Notifications
You must be signed in to change notification settings - Fork 0
/
health_check.py
50 lines (43 loc) · 1.55 KB
/
health_check.py
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
#!/usr/bin/env python3
import socket
import shutil
import psutil
import emails
def check_localhost():
localhost = socket.gethostbyname('localhost')
return localhost== "127.0.0.1"
def check_disk_usage(disk):
"""Verifies that there's enough free space on disk"""
du = shutil.disk_usage(disk)
free = du.free / du.total * 100
return free > 20
def check_memory_usage():
"""Verifies that there's enough free space on disk"""
mu = psutil.virtual_memory().available
total = mu / (1024.0 ** 2)
return total > 500
def check_cpu_usage():
"""Verifies that there's enough unused CPU"""
usage = psutil.cpu_percent(1)
return usage < 80
def send_email(subject):
email = emails.generate_email("[email protected]", "[email protected]",
subject,
"Please check your system and resolve the issue as soon as possible.", "")
emails.send_email(email)
# If there's not enough disk, or not enough CPU, print an error
if not check_cpu_usage() :
subject="Error - CPU usage is over 80%"
print(subject)
send_email(subject)
if not check_memory_usage():
subject = "Error - Available memory is less than 500MB"
print(subject)
if not check_disk_usage('/') :
subject = "Error - Available disk space is less than 20%"
print(subject)
send_email(subject)
if not check_localhost():
subject = "Error - localhost cannot be resolved to 127.0.0.1"
print(subject)
send_email(subject)