Fix cartography-cncf#1326: update CVE request timeouts (cartography-c…

…ncf#1327) Updates the timeouts for requests to be the same as https://github.com/lyft/cartography/blob/07243c224c3e27fb9022d88a95b4331c10cd76ba/cartography/intel/github/util.py#L19 Signed-off-by: chandanchowdhury <[email protected]>
chandanchowdhury · Nov 27, 2024 · 5e643b1 · 5e643b1
1 parent 7e4b86a
commit 5e643b1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cartography/intel/cve/feed.py b/cartography/intel/cve/feed.py
@@ -23,7 +23,8 @@
 logger = logging.getLogger(__name__)
 
 MAX_RETRIES = 3
-REQUEST_TIMEOUT = 10
+# Connect and read timeouts of 60 seconds each; see https://requests.readthedocs.io/en/master/user/advanced/#timeouts
+CONNECT_AND_READ_TIMEOUT = (60, 60)
 CVE_FEED_ID = "NIST_NVD"
 BATCH_SIZE_DAYS = 120
 RESULTS_PER_PAGE = 2000
@@ -87,7 +88,7 @@ def _call_cves_api(url: str, api_key: str, params: Dict[str, Any]) -> Dict[Any,
     while params["resultsPerPage"] > 0 or params["startIndex"] < totalResults:
         try:
             res = requests.get(
-                url, params=params, headers=headers, timeout=REQUEST_TIMEOUT,
+                url, params=params, headers=headers, timeout=CONNECT_AND_READ_TIMEOUT,
             )
             res.raise_for_status()
         except requests.exceptions.HTTPError: