43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
import json
|
|
import time
|
|
import random
|
|
import requests
|
|
import warnings
|
|
|
|
import arjun.core.config as mem
|
|
|
|
from arjun.core.utils import dict_to_xml
|
|
|
|
warnings.filterwarnings('ignore') # Disable SSL related warnings
|
|
|
|
def requester(request, payload={}):
|
|
"""
|
|
central function for making http requests
|
|
returns str on error otherwise response object of requests library
|
|
"""
|
|
if 'include' in request and request['include']:
|
|
payload.update(request['include'])
|
|
if mem.var['stable']:
|
|
mem.var['delay'] = random.choice(range(6, 12))
|
|
time.sleep(mem.var['delay'])
|
|
url = request['url']
|
|
if mem.var['kill']:
|
|
return 'killed'
|
|
try:
|
|
if request['method'] == 'GET':
|
|
response = requests.get(url, params=payload, headers=request['headers'], verify=False, timeout=mem.var['timeout'])
|
|
elif request['method'] == 'JSON':
|
|
if mem.var['include'] and '$arjun$' in mem.var['include']:
|
|
payload = mem.var['include'].replace('$arjun$', json.dumps(payload).rstrip('}').lstrip('{'))
|
|
response = requests.post(url, data=payload, headers=request['headers'], verify=False, timeout=mem.var['timeout'])
|
|
else:
|
|
response = requests.post(url, json=payload, headers=request['headers'], verify=False, timeout=mem.var['timeout'])
|
|
elif request['method'] == 'XML':
|
|
payload = mem.var['include'].replace('$arjun$', dict_to_xml(payload))
|
|
response = requests.post(url, data=payload, headers=request['headers'], verify=False, timeout=mem.var['timeout'])
|
|
else:
|
|
response = requests.post(url, data=payload, headers=request['headers'], verify=False, timeout=mem.var['timeout'])
|
|
return response
|
|
except Exception as e:
|
|
return str(e)
|