get-with-headers.py
60 lines
| 1.6 KiB
| text/x-python
|
PythonLexer
/ tests / get-with-headers.py
|
r2532 | #!/usr/bin/env python | ||
|
r8447 | """This does HTTP GET requests given a host:port and path and returns | ||
|
r2532 | a subset of the headers plus the body of the result.""" | ||
|
r10905 | import httplib, sys | ||
|
r7054 | |||
try: | ||||
import msvcrt, os | ||||
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) | ||||
msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) | ||||
except ImportError: | ||||
pass | ||||
|
r12182 | twice = False | ||
if '--twice' in sys.argv: | ||||
sys.argv.remove('--twice') | ||||
twice = True | ||||
|
r18400 | headeronly = False | ||
if '--headeronly' in sys.argv: | ||||
sys.argv.remove('--headeronly') | ||||
headeronly = True | ||||
|
r12182 | |||
|
r12250 | reasons = {'Not modified': 'Not Modified'} # python 2.4 | ||
|
r12182 | tag = None | ||
def request(host, path, show): | ||||
|
r17017 | assert not path.startswith('/'), path | ||
|
r12182 | global tag | ||
headers = {} | ||||
if tag: | ||||
headers['If-None-Match'] = tag | ||||
|
r5561 | |||
|
r12182 | conn = httplib.HTTPConnection(host) | ||
|
r17017 | conn.request("GET", '/' + path, None, headers) | ||
|
r12182 | response = conn.getresponse() | ||
|
r12250 | print response.status, reasons.get(response.reason, response.reason) | ||
|
r18380 | if show[:1] == ['-']: | ||
|
r18393 | show = sorted(h for h, v in response.getheaders() | ||
if h.lower() not in show) | ||||
|
r12182 | for h in [h.lower() for h in show]: | ||
if response.getheader(h, None) is not None: | ||||
print "%s: %s" % (h, response.getheader(h)) | ||||
|
r18400 | if not headeronly: | ||
data = response.read() | ||||
sys.stdout.write(data) | ||||
|
r12182 | |||
|
r18400 | if twice and response.getheader('ETag', None): | ||
tag = response.getheader('ETag') | ||||
|
r12182 | |||
return response.status | ||||
status = request(sys.argv[1], sys.argv[2], sys.argv[3:]) | ||||
if twice: | ||||
status = request(sys.argv[1], sys.argv[2], sys.argv[3:]) | ||||
if 200 <= status <= 305: | ||||
|
r5561 | sys.exit(0) | ||
sys.exit(1) | ||||