Wait for panels state=Done

grafana-toolbox · Oct 15, 2023 · e6bc987 · e6bc987
1 parent 6350326
commit e6bc987
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/grafanimate/marionette.py b/grafanimate/marionette.py
@@ -4,6 +4,7 @@
 import atexit
 import json
 import logging
+import time
 from collections import OrderedDict
 
 import where
@@ -170,6 +171,16 @@ def render_image(self, element=None):
         """
         Return screenshot from element.
         """
+        while True:
+            result = self.marionette.execute_script('''
+            panels = Object.values(window.wrappedJSObject.grafanaRuntime.getPanelData())
+            return panels.length && panels.every(function(o) {return o?.state=='Done'})
+            ''')
+            if result:
+                break
+            logger.info("waiting for panels to load")
+            time.sleep(0.1)
+
         image = self.marionette.screenshot(element=element, format="binary")
         return image