Try new way to run cells chrome and firefox

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -60,8 +60,8 @@ def main():
     # driver option is chosen
     if args.driver == "chrome":
         options = ChromeOptions()
-        options.add_argument("--headless")
-        options.add_argument("--no-sandbox")
+        #options.add_argument("--headless")
+        #options.add_argument("--no-sandbox")
         driver = webdriver.Chrome(options=options)
 
     elif args.driver == "firefox":
@@ -106,12 +106,6 @@ def main():
     if args.driver == "chrome" or args.driver == "firefox":
         print("Running Cells")
         while True:
-            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()
-
             focused_cell = driver.find_element(
                 By.CSS_SELECTOR, ".jp-Notebook-cell.jp-mod-selected"
             )
@@ -130,27 +124,29 @@ def main():
 
             if cell_is_waiting_for_input(driver):
                 print("Cell requesting input")
-                actions.send_keys(f"{args.stdin}").perform()
-                time.sleep(1.15)
-                actions.send_keys(Keys.CONTROL, Keys.ENTER)
-                time.sleep(1.15)
-                actions.move_to_element(focused_cell).perform()
-                time.sleep(1.15)
+                input_box = WebDriverWait(driver, 5).until(
+                    EC.visibility_of_element_located(
+                        (By.CSS_SELECTOR, ".jp-Stdin-input")
+                    )
+                )
+                input_box.click()
+                input_box.send_keys(f"{args.stdin}")
+                time.sleep(10)
+                input_box.send_keys(Keys.CONTROL, Keys.ENTER)
+                next_cell = focused_cell.find_element(
+                    By.XPATH,
+                    "following-sibling::div[contains(@class,'jp-Notebook-cell')][1]",
+                )
                 driver.execute_script(
-                    """
-                var el = document.activeElement;
-                var evt = new KeyboardEvent('keydown', {
-                key: 'ArrowDown',
-                code: 'ArrowDown',
-                keyCode: 40,
-                which: 40,
-                bubbles: true
-                });
-                el.dispatchEvent(evt);
-                """
+                    "arguments[0].scrollIntoView({block:'center'});", next_cell
                 )
-                time.sleep(1.15)
-            time.sleep(1.15)
+                next_cell.click()
+            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)
 
     elif args.driver == "safari":
         print("Running all cells using Shift+Enter...")