| lothiraldan | |
| durin42 |
| hg-reviewers |
I'm about to add another flag and I don't want to deal with this
organic, artisinal argument parser.
| Lint Skipped |
| Unit Tests Skipped |
| #!/usr/bin/env python | #!/usr/bin/env python | ||||
| """This does HTTP GET requests given a host:port and path and returns | """This does HTTP GET requests given a host:port and path and returns | ||||
| a subset of the headers plus the body of the result.""" | a subset of the headers plus the body of the result.""" | ||||
| from __future__ import absolute_import, print_function | from __future__ import absolute_import, print_function | ||||
| import argparse | |||||
| import json | import json | ||||
| import os | import os | ||||
| import sys | import sys | ||||
| from mercurial import ( | from mercurial import ( | ||||
| util, | util, | ||||
| ) | ) | ||||
| httplib = util.httplib | httplib = util.httplib | ||||
| try: | try: | ||||
| import msvcrt | import msvcrt | ||||
| msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) | msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) | ||||
| msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) | msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) | ||||
| except ImportError: | except ImportError: | ||||
| pass | pass | ||||
| twice = False | parser = argparse.ArgumentParser() | ||||
| if '--twice' in sys.argv: | parser.add_argument('--twice', action='store_true') | ||||
| sys.argv.remove('--twice') | parser.add_argument('--headeronly', action='store_true') | ||||
| twice = True | parser.add_argument('--json', action='store_true') | ||||
| headeronly = False | parser.add_argument('--hgproto') | ||||
| if '--headeronly' in sys.argv: | parser.add_argument('host') | ||||
| sys.argv.remove('--headeronly') | parser.add_argument('path') | ||||
| headeronly = True | parser.add_argument('show', nargs='*') | ||||
| formatjson = False | |||||
| if '--json' in sys.argv: | args = parser.parse_args() | ||||
| sys.argv.remove('--json') | |||||
| formatjson = True | twice = args.twice | ||||
| headeronly = args.headeronly | |||||
| hgproto = None | formatjson = args.json | ||||
| if '--hgproto' in sys.argv: | hgproto = args.hgproto | ||||
| idx = sys.argv.index('--hgproto') | |||||
| hgproto = sys.argv[idx + 1] | |||||
| sys.argv.pop(idx) | |||||
| sys.argv.pop(idx) | |||||
| 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 | ||||
| else: | else: | ||||
| sys.stdout.write(data) | sys.stdout.write(data) | ||||
| 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(sys.argv[1], sys.argv[2], sys.argv[3:]) | status = request(args.host, args.path, args.show) | ||||
| if twice: | if twice: | ||||
| status = request(sys.argv[1], sys.argv[2], sys.argv[3:]) | 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) | ||||