added disk space and cpu_usuage deets

Author: Sbasu2512

service/helper.py

@@ -1,3 +1,19 @@
+import os
+import time
+
+
+def kb_to_mb(kb_value):
+    return kb_value / 1024
+
+
+def bytes_to_mb(value):
+    return value / (1024**2)
+
+
+def mb_to_gb(mb_value):
+    return mb_value / 1024
+
+
 def get_memory_info():
     mem_info = {}
 
@@ -11,3 +27,59 @@ def get_memory_info():
             mem_info[key] = value
 
     return mem_info
+
+
+def get_disk_usage(path="/"):
+    # Get the file system statistics for the specified path
+    statvfs = os.statvfs(path)
+
+    # Calculate the disk space in bytes
+    total_space = statvfs.f_frsize * statvfs.f_blocks  # Total space
+    free_space = statvfs.f_frsize * statvfs.f_bfree  # Free space
+    # Space available to non-superuser
+    available_space = statvfs.f_frsize * statvfs.f_bavail
+    # Used space
+    used_space = total_space - free_space
+
+    # Convert bytes to MB
+    total_space_mb = bytes_to_mb(total_space)
+    free_space_mb = bytes_to_mb(free_space)
+    used_space_mb = bytes_to_mb(used_space)
+
+    return {
+        "total_space_mb": total_space_mb,
+        "used_space_mb": used_space_mb,
+        "free_space_mb": free_space_mb,
+    }
+
+
+def get_cpu_usage():
+    # Read the initial CPU stats
+    with open("/proc/stat", "r") as f:
+        first_line = f.readline()
+
+    # Extract CPU values from the first line
+    cpu_times = list(
+        map(int, first_line.split()[1:])
+    )  # Skip the first element (CPU label)
+    total_time1 = sum(cpu_times)  # Total time spent in all states
+
+    # Wait for a short period (1 second)
+    time.sleep(1)
+
+    # Read the CPU stats again
+    with open("/proc/stat", "r") as f:
+        second_line = f.readline()
+
+    # Extract CPU values from the second line
+    cpu_times = list(map(int, second_line.split()[1:]))
+    total_time2 = sum(cpu_times)
+
+    # Calculate the differences
+    total_diff = total_time2 - total_time1
+    idle_diff = cpu_times[3] - cpu_times[3]  # Idle time is usually the fourth value
+
+    # Calculate CPU usage percentage
+    cpu_usage = 100 * (1 - (idle_diff / total_diff))
+
+    return cpu_usage

service/views.py

@@ -19,7 +19,7 @@
 from io import StringIO
 from datetime import datetime
 from collections import defaultdict
-from .helper import get_memory_info
+from .helper import get_memory_info, kb_to_mb, get_disk_usage, get_cpu_usage
 
 APP_START_TIME = datetime.now()
 
@@ -38,18 +38,36 @@ def health_check(request, *args, **kwargs):
 
         memory_info = get_memory_info()
 
-        # Extract specific memory details (if available)
-        total_memory = memory_info.get("MemTotal")
-        free_memory = memory_info.get("MemFree")
-        available_memory = memory_info.get("MemAvailable")
-        cached_memory = memory_info.get("Cached")
-        buffered_memory = memory_info.get("Buffers")
-
-        print(f"Total Memory: {total_memory}")
-        print(f"Free Memory: {free_memory}")
-        print(f"Available Memory: {available_memory}")
-        print(f"Cached Memory: {cached_memory}")
-        print(f"Buffered Memory: {buffered_memory}")
+        # Extract specific memory details
+        total_memory = kb_to_mb(memory_info.get("MemTotal"))
+        free_memory = kb_to_mb(memory_info.get("MemFree"))
+        available_memory = kb_to_mb(memory_info.get("MemAvailable"))
+
+        used_memory = total_memory - available_memory
+
+        used_memory_percentage = (used_memory / total_memory) * 100
+
+        data["free_memory"] = f"{free_memory:.2f} MB"
+        data["total_memory"] = f"{total_memory:.2f} MB"
+        data["memory_usage"] = f"{used_memory_percentage:.2f} %"
+
+        disk_usage = get_disk_usage("/")
+
+        total_disk_space_gb = disk_usage["total_space_mb"] / 1024
+        disk_usage_percentage = (
+            disk_usage["used_space_mb"] / disk_usage["total_space_mb"]
+        ) / 100
+
+        data["total_space_mb"] = f"{disk_usage['total_space_mb']:.2f} MB"
+        data["used_space_mb"] = f"{disk_usage['used_space_mb']:.2f} MB"
+        data["free_space_mb"] = f"{disk_usage['free_space_mb']:.2f} MB"
+        data["disk_usage_percentage"] = (
+            f"{disk_usage_percentage:.2f} % of {total_disk_space_gb:.2f}"
+        )
+
+        cpu_usage = get_cpu_usage()
+
+        data["cpu_usuage"] = f"{cpu_usage:.2f}%"
 
         # Check app status
         data["status"] = "OK"