BasePage.py 文件源码

def __init__(self, driver, browser='chrome'):
        """
        initialize selenium webdriver, use chrome as default webdriver
        """
        if not driver:
            if browser == "firefox" or browser == "ff":
                driver = webdriver.Firefox()
            elif browser == "chrome":
                driver = webdriver.Chrome()
            elif browser == "internet explorer" or browser == "ie":
                driver = webdriver.Ie()
            elif browser == "opera":
                driver = webdriver.Opera()
            elif browser == "phantomjs":
                driver = webdriver.PhantomJS()
            try:
                self.driver = driver
                self.wait = WebDriverWait(self.driver, 20)
            except Exception:
                raise NameError("Not found %s browser,You can enter 'ie', 'ff' or 'chrome'." % browser)
        else:
            self.driver = driver
            self.wait = WebDriverWait(self.driver, 20)