def initialize_driver(cls, driver_type=None): if driver_type is not None: if driver_type == 'chrome': _chrome_driver_file_location = FileLocator.read_config_get_file_location('DriverConfiguration', 'ChromeDriverLocation') cls._driver = webdriver.Chrome(_chrome_driver_file_location) elif driver_type == 'ie': _ie_driver_file_location = FileLocator.read_config_get_file_location('DriverConfiguration', 'IEdriverLocation') cls._driver = webdriver.Ie(_ie_driver_file_location) elif driver_type == 'firefox': cls._driver = webdriver.Firefox() else: raise WebDriverException("Please provide a valid web driver! options: chrome | ie") else: cls._driver = webdriver.Firefox()