python 小白求助一个简单的问题

我有下面2段重复的代码。想把他写成一个function。

1. driver = webdriver.Chrome()
   
2. driver.get('http://www.baidu.com')
   
3. time.sleep(2)
   
4. driver.quit()
   
5. 
   
6. driver2 = webdriver.Firefox()
   
7. driver2.get('http://www.baidu.com')
   
8. time.sleep(2)
   
9. driver2.quit()

复制代码

我写成了下面的代码。但是好像webdriver_name 根本传不进去。是需要一种别的写法吗？

1. def go_to_baidu(webdriver_name):
   
2.     driver= webdriver.webdriver_name()
   
3.     driver.get('http://www.baidu.com')
   
4.     time.sleep(2)
   
5.     driver.quit()

复制代码

求大神们指点。万分感谢！

写一个函数，传入字符串，判断获取的字符串是百度还是火狐，然后根据条件来初始化不同的类，前往不同的地址，你看看这样可不可以

就不能入参字符串判断下？
def go_to_baidu(webdriver_name):
    driver = None
    if webdriver_name = "Chrome":
          driver = webdriver.Chrome()
    else:
          driver = webdriver.Firefox()
    # driver= webdriver.webdriver_name()
    driver.get('http://www.baidu.com')
    time.sleep(2)
    driver.quit()
还有一种方法，使用eval。
def go_to_baidu(webdriver_name):
    # webdriver_name入参字符串必须正确：Chrome或Firefox
    driver = eval("webdriver.%s()" % webdriver_name)
    driver.get('http://www.baidu.com')
    time.sleep(2)
    driver.quit()