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()
