Python Selenium WebDriver how to add timeout to get(url) function

Question

I am running a simple piece of code that downloads a website through proxy, however sometimes the proxy can be slow and this can cause the WebDriver .get(url) request to block indefinitely.

Is there a simple piece of Python code for WebDriver that will set a timeout for this function? Through searching I've only found techniques that work for java.

driver.get(url)

Answer 1

For all the web cretents out there what I used to solve this problem was this. Selenium uses the socket library so I set a timeout on the socket module, this throws an error which we can use to send the escape key to the browser (which stops the page loading):

socket.setdefaulttimeout(2)
try:
     driver.get(pageLink)
except socket.timeout:
     #send ESCAPE key to browser