Chrome驱动程序可以解决GeckoDriver的问题

Question

我正在编写一个脚本来下载我正在处理的项目的文件。我的公司在Chrome上阻止了该网站，但没有使用FireFox，因此我决定尝试使用Geckodriver，但仍然遇到此问题：

AttributeError: module 'selenium.webdriver' has no attribute 'FireFox'

这是我的代码：

import selenium.webdriver as webdriver
from selenium.webdriver import firefox
import os

def init_driver():
    driver = 
webdriver.FireFox(r"C:\Users\Michael.Spencer\Documents\geckodriver.exe")
    driver.wait = webdriver.support.ui.WebDriverWait(driver, 5)
    return driver

def download(driver):

    driver.get("http://www.google.com")

def main():
    driver = init_driver()
    download(driver)
    driver.close()

main()

如果我使用

，这样可以正常工作

webdrive.Chrome(r"C:\Users\Michael.Spencer\Documents\chromedriver.exe")

当我开始输入时，Firefox甚至会弹出，所以此时我不知道如何继续。

Answer 1

试试这个，我可能错了，但它可能是你导入的方式

from selenium import webdriver
driver = webdriver.Firefox()