封装启动浏览器方法

封装启动浏览器方法

#coding=utf-8
from selenium import  webdriver
def browser(browser='firefox'):
    '''打开火狐、谷歌浏览器用法，如谷歌：driver=browser('chrome')'''
    try:
       if browser=="firefox":
        driver=webdriver.Firefox()
        return driver
       elif browser=="chrome":
           driver=webdriver.Chrome()
           return driver
       else:
           print("Not found browser!You can enter 'firefox','chrome'")
    except Exception as msg:
        print("open browser error:%s"%msg)

if __name__=="__main__":
    # 用火狐浏览器打开百度网页
    driver_firefox=browser()
    driver_firefox.get("https://www.baidu.com")
    # 获取浏览器的名称
    print("open browser:%s"%driver_firefox.name)
    # 获取网页名称
    print(driver_firefox.title)


    # 用谷歌浏览器打开天龟网页
    driver_chrome=browser("chrome")
    driver_chrome.get("http://www.tianguiedu.com")
    # 获取浏览器的名称
    print("open browser: %s"%driver_chrome.name)
    # 获取网页名称
    print(driver_chrome.title)
    driver_chrome.quit()