| lothiraldan | |
| durin42 |
| hg-reviewers |
We add the ability to specify arbitrary HTTP request headers and
to save the HTTP response body to a file. These will be used in
upcoming commits.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| except ImportError: | except ImportError: | ||||
| pass | pass | ||||
| parser = argparse.ArgumentParser() | parser = argparse.ArgumentParser() | ||||
| parser.add_argument('--twice', action='store_true') | parser.add_argument('--twice', action='store_true') | ||||
| parser.add_argument('--headeronly', action='store_true') | parser.add_argument('--headeronly', action='store_true') | ||||
| parser.add_argument('--json', action='store_true') | parser.add_argument('--json', action='store_true') | ||||
| parser.add_argument('--hgproto') | parser.add_argument('--hgproto') | ||||
| parser.add_argument('--requestheader', nargs='*', default=[], | |||||
| help='Send an additional HTTP request header. Argument ' | |||||
| 'value is <header>=<value>') | |||||
| parser.add_argument('--bodyfile', | |||||
| help='Write HTTP response body to a file') | |||||
| parser.add_argument('host') | parser.add_argument('host') | ||||
| parser.add_argument('path') | parser.add_argument('path') | ||||
| parser.add_argument('show', nargs='*') | parser.add_argument('show', nargs='*') | ||||
| args = parser.parse_args() | args = parser.parse_args() | ||||
| twice = args.twice | twice = args.twice | ||||
| headeronly = args.headeronly | headeronly = args.headeronly | ||||
| formatjson = args.json | formatjson = args.json | ||||
| hgproto = args.hgproto | hgproto = args.hgproto | ||||
| requestheaders = args.requestheader | |||||
| tag = None | tag = None | ||||
| def request(host, path, show): | def request(host, path, show): | ||||
| assert not path.startswith('/'), path | assert not path.startswith('/'), path | ||||
| global tag | global tag | ||||
| headers = {} | headers = {} | ||||
| if tag: | if tag: | ||||
| headers['If-None-Match'] = tag | headers['If-None-Match'] = tag | ||||
| if hgproto: | if hgproto: | ||||
| headers['X-HgProto-1'] = hgproto | headers['X-HgProto-1'] = hgproto | ||||
| for header in requestheaders: | |||||
| key, value = header.split('=', 1) | |||||
| headers[key] = value | |||||
| conn = httplib.HTTPConnection(host) | conn = httplib.HTTPConnection(host) | ||||
| conn.request("GET", '/' + path, None, headers) | conn.request("GET", '/' + path, None, headers) | ||||
| response = conn.getresponse() | response = conn.getresponse() | ||||
| print(response.status, response.reason) | print(response.status, response.reason) | ||||
| if show[:1] == ['-']: | if show[:1] == ['-']: | ||||
| show = sorted(h for h, v in response.getheaders() | show = sorted(h for h, v in response.getheaders() | ||||
| if h.lower() not in show) | if h.lower() not in show) | ||||
| for h in [h.lower() for h in show]: | for h in [h.lower() for h in show]: | ||||
| if response.getheader(h, None) is not None: | if response.getheader(h, None) is not None: | ||||
| print("%s: %s" % (h, response.getheader(h))) | print("%s: %s" % (h, response.getheader(h))) | ||||
| if not headeronly: | if not headeronly: | ||||
| print() | print() | ||||
| data = response.read() | data = response.read() | ||||
| if args.bodyfile: | |||||
| bodyfh = open(args.bodyfile, 'wb') | |||||
| else: | |||||
| bodyfh = sys.stdout | |||||
| # Pretty print JSON. This also has the beneficial side-effect | # Pretty print JSON. This also has the beneficial side-effect | ||||
| # of verifying emitted JSON is well-formed. | # of verifying emitted JSON is well-formed. | ||||
| if formatjson: | if formatjson: | ||||
| # json.dumps() will print trailing newlines. Eliminate them | # json.dumps() will print trailing newlines. Eliminate them | ||||
| # to make tests easier to write. | # to make tests easier to write. | ||||
| data = json.loads(data) | data = json.loads(data) | ||||
| lines = json.dumps(data, sort_keys=True, indent=2).splitlines() | lines = json.dumps(data, sort_keys=True, indent=2).splitlines() | ||||
| for line in lines: | for line in lines: | ||||
| print(line.rstrip()) | bodyfh.write(line.rstrip()) | ||||
| bodyfh.write(b'\n') | |||||
| else: | else: | ||||
| sys.stdout.write(data) | bodyfh.write(data) | ||||
| if args.bodyfile: | |||||
| bodyfh.close() | |||||
| if twice and response.getheader('ETag', None): | if twice and response.getheader('ETag', None): | ||||
| tag = response.getheader('ETag') | tag = response.getheader('ETag') | ||||
| return response.status | return response.status | ||||
| status = request(args.host, args.path, args.show) | status = request(args.host, args.path, args.show) | ||||
| if twice: | if twice: | ||||
| status = request(args.host, args.path, args.show) | status = request(args.host, args.path, args.show) | ||||
| if 200 <= status <= 305: | if 200 <= status <= 305: | ||||
| sys.exit(0) | sys.exit(0) | ||||
| sys.exit(1) | sys.exit(1) | ||||