Selenium Snippets

selenium_snippets.py

# to make a headless browser
geckodriver_autoinstaller.install()
options = Options()
options.headless = True
driver = webdriver.Firefox(options=options)



def setup_driver(headless: bool):
    """
    Only download the driver if not found on system as geckodriver_autoinstaller.install() will naively download
    """
    gecko_path = shutil.which("geckodriver") or shutil.which("geckodriver.exe")
    if not gecko_path:
        try:
            gecko_path = glob.glob('*/*geckodriver')[0]
        except IndexError:
            print('Installing geckodriver...')
            gecko_path = geckodriver_autoinstaller.install(True)
            print('Installed geckodriver to', gecko_path)
    options = Options()
    options.headless = headless
    driver = webdriver.Firefox(options=options, executable_path=gecko_path)
    return driver