Implements: #1353

Author: micafer

IM/REST.py

@@ -19,6 +19,7 @@
 import json
 import base64
 import bottle
+import datetime
 
 from IM.InfrastructureInfo import IncorrectVMException, DeletedVMException, IncorrectStateException
 from IM.InfrastructureManager import (InfrastructureManager, DeletedInfrastructureException,
@@ -1049,14 +1050,42 @@ def ReturnOptions(**kwargs):
 
 
 @app.route('/version', method='GET')
-def RESTGeVersion():
+def RESTGetVersion():
     try:
         from IM import __version__ as version
         return format_output(version, field_name="version")
     except Exception as ex:
         return return_error(400, "Error getting IM version: %s" % get_ex_error(ex))
 
 
+@app.route('/stats', method='GET')
+def RESTGetStats():
+    try:
+        auth = get_auth_header()
+    except Exception:
+        return return_error(401, "No authentication data provided")
+
+    try:
+        if "init_date" in bottle.request.params.keys():
+            init_date = bottle.request.params.get("init_date").lower()
+            init_date = init_date.replace("/", "-")
+            parts = init_date.split("-")
+            try:
+                year = int(parts[0])
+                month = int(parts[1])
+                day = int(parts[2])
+                datetime.date(year, month, day)
+            except Exception:
+                return return_error(400, "Incorrect format in init_date parameter: YYYY/MM/dd")
+        else:
+            init_date = "1970-01-01"
+
+        return InfrastructureManager.GetStats(init_date, auth)
+    except Exception as ex:
+        logger.exception("Error getting stats")
+        return return_error(400, "Error getting stats: %s" % get_ex_error(ex))
+
+
 @app.route('/infrastructures/:infid/vms/:vmid/disks/:disknum/snapshot', method='PUT')
 def RESTCreateDiskSnapshot(infid=None, vmid=None, disknum=None):
     try:

IM/Stats.py

@@ -87,7 +87,18 @@ def get_stats(init_date="1970-01-01", auth=None):
         - init_date(str): Only will be returned infrastructure created afther this date.
         - auth(Authentication): parsed authentication tokens.
 
-        Return: a list of dict with the stats.
+        Return: a list of dict with the stats with the following format:
+            {'creation_date': '2022-03-07 13:16:14',
+             'icon': 'kubernetes',
+             'vm_count': 2,
+             'cpu_count': 4,
+             'memory_size': 1024,
+             'cloud_type': 'OSCAR',
+             'cloud_host': 'sharp-elbakyan5.im.grycap.net',
+             'hybrid': False,
+             'im_user': '__OPENID__mcaballer',
+             'inf_id': '1',
+             'last_date': '2022-03-23'}
         """
         stats = []
         db = DataBase(Config.DATA_DB)

test/unit/REST.py

@@ -53,11 +53,12 @@
                      RESTStartVM,
                      RESTStopVM,
                      RESTRebootVM,
-                     RESTGeVersion,
+                     RESTGetVersion,
                      RESTCreateDiskSnapshot,
                      RESTImportInfrastructure,
                      RESTGetCloudInfo,
                      RESTChangeInfrastructureAuth,
+                     RESTGetStats,
                      return_error,
                      format_output)
 
@@ -1074,6 +1075,21 @@ def test_ChangeInfrastructureAuth(self, ChangeInfrastructureAuth, bottle_request
                                                                                        "password": "new_pass"}])
         self.assertEqual(ChangeInfrastructureAuth.call_args_list[0][0][2], True)
 
+    @patch("bottle.request")
+    @patch("IM.InfrastructureManager.InfrastructureManager.GetStats")
+    def test_GetStats(self, GetStats, bottle_request):
+        """Test REST GetStats."""
+        bottle_request.return_value = MagicMock()
+        bottle_request.headers = {"AUTHORIZATION": "type = InfrastructureManager; username = user; password = pass"}
+        GetStats.return_value = [{"stats"}]
+        bottle_request.params = {'init_date': '2010-01-01'}
+        res = RESTGetStats()
+        self.assertEqual(res, [{"stats"}])
+        self.assertEqual(GetStats.call_args_list[0][0][0], '2010-01-01')
+        self.assertEqual(GetStats.call_args_list[0][0][1].auth_list, [{"type": "InfrastructureManager",
+                                                                       "username": "user",
+                                                                       "password": "pass"}])
+
 
 if __name__ == "__main__":
     unittest.main()