def build_httpie_command(request):
parts = ['http']
redir_input = ''
# Method
if request.command != 'GET':
parts.append(request.command)
# URL
parts.append(shlex_quote(request.url()))
# Authorization (prepare)
method, token = request.auth()
# Headers
for header in sorted(request.headers):
if header in EXCLUDE_HEADERS_HTTP:
continue
parts.append('{}:{}'.format(header, shlex_quote(
request.headers[header]))) # noqa
if method != 'Basic' and 'Authorization' in request.headers:
header = 'Authorization'
parts.append('{}:{}'.format(header, shlex_quote(
request.headers[header]))) # noqa
# JSON or raw data
data = maybe_str(request.data())
if data:
content_type = request.headers.get('Content-Type', '')
if is_json(content_type):
# We need to explicitly set the separators to get consistent
# whitespace handling across Python 2 and 3. See
# https://bugs.python.org/issue16333 for details.
redir_input = shlex_quote(
json.dumps(data,
indent=2,
sort_keys=True,
separators=(',', ': ')))
elif is_form(content_type):
parts[0] += ' --form'
parts.append(data.replace('\n', ' '))
else:
redir_input = shlex_quote(data)
# Authorization
if method == 'Basic':
parts.append('-a {}'.format(token))
cmd = ' '.join(parts)
if not redir_input:
return cmd
else:
return 'echo {} | {}'.format(redir_input, cmd)
def data(self):
payload_bytes = self.rfile.read()
if payload_bytes:
if is_json(self.headers.get('Content-Type', '')):
assert isinstance(payload_bytes, bytes)
payload_str = payload_bytes.decode('utf-8')
return ordered(json.loads(payload_str))
else:
return payload_bytes
def build_wget_command(request):
parts = ['wget', '-S', '-O-']
# Method
if request.command not in ['GET', 'POST']:
parts.append('--method={}'.format(request.command))
# URL
parts.append(shlex_quote(request.url()))
# Authorization (prepare)
method, token = request.auth()
# Headers
for header in sorted(request.headers):
if header in EXCLUDE_HEADERS:
continue
header_line = shlex_quote('{}: {}'.format(
header, request.headers[header])) # noqa: E501
parts.append('--header={}'.format(header_line))
if method != 'Basic' and 'Authorization' in request.headers:
header = 'Authorization'
header_line = shlex_quote('{}: {}'.format(
header, request.headers[header])) # noqa: E501
parts.append('--header={}'.format(header_line))
# JSON or raw data
data = maybe_str(request.data())
if data:
if is_json(request.headers.get('Content-Type', '')):
data = json.dumps(data)
if request.command == 'POST':
parts.append('--post-data=\'{}\''.format(data))
elif request.command != 'POST':
parts.append('--body-data=\'{}\''.format(data))
# Authorization
if method == 'Basic':
user, password = token.split(':')
parts.append('--auth-no-challenge')
parts.append('--user={}'.format(user))
parts.append('--password={}'.format(password))
return ' '.join(parts)
def build_curl_command(request):
parts = ['curl', '-i']
# Method
if request.command != 'GET':
parts.append('-X {}'.format(request.command))
# URL
parts.append(shlex_quote(request.url()))
# Authorization (prepare)
method, token = request.auth()
# Headers
for header in sorted(request.headers):
if header in EXCLUDE_HEADERS:
continue
header_line = shlex_quote('{}: {}'.format(
header, request.headers[header])) # noqa: E501
parts.append('-H {}'.format(header_line))
if method != 'Basic' and 'Authorization' in request.headers:
header = 'Authorization'
header_line = shlex_quote('{}: {}'.format(
header, request.headers[header])) # noqa: E501
parts.append('-H {}'.format(header_line))
# JSON
data = maybe_str(request.data())
if data:
if is_json(request.headers.get('Content-Type', '')):
data = json.dumps(data)
parts.append('--data-raw \'{}\''.format(data))
# Authorization
if method == 'Basic':
parts.append('--user {}'.format(token))
return ' '.join(parts)
def build_requests_command(request):
# Method
tree = ast.parse('requests.{}()'.format(request.command.lower()))
call = tree.body[0].value
call.keywords = []
# URL
call.args.append(ast.Str(request.url()))
# Authorization (prepare)
method, token = request.auth()
# Headers
header_keys = []
header_values = []
for header in sorted(request.headers):
if header in EXCLUDE_HEADERS_REQUESTS:
continue
header_keys.append(ast.Str(header))
header_values.append(ast.Str(request.headers[header]))
if method != 'Basic' and 'Authorization' in request.headers:
header_keys.append(ast.Str('Authorization'))
header_values.append(ast.Str(request.headers['Authorization']))
if header_keys and header_values:
call.keywords.append(
ast.keyword('headers', ast.Dict(header_keys, header_values)))
# JSON or raw data
data = maybe_str(request.data())
def astify_json_obj(obj):
obj = maybe_str(obj)
if isinstance(obj, str):
return ast.Str(obj)
elif isinstance(obj, bool):
return ast.Name(obj, ast.Load())
elif isinstance(obj, int):
return ast.Name(obj, ast.Load())
elif isinstance(obj, float):
return ast.Name(obj, ast.Load())
elif isinstance(obj, list):
json_values = []
for v in obj:
json_values.append(astify_json_obj(v))
return ast.List(json_values, ast.Load())
elif isinstance(obj, dict):
json_values = []
json_keys = []
for k, v in obj.items():
json_keys.append(ast.Str(maybe_str(k)))
json_values.append(astify_json_obj(v))
return ast.Dict(json_keys, json_values)
else:
raise Exception('Cannot astify {0:s}'.format(str(obj)))
if data:
if is_json(request.headers.get('Content-Type', '')):
call.keywords.append(ast.keyword('json', astify_json_obj(data)))
else:
call.keywords.append(ast.keyword('data', ast.Str(data)))
# Authorization
if method == 'Basic':
token = maybe_str(token)
call.keywords.append(
ast.keyword('auth',
ast.Tuple(tuple(map(ast.Str, token.split(':'))),
None)))
return astunparse.unparse(tree).strip()
def test_is_json(ctype, expected):
assert utils.is_json(ctype) is expected
