Added SSL Certificate Verification Options

Added SSL Certificate Verification Options
This commit is contained in:
swatcats
2024-03-02 16:34:51 -05:00
parent c68a083e97
commit 0ff9d53e43
4 changed files with 21 additions and 9 deletions

View File

@@ -10,7 +10,7 @@ from config.config import settings_dict
async def rest_get(url, api_key=None, params=None, cookies=None):
try:
headers = {'X-Api-Key': api_key} if api_key else None
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.get(url, params=params, headers=headers, cookies=cookies))
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.get(url, params=params, headers=headers, cookies=cookies, verify=settings_dict['SSL_VERIFICATION']))
response.raise_for_status()
return response.json()
except requests.exceptions.HTTPError as e:
@@ -26,7 +26,7 @@ async def rest_delete(url, api_key, params=None):
if settings_dict['TEST_RUN']: return
try:
headers = {'X-Api-Key': api_key}
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.delete(url, params=params, headers=headers))
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.delete(url, params=params, headers=headers, verify=settings_dict['SSL_VERIFICATION']))
response.raise_for_status()
if response.status_code in [200, 204]:
return None
@@ -42,7 +42,7 @@ async def rest_delete(url, api_key, params=None):
async def rest_post(url, data=None, json=None, headers=None, cookies=None):
if settings_dict['TEST_RUN']: return
try:
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.post(url, data=data, json=json, headers=headers, cookies=cookies))
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.post(url, data=data, json=json, headers=headers, cookies=cookies, verify=settings_dict['SSL_VERIFICATION']))
response.raise_for_status()
if response.status_code in (200,201):
return None
@@ -60,7 +60,7 @@ async def rest_put(url, api_key, data):
if settings_dict['TEST_RUN']: return
try:
headers = {'X-Api-Key': api_key} | {"content-type": "application/json"}
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.put(url, data=data, headers=headers))
response = await asyncio.get_event_loop().run_in_executor(None, lambda: requests.put(url, data=data, headers=headers, verify=settings_dict['SSL_VERIFICATION']))
response.raise_for_status()
return response.json()
except RequestException as e: