forked from extern/httpie-cli
218 lines
8.6 KiB
Python
218 lines
8.6 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import sys
|
|
import json
|
|
import argparse
|
|
from collections import namedtuple
|
|
import requests
|
|
from requests.structures import CaseInsensitiveDict
|
|
from . import pretty
|
|
from . import __version__ as version
|
|
from . import __doc__ as doc
|
|
|
|
|
|
DEFAULT_UA = 'HTTPie/%s' % version
|
|
SEP_COMMON = ':'
|
|
SEP_DATA = '='
|
|
SEP_DATA_RAW = ':='
|
|
TYPE_FORM = 'application/x-www-form-urlencoded; charset=utf-8'
|
|
TYPE_JSON = 'application/json; charset=utf-8'
|
|
PRETTIFY_STDOUT_TTY_ONLY = object()
|
|
|
|
|
|
KeyValue = namedtuple('KeyValue', ['key', 'value', 'sep'])
|
|
|
|
|
|
class KeyValueType(object):
|
|
|
|
def __init__(self, separators):
|
|
self.separators = separators
|
|
|
|
def __call__(self, string):
|
|
found = dict((string.find(sep), sep)
|
|
for sep in self.separators
|
|
if string.find(sep) != -1)
|
|
|
|
if not found:
|
|
#noinspection PyExceptionInherit
|
|
raise argparse.ArgumentTypeError(
|
|
'"%s" is not a valid value' % string)
|
|
sep = found[min(found.keys())]
|
|
key, value = string.split(sep, 1)
|
|
return KeyValue(key=key, value=value, sep=sep)
|
|
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description=doc.strip())
|
|
|
|
|
|
# Content type.
|
|
group_type = parser.add_mutually_exclusive_group(required=False)
|
|
group_type.add_argument('--json', '-j', action='store_true',
|
|
help='Serialize data items as a JSON object and set'
|
|
' Content-Type to application/json, if not specified.')
|
|
group_type.add_argument('--form', '-f', action='store_true',
|
|
help='Serialize data items as form values and set'
|
|
' Content-Type to application/x-www-form-urlencoded,'
|
|
' if not specified.')
|
|
|
|
# Output options.
|
|
parser.add_argument('--traceback', action='store_true', default=False,
|
|
help='Print a full exception traceback should one'
|
|
' be raised by `requests`.')
|
|
group_pretty = parser.add_mutually_exclusive_group(required=False)
|
|
group_pretty.add_argument('--pretty', '-p', dest='prettify', action='store_true',
|
|
default=PRETTIFY_STDOUT_TTY_ONLY,
|
|
help='If stdout is a terminal, '
|
|
' the response is prettified by default (colorized and'
|
|
' indented if it is JSON). This flag ensures'
|
|
' prettifying even when stdout is redirected.')
|
|
group_pretty.add_argument('--ugly', '-u', help='Do not prettify the response.',
|
|
dest='prettify', action='store_false')
|
|
group_only = parser.add_mutually_exclusive_group(required=False)
|
|
group_only.add_argument('--headers', '-t', dest='print_body',
|
|
action='store_false', default=True,
|
|
help='Print only the response headers.')
|
|
group_only.add_argument('--body', '-b', dest='print_headers',
|
|
action='store_false', default=True,
|
|
help='Print only the response body.')
|
|
parser.add_argument('--style', '-s', dest='style', default='solarized', metavar='STYLE',
|
|
choices=pretty.AVAILABLE_STYLES,
|
|
help='Output coloring style, one of %s. Defaults to solarized.'
|
|
% ', '.join(sorted(pretty.AVAILABLE_STYLES)))
|
|
|
|
# ``requests.request`` keyword arguments.
|
|
parser.add_argument('--auth', '-a', help='username:password',
|
|
type=KeyValueType(SEP_COMMON))
|
|
parser.add_argument('--verify',
|
|
help='Set to "yes" to check the host\'s SSL certificate.'
|
|
' You can also pass the path to a CA_BUNDLE'
|
|
' file for private certs. You can also set '
|
|
'the REQUESTS_CA_BUNDLE environment variable.')
|
|
parser.add_argument('--proxy', default=[], action='append',
|
|
type=KeyValueType(SEP_COMMON),
|
|
help='String mapping protocol to the URL of the proxy'
|
|
' (e.g. http:foo.bar:3128).')
|
|
parser.add_argument('--allow-redirects', default=False, action='store_true',
|
|
help='Set this flag if full redirects are allowed'
|
|
' (e.g. re-POST-ing of data at new ``Location``)')
|
|
parser.add_argument('--file', metavar='PATH', type=argparse.FileType(),
|
|
default=[], action='append',
|
|
help='File to multipart upload')
|
|
parser.add_argument('--timeout', type=float,
|
|
help='Float describes the timeout of the request'
|
|
' (Use socket.setdefaulttimeout() as fallback).')
|
|
|
|
# Positional arguments.
|
|
parser.add_argument('method', metavar='METHOD',
|
|
help='HTTP method to be used for the request'
|
|
' (GET, POST, PUT, DELETE, PATCH, ...).')
|
|
parser.add_argument('url', metavar='URL',
|
|
help='Protocol defaults to http:// if the'
|
|
' URL does not include it.')
|
|
parser.add_argument('items', nargs='*',
|
|
type=KeyValueType([SEP_COMMON, SEP_DATA, SEP_DATA_RAW]),
|
|
help='HTTP header (key:value), data field (key=value)'
|
|
' or raw JSON field (field:=value).')
|
|
|
|
|
|
def main(args=None,
|
|
stdin=sys.stdin,
|
|
stdin_isatty=sys.stdin.isatty(),
|
|
stdout=sys.stdout,
|
|
stdout_isatty=sys.stdout.isatty()):
|
|
|
|
args = parser.parse_args(args if args is not None else sys.argv[1:])
|
|
do_prettify = (args.prettify is True or
|
|
(args.prettify == PRETTIFY_STDOUT_TTY_ONLY and stdout_isatty))
|
|
# Parse request headers and data from the command line.
|
|
headers = CaseInsensitiveDict()
|
|
headers['User-Agent'] = DEFAULT_UA
|
|
data = {}
|
|
for item in args.items:
|
|
value = item.value
|
|
if item.sep == SEP_COMMON:
|
|
target = headers
|
|
else:
|
|
if not stdin_isatty:
|
|
parser.error('Request body (stdin) and request '
|
|
'data (key=value) cannot be mixed.')
|
|
if item.sep == SEP_DATA_RAW:
|
|
try:
|
|
value = json.loads(item.value)
|
|
except ValueError:
|
|
if args.traceback:
|
|
raise
|
|
parser.error('%s:=%s is not valid JSON'
|
|
% (item.key, item.value))
|
|
target = data
|
|
target[item.key] = value
|
|
|
|
if not stdin_isatty:
|
|
data = stdin.read()
|
|
|
|
# JSON/Form content type.
|
|
if args.json or (not args.form and data):
|
|
if stdin_isatty:
|
|
data = json.dumps(data)
|
|
if 'Content-Type' not in headers and (data or args.json):
|
|
headers['Content-Type'] = TYPE_JSON
|
|
elif 'Content-Type' not in headers:
|
|
headers['Content-Type'] = TYPE_FORM
|
|
|
|
# Fire the request.
|
|
try:
|
|
response = requests.request(
|
|
method=args.method.lower(),
|
|
url=args.url if '://' in args.url else 'http://%s' % args.url,
|
|
headers=headers,
|
|
data=data,
|
|
verify=True if args.verify == 'yes' else args.verify,
|
|
timeout=args.timeout,
|
|
auth=(args.auth.key, args.auth.value) if args.auth else None,
|
|
proxies=dict((p.key, p.value) for p in args.proxy),
|
|
files=dict((os.path.basename(f.name), f) for f in args.file),
|
|
allow_redirects=args.allow_redirects,
|
|
)
|
|
except (KeyboardInterrupt, SystemExit) as e:
|
|
sys.stderr.write('\n')
|
|
sys.exit(1)
|
|
except Exception as e:
|
|
if args.traceback:
|
|
raise
|
|
sys.stderr.write(str(e.message) + '\n')
|
|
sys.exit(1)
|
|
|
|
# Reconstruct the raw response.
|
|
encoding = response.encoding or 'ISO-8859-1'
|
|
original = response.raw._original_response
|
|
status_line, headers, body = (
|
|
'HTTP/{version} {status} {reason}'.format(
|
|
version='.'.join(str(original.version)),
|
|
status=original.status, reason=original.reason,
|
|
),
|
|
str(original.msg).decode(encoding),
|
|
response.content.decode(encoding) if response.content else u''
|
|
)
|
|
|
|
if do_prettify:
|
|
prettify = pretty.PrettyHttp(args.style)
|
|
if args.print_headers:
|
|
status_line = prettify.headers(status_line)
|
|
headers = prettify.headers(headers)
|
|
if args.print_body and 'Content-Type' in response.headers:
|
|
body = prettify.body(body, response.headers['Content-Type'])
|
|
|
|
# Output.
|
|
if args.print_headers:
|
|
stdout.write(status_line)
|
|
stdout.write('\n')
|
|
stdout.write(headers.encode('utf-8'))
|
|
stdout.write('\n\n')
|
|
if args.print_body:
|
|
stdout.write(body.encode('utf-8'))
|
|
stdout.write('\n')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|