Skip to content

Commit d4a9461

Browse files
committed
Added a configurable timeout for requests
1 parent c6630ac commit d4a9461

File tree

2 files changed

+14
-2
lines changed

2 files changed

+14
-2
lines changed

socketsecurity/core/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ def do_request(
199199
verify=verify
200200
)
201201
except ReadTimeout:
202-
raise RequestTimeoutExceeded(f"Configured timeout {timeout} reached for request for path {path}")
202+
raise RequestTimeoutExceeded(f"Configured timeout {timeout} reached for request for path {url}")
203203
output_headers = headers.copy()
204204
output_headers['Authorization'] = "API_KEY_REDACTED"
205205
output = {

socketsecurity/socketcli.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,16 @@
161161
default=False
162162
)
163163

164+
parser.add_argument(
165+
'--timeout',
166+
default=1200,
167+
help='Timeout configuration for each request. Defaults to 1200 and applies to each unique HTTP request',
168+
required=False,
169+
type=float
170+
)
171+
172+
173+
164174

165175
def output_console_comments(diff_report: Diff, sbom_file_name: str = None) -> None:
166176
if diff_report.id != "NO_DIFF_RAN":
@@ -252,6 +262,8 @@ def main_code():
252262
ignore_commit_files = arguments.ignore_commit_files
253263
disable_blocking = arguments.disable_blocking
254264
allow_unverified = arguments.allow_unverified
265+
timeout = arguments.timeout
266+
255267
if disable_blocking:
256268
global blocking_disabled
257269
blocking_disabled = True
@@ -308,7 +320,7 @@ def main_code():
308320
default_branch = scm.is_default_branch
309321

310322
base_api_url = os.getenv("BASE_API_URL") or None
311-
core = Core(token=api_token, request_timeout=1200, base_api_url=base_api_url, allow_unverified=allow_unverified)
323+
core = Core(token=api_token, request_timeout=timeout, base_api_url=base_api_url, allow_unverified=allow_unverified)
312324
no_change = True
313325
if ignore_commit_files:
314326
no_change = False

0 commit comments

Comments
 (0)