1、if 判断写法:
def open_browser(text):
driver_type = text
if driver_type == "Chrome":
driver = webdriver.Chrome()
if driver_type == "Firefox":
driver = webdriver.Firefox()
if driver_type == "Edge":
driver = webdriver.Edge()
if driver_type == "Ie":
driver = webdriver.Ie()
return driver
2、根据字符串来识别字典里数据
def open_browser(driver_type):
driver_dict={
"Chrome": webdriver.Chrome,
"firefox": webdriver.Firefox,
"Edge": webdriver.Edge,
"ie": webdriver.Ie
}
driver = driver_dict[driver_type]()
return driver
3、根据反射机制来创建对象
def open_browser(driver_type):
try:
# getattr(webdriver, driver_type)() 这个括号一个定要记得写,不然浏览器会启动不了
# 个人猜想:getattr() 方法返回的应该是一个字符串,只有在后面加了(),才会成为函数的调用
driver = getattr(webdriver, driver_type)()
except Exception as e:
print(e)
driver = webdriver.Chrome()
return driver