Update automated-notebook-run-script.py

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -109,7 +109,7 @@ def main():
             notebook_area.send_keys(Keys.SHIFT, Keys.ENTER)
             time.sleep(0.5)
 
-        time.sleep(5)
+        time.sleep(600)
 
     if args.driver == "chrome" or args.driver == "firefox":
         print("Saving notebook")
@@ -168,9 +168,13 @@ def main():
     return deepQuerySelector(document, "jp-button[data-command='docmanager:download']");
     """
 
-    download_button = driver.execute_script(search_script)
+    download_button = WebDriverWait(driver, 20).until(
+    lambda d: d.execute_script(search_script)
+    )
+
+    print("Found element:", download_button)
 
-    time.sleep(2)
+    time.sleep(20)
     driver.execute_script(
         """
         const el = arguments[0];
@@ -190,46 +194,8 @@ def main():
         """,
         download_button,
     )
-    applescript = r'''
-set buttonInfo to {}
-
-tell application "System Events"
-    repeat with proc in (every process whose background only is false)
-        set procName to name of proc
-        try
-            repeat with w in windows of proc
-                repeat with b in (buttons of w)
-                    set bName to ""
-                    try
-                        set bName to name of b
-                    end try
-                    if bName is missing value or bName = "" then
-                        try
-                            set bName to description of b
-                        end try
-                    end if
-                    if bName is missing value or bName = "" then
-                        try
-                            set bName to role description of b
-                        end try
-                    end if
-                    if bName is missing value or bName = "" then
-                        set bName to "«no name»"
-                    end if
-
-                    set end of buttonInfo to procName & " — " & bName
-                end repeat
-            end repeat
-        end try
-    end repeat
-end tell
-
-return buttonInfo
-'''
-
-    result = subprocess.run(["osascript", "-e", applescript], capture_output=True, text=True)
-    print(result.stdout)
-    time.sleep(2)
+
+    time.sleep(20)
 
     # Close browser
     driver.quit()