Selenium Wait For Page Load

gistfile1.java

public static void waitForPageLoad(WebDriver drv) {
	sleep(50);		//	Make sure new page init has started...
	try {
		(new WebDriverWait(drv, 5))
        .until(new ExpectedCondition<Boolean>() {
            public Boolean apply(WebDriver d) {
                return  ((JavascriptExecutor)d).executeScript("return document.readyState").equals("complete");
            }
        });
	} catch (TimeoutException ex) {
		 ((JavascriptExecutor)drv).executeScript("window.stop();");
		 sleep(500);
	}
}

Do you have a python equivalent

We have a python example here