Execute notebooks faster for Chrome and Firefox (to be tested with Safari before porting changes to that browser too)

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -94,14 +94,13 @@ def main():
     )
     driver.execute_script("arguments[0].click();", kernel_button)
     driver.switch_to.active_element.send_keys(Keys.TAB)
-    time.sleep(2)
+    time.sleep(1)
     actions.send_keys(f"{args.kernel}").perform()
-    time.sleep(2)
+    time.sleep(1)
     actions.send_keys(Keys.TAB).perform()
-    time.sleep(2)
+    time.sleep(1)
     actions.send_keys(Keys.ENTER).perform()
-    time.sleep(2)
-    time.sleep(0.5)
+    time.sleep(1)
 
     # This will run all the cells of the chosen notebook
     if args.driver == "chrome" or args.driver == "firefox":
@@ -110,7 +109,6 @@ def main():
             focused_cell = driver.find_element(
                 By.CSS_SELECTOR, ".jp-Notebook-cell.jp-mod-selected"
             )
-            print(focused_cell.text)
             editor_divs = focused_cell.find_elements(
                 By.CSS_SELECTOR, ".jp-InputArea-editor div"
             )
@@ -132,7 +130,7 @@ def main():
                 )
                 input_box.click()
                 input_box.send_keys(f"{args.stdin}")
-                time.sleep(10)
+                time.sleep(1)
                 input_box.send_keys(Keys.CONTROL, Keys.ENTER)
                 next_cell = focused_cell.find_element(
                     By.XPATH,
@@ -141,14 +139,32 @@ def main():
                 driver.execute_script(
                     "arguments[0].scrollIntoView({block:'center'});", next_cell
                 )
-                print(next_cell.text)
                 next_cell.click()
+                while True:
+                    spans = driver.find_elements(By.CSS_SELECTOR, "span.jp-StatusBar-TextItem")
+                    status_span = spans[2]
+                    text = status_span.text
+
+                    if "Idle" in text:
+                        break
+                    time.sleep(0.01)
+                print(focused_cell.text)
+                focused_cell=next_cell
             run_menu = wait.until(
                 EC.element_to_be_clickable((By.XPATH, "//li[normalize-space()='Run']"))
             )
-            actions.move_to_element(run_menu).pause(0.1).click().perform()
-            actions.send_keys(Keys.DOWN).send_keys(Keys.ENTER).perform()
-            time.sleep(1.0)
+            actions.move_to_element(run_menu).pause(0.05).click().perform()
+            actions.send_keys(Keys.DOWN).send_keys(Keys.ENTER).pause(0.1).perform()
+            if not cell_is_waiting_for_input(driver):
+                while True:
+                    spans = driver.find_elements(By.CSS_SELECTOR, "span.jp-StatusBar-TextItem")
+                    status_span = spans[2]
+                    text = status_span.text
+
+                    if "Idle" in text:
+                        print(focused_cell.text)
+                        break
+                    time.sleep(0.01)
 
     elif args.driver == "safari":
         print("Running all cells using Shift+Enter...")