Refactor create_sonar_query function and update status code handling

talltechy · Apr 19, 2024 · 4a13ad4 · 4a13ad4
1 parent 089cff9
commit 4a13ad4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/rapid7/api_r7_isvm_sonar_add.py b/src/rapid7/api_r7_isvm_sonar_add.py
@@ -6,7 +6,7 @@
 import requests
 from dotenv import dotenv_values
 from requests.auth import HTTPBasicAuth
-from status_codes import create_sonar_query_sm
+from api_r7_status_codes import create_sonar_query_sm
 
 secrets = dotenv_values(".env")
 
@@ -35,7 +35,7 @@ def create_sonar_query(url, name, criteria, username, password):
         "name": name,
         "criteria": criteria
     }
-    response = requests.post(url, auth=HTTPBasicAuth(username, password), headers=headers, json=payload, timeout=1, verify=False)
+    response = requests.post(url, auth=HTTPBasicAuth(username, password), headers=headers, json=payload, timeout=10, verify=False)
 
     # Get the user-friendly message based on the status code
     message = create_sonar_query_sm(response.status_code, response)

diff --git a/src/rapid7/status_codes.py → src/rapid7/api_r7_status_codes.py b/src/rapid7/status_codes.py → src/rapid7/api_r7_status_codes.py