CPU, memory and disk usage monitoring methods added

Author: jeremymaya

README.md

@@ -13,7 +13,21 @@ Proactive monitoring of the system is crucial for ensuring uptime. Critial syste
 
 ### Core Temperature
 
-Core temperature of the Raspberry Pi is monitored using ```check_temperature``` method. If the core temperature rises above 80.0'C, the method sends an alsert email.
+Core temperature is monitored using ```check_temperature``` method. If the core temperature rises above 80.0'C, the method sends an alert email.
+
+### CPU Usage
+
+CPU usage is monitored using ```check_cpu_usage``` method. If the CPU usage is over 80%, the method sends an alert email.
+
+### Memory Usage
+
+Memory usage is monitored using ```check_memoery_usage``` method. If the available memory is less than 500MB, the method sends an alert email.
+
+### Disk Usage
+
+Disk usage is monitored using ```check_disk_usage``` method. If the available memory is less than 20%, the method sends an alert email.
+
+---
 
 ## Credits
 

system_health_check.py

@@ -2,6 +2,8 @@
 
 import sys
 import os
+import shutil
+import psutil
 import emails
 import re
 from decouple import config
@@ -14,6 +16,20 @@ def check_temperature():
     temp = re.findall(r'\d*\.?\d+', measure_temp)
     return float(temp[0]) < 80.0
 
+def check_cpu_usage():
+    cpu_usage = psutil.cpu_percent(1)
+    return cpu_usage < 80
+
+def check_memory_usage():
+    memoery_usage = psutil.virtual_memory().available
+    used = memoery_usage / (1024.0 ** 2)
+    return used > 500
+
+def check_disk_usage(disk):
+    disk_usage = shutil.disk_usage(disk)
+    free = disk_usage.free / disk_usage.total * 100
+    return free > 20
+
 def send_alert(subject):
     sender = SENDER
     recipient = RECIPIENT
@@ -22,8 +38,18 @@ def send_alert(subject):
     emails.send_email(message)
 
 def main(argv):
-    subject = "Alert - The core temperature is over 80.0'C"
-    send_alert(subject)
+    if not check_temperature():
+        subject = "Alert - Core temperature is over 80.0'C"
+        send_alert(subject)
+    if not check_cpu_usage() :
+        subject="Alert - CPU usage is over 80%"
+        send_alert(subject)
+    if not check_memory_usage():
+        subject = "Alert - Available memory is less than 500MB"
+        send_alert(subject)
+    if not check_disk_usage('/') :
+        subject = "Alert - Available disk space is less than 20%"
+        send_alert(subject)
 
 if __name__ == "__main__":
     main(sys.argv)