#!/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()