Add "timeout" to "requests.get()" calls

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -111,20 +111,20 @@ def requests_get(url, proxy_string=None):
     if proxy_string:
         proxies = {protocol: proxy_string}
         try:
-            response = requests.get(url, proxies=proxies)
+            response = requests.get(url, proxies=proxies, timeout=3)
         except Exception:
             # Prevent SSLCertVerificationError / CERTIFICATE_VERIFY_FAILED
             url = url.replace("https://", "http://")
             time.sleep(0.04)
-            response = requests.get(url, proxies=proxies)
+            response = requests.get(url, proxies=proxies, timeout=4)
     else:
         try:
-            response = requests.get(url)
+            response = requests.get(url, timeout=3)
         except Exception:
             # Prevent SSLCertVerificationError / CERTIFICATE_VERIFY_FAILED
             url = url.replace("https://", "http://")
             time.sleep(0.04)
-            response = requests.get(url)
+            response = requests.get(url, timeout=4)
     return response
 
 

seleniumbase/fixtures/js_utils.py

@@ -531,7 +531,7 @@ def add_css_style(driver, css_style):
 def add_js_code_from_link(driver, js_link):
     if js_link.startswith("//"):
         js_link = "http:" + js_link
-    js_code = requests.get(js_link).text
+    js_code = requests.get(js_link, timeout=5).text
     add_js_code_script = (
         """var body_tag=document.getElementsByTagName('body').item(0);"""
         """var script_tag=document.createElement("script");"""

seleniumbase/fixtures/page_utils.py

@@ -245,7 +245,7 @@ def _download_file_to(file_url, destination_folder, new_file_name=None):
         file_name = new_file_name
     else:
         file_name = file_url.split("/")[-1]
-    r = requests.get(file_url)
+    r = requests.get(file_url, timeout=5)
     file_path = os.path.join(destination_folder, file_name)
     download_file_lock = fasteners.InterProcessLock(
         constants.MultiBrowser.DOWNLOAD_FILE_LOCK

seleniumbase/undetected/__init__.py

@@ -143,7 +143,7 @@ def __init__(
 
         special_port_free = False  # If the port isn't free, don't use 9222
         try:
-            res = requests.get("http://127.0.0.1:9222")
+            res = requests.get("http://127.0.0.1:9222", timeout=1)
             if res.status_code != 200:
                 raise Exception("The port is free! It will be used!")
         except Exception: