Implementing a webdriver factory pattern to set the browser configuration

webdriverfactory.py

import selenium.webdriver as webdriver

class WebdriverFactory:

    @staticmethod
    def set_browser(browserName='chrome'):  #sets chrome as default browser
        if(browserName == 'firefox'):
            driver = webdriver.Firefox()
        elif(browserName == 'chrome'):
            driver = webdriver.Chrome()
        elif(browserName == 'ie'):
            driver = webdriver.Ie()
        elif(browserName == 'safari'):
            driver = webdriver.Safari()
        elif(browserName == 'edge'):
            driver = webdriver.Edge()
        else:
            raise Exception("Invalid browser name: " + browserName + ". Accepted values: firefox, chrome, ie, safari and edge.")
        return driver #returns a webdriver browser object

    def set_remote_browser():
        # TODO: set capabilities for each remote browser
        pass
      

browser = WebdriverFactory.set_browser("firefox")
browser.get("https://github.com/")
print browser.current_url