REQUEST_TIMEOUT Env Var (#541)

dylanpulver · web-flow · commit f15d7908d27f · 2024-06-28T11:35:29.000-04:00
diff --git a/safety/constants.py b/safety/constants.py
@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
 import configparser
+import os
 from enum import Enum
 from pathlib import Path
 from typing import Optional
 
-
 JSON_SCHEMA_VERSION = '2.0.0'
 
 # TODO fix this
@@ -15,8 +15,8 @@
 DIR_NAME = ".safety"
 
 def get_system_dir() -> Path:
-    import sys
     import os
+    import sys
     raw_dir = os.getenv("SAFETY_SYSTEM_CONFIG_PATH")
     app_data = os.environ.get('ALLUSERSPROFILE', None)
 
@@ -98,7 +98,8 @@ def get_config_setting(name: str) -> Optional[str]:
     DATA_API_BASE_URL
 ]
 
-REQUEST_TIMEOUT = 30
+# Fetch the REQUEST_TIMEOUT from the environment variable, defaulting to 30 if not set
+REQUEST_TIMEOUT = int(os.getenv("SAFETY_REQUEST_TIMEOUT", 30))
 
 # Colors
 YELLOW = 'yellow'
@@ -122,19 +123,19 @@ def get_config_setting(name: str) -> Optional[str]:
 MSG_NO_AUTHD_CICD_PROD_STG_ORG = \
 f"""
 Login to get your API key
-    
+
 To log in: [link]{{LOGIN_URL}}[/link]
-    
+
 Read more at: [link]{DOCS_API_KEY_URL}[/link]
 """
 
 MSG_NO_AUTHD_NOTE_CICD_PROD_STG_TPL = \
 f"""
 Login or register for a free account to get your API key
-    
+
 To log in: [link]{{LOGIN_URL}}[/link]
 To register: [link]{{SIGNUP_URL}}[/link]
-    
+
 Read more at: [link]{DOCS_API_KEY_URL}[/link]
 """
 
