updated get_ram_and_disk method

2024-06-28 18:12:11 +05:30 · 2024-06-28 18:12:11 +05:30 · ea4675a56c
commit ea4675a56c
parent bd6ecc3519
1 changed files with 8 additions and 7 deletions
--- a/agent.py
+++ b/agent.py
@ -145,13 +145,14 @@ class ServerData:
        for device in os.listdir('/sys/block'):
            device_path = '/sys/block/{}/device'.format(device)
            size_path = '/sys/block/{}/size'.format(device)
-            if os.path.islink(device_path):
+            if not os.path.islink(device_path):
+                continue
            try:
                with open(size_path, 'r') as f:
                    size = int(f.read().strip())
                    disk += size
            except Exception:
-                    pass  # Skip the device if any exception occurs
+                pass

        disk = disk * 512 // (1024**3)  # convert to GB
        logging.info("RAM: {}MB, Disk: {}GB".format(ram, disk))