Add total RAM, CPUs and disk size to servers summary as well as disk usage in percent. Fixes https://github.com/GNS3/gns3-server/issues/1532

gns3server/controller/compute.py

@@ -80,9 +80,13 @@ class Compute:
         self._set_auth(user, password)
         self._cpu_usage_percent = None
         self._memory_usage_percent = None
+        self._disk_usage_percent = None
         self._last_error = None
         self._capabilities = {
             "version": None,
+            "cpus": None,
+            "memory": None,
+            "disk_size": None,
             "node_types": []
         }
         self.name = name
@@ -270,6 +274,10 @@ class Compute:
     def memory_usage_percent(self):
         return self._memory_usage_percent
 
+    @property
+    def disk_usage_percent(self):
+        return self._disk_usage_percent
+
     def __json__(self, topology_dump=False):
         """
         :param topology_dump: Filter to keep only properties require for saving on disk
@@ -292,6 +300,7 @@ class Compute:
             "connected": self._connected,
             "cpu_usage_percent": self._cpu_usage_percent,
             "memory_usage_percent": self._memory_usage_percent,
+            "disk_usage_percent": self._disk_usage_percent,
             "capabilities": self._capabilities,
             "last_error": self._last_error
         }
@@ -437,6 +446,7 @@ class Compute:
                         if action == "ping":
                             self._cpu_usage_percent = event["cpu_usage_percent"]
                             self._memory_usage_percent = event["memory_usage_percent"]
+                            self._disk_usage_percent = event["disk_usage_percent"]
                             #FIXME: slow down number of compute events
                             self._controller.notification.controller_emit("compute.updated", self.__json__())
                         else:
@@ -461,6 +471,7 @@ class Compute:
 
         self._cpu_usage_percent = None
         self._memory_usage_percent = None
+        self._disk_usage_percent = None
         self._controller.notification.controller_emit("compute.updated", self.__json__())
 
     def _getUrl(self, path):

gns3server/handlers/api/compute/capabilities_handler.py

@@ -16,11 +16,13 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 import sys
+import psutil
 
 from gns3server.web.route import Route
 from gns3server.schemas.capabilities import CAPABILITIES_SCHEMA
 from gns3server.version import __version__
 from gns3server.compute import MODULES
+from gns3server.utils.path import get_default_project_directory
 
 
 class CapabilitiesHandler:
@@ -38,5 +40,8 @@ class CapabilitiesHandler:
         response.json({
             "version": __version__,
             "platform": sys.platform,
+            "cpus": psutil.cpu_count(logical=True),
+            "memory": psutil.virtual_memory().total,
+            "disk_size": psutil.disk_usage(get_default_project_directory()).total,
             "node_types": node_types
         })

gns3server/handlers/api/compute/project_handler.py

@@ -20,12 +20,12 @@ import asyncio
 import json
 import os
 import psutil
-import tempfile
 
 from gns3server.web.route import Route
 from gns3server.compute.project_manager import ProjectManager
 from gns3server.compute import MODULES
 from gns3server.utils.cpu_percent import CpuPercent
+from gns3server.utils.path import get_default_project_directory
 
 from gns3server.schemas.project import (
     PROJECT_OBJECT_SCHEMA,
@@ -211,6 +211,7 @@ class ProjectHandler:
         # Non blocking call in order to get cpu usage. First call will return 0
         stats["cpu_usage_percent"] = CpuPercent.get(interval=None)
         stats["memory_usage_percent"] = psutil.virtual_memory().percent
+        stats["disk_usage_percent"] = psutil.disk_usage(get_default_project_directory()).percent
         return {"action": "ping", "event": stats}
 
     @Route.get(

gns3server/handlers/api/compute/server_handler.py

@@ -25,6 +25,7 @@ from gns3server.schemas.server_statistics import SERVER_STATISTICS_SCHEMA
 from gns3server.compute.port_manager import PortManager
 from gns3server.utils.cpu_percent import CpuPercent
 from gns3server.version import __version__
+from gns3server.utils.path import get_default_project_directory
 from aiohttp.web import HTTPConflict
 
 
@@ -61,7 +62,7 @@ class ServerHandler:
             load_average_percent = [int(x / psutil.cpu_count() * 100) for x in psutil.getloadavg()]
             memory_percent = int(psutil.virtual_memory().percent)
             swap_percent = int(psutil.swap_memory().percent)
-            disk_usage_percent = int(psutil.disk_usage('/').percent)
+            disk_usage_percent = int(psutil.disk_usage(get_default_project_directory()).percent)
         except psutil.Error as e:
             raise HTTPConflict(text="Psutil error detected: {}".format(e))
         response.json({"memory_total": memory_total,

gns3server/notification_queue.py

@@ -20,6 +20,7 @@ import json
 import psutil
 
 from gns3server.utils.cpu_percent import CpuPercent
+from gns3server.utils.path import get_default_project_directory
 
 
 class NotificationQueue(asyncio.Queue):
@@ -55,6 +56,7 @@ class NotificationQueue(asyncio.Queue):
         # Non blocking call in order to get cpu usage. First call will return 0
         msg["cpu_usage_percent"] = CpuPercent.get(interval=None)
         msg["memory_usage_percent"] = psutil.virtual_memory().percent
+        msg["disk_usage_percent"] = psutil.disk_usage(get_default_project_directory()).percent
         return msg
 
     async def get_json(self, timeout):

gns3server/schemas/capabilities.py

@@ -36,7 +36,19 @@ CAPABILITIES_SCHEMA = {
         "platform": {
             "type": "string",
             "description": "Platform where the compute is running"
-        }
+        },
+        "cpus": {
+            "description": "Number of CPUs on this compute. Read only",
+            "type": ["integer", "null"],
+        },
+        "memory": {
+            "description": "Amount of memory on this compute. Read only",
+            "type": ["integer", "null"],
+        },
+        "disk_size": {
+            "description": "Disk size on this compute. Read only",
+            "type": ["integer", "null"],
+        },
     },
     "additionalProperties": False
 }

gns3server/schemas/compute.py

@@ -104,6 +104,12 @@ COMPUTE_OBJECT_SCHEMA = {
             "maximum": 100,
             "minimum": 0
         },
+        "disk_usage_percent": {
+            "description": "Disk usage of the compute. Read only",
+            "type": ["number", "null"],
+            "maximum": 100,
+            "minimum": 0
+        },
         "last_error": {
             "description": "Last error on the compute",
             "type": ["string", "null"]