fix: summary time travel

Change-Id: I68e489f7d35b47aaadcbb842d9213e66147bd766
redhat-performance · May 26, 2020 · cbcda5f · cbcda5f
1 parent d5d047b
commit cbcda5f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/bin/quads-cli b/bin/quads-cli
@@ -764,7 +764,7 @@ def main(_args):
                 _args.datearg, "%Y-%m-%d %H:%M"
             ).isoformat()
         try:
-            summary = quads.get_summary()
+            summary = quads.get_summary(**_kwargs)
         except ConnectionError:
             logger.error(
                 "Could not connect to the quads-server, verify service is up and running."

diff --git a/quads/api_v2.py b/quads/api_v2.py
@@ -153,11 +153,16 @@ def GET(self, **data):
         if self.name == "summary":
             _clouds = model.Cloud.objects().all()
             clouds_summary = []
+            total_count = 0
             for cloud in _clouds:
                 if cloud.name == "cloud01":
                     count = model.Host.objects(cloud=cloud).count()
                 else:
-                    count = self.model.current_schedule(cloud=cloud).count()
+                    date = datetime.datetime.now()
+                    if "date" in data:
+                        date = datetime.datetime.strptime(data["date"], "%Y-%m-%dT%H:%M:%S")
+                    count = self.model.current_schedule(cloud=cloud, date=date).count()
+                    total_count += count
                 clouds_summary.append(
                     {
                         "name": cloud.name,
@@ -170,6 +175,11 @@ def GET(self, **data):
                         "validated": cloud.validated,
                     }
                 )
+            if "date" in data:
+                host_count = model.Host.objects().count()
+                for cloud in clouds_summary:
+                    if cloud["name"] == "cloud01":
+                        cloud["count"] = host_count - total_count
 
             return json.dumps(clouds_summary)