Merge pull request #161 from casschin/get-response-code-timeout

Get response code timeout
mozilla · Feb 4, 2013 · bbb784d · bbb784d
2 parents 14dd4aa + bf81bb7
commit bbb784d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pages/page.py b/pages/page.py
@@ -15,6 +15,7 @@
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.common.exceptions import NoSuchElementException
 from selenium.common.exceptions import ElementNotVisibleException
+from requests.exceptions import Timeout
 
 http_regex = re.compile('https?://((\w+\.)+\w+\.\w+)')
 
@@ -100,5 +101,8 @@ def wait_for_ajax(self):
     def get_response_code(self, url):
         # return the response status
         requests_config = {'max_retries': 5}
-        r = requests.get(url, verify=False, config=requests_config)
-        return r.status_code
+        try:
+            r = requests.get(url, verify=False, allow_redirects=True, config=requests_config, timeout=self.timeout)
+            return r.status_code
+        except Timeout:
+            return 408