Run Chrome and Firefox in headless mode

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -1,5 +1,7 @@
 import argparse
 from selenium import webdriver
+from selenium.webdriver.chrome.options import Options as ChromeOptions
+from selenium.webdriver.firefox.options import Options as FirefoxOptions
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.common.action_chains import ActionChains
@@ -24,10 +26,15 @@ def main():
     # This will start the right driver depending on what
     # driver option is chosen
     if args.driver == "chrome":
-        driver = webdriver.Chrome()
+        options = ChromeOptions()
+        options.add_argument("--headless")
+        options.add_argument("--no-sandbox")
+        driver = webdriver.Chrome(options=options)
 
     elif args.driver == "firefox":
-        driver = webdriver.Firefox()
+        options = FirefoxOptions()
+        options.add_argument("--headless")
+        driver = webdriver.Firefox(options=options)
 
     elif args.driver == "safari":
         driver = webdriver.Safari()