with contextlib.closing(webdriver.Firefox()) as driver:
    driver.get('http://www.google.com')
wait = ui.WebDriverWait(driver, 10)
How can I make Selenium/Python wait for the user to login before continuing to run?