Show More
@@ -1,60 +1,61 b'' | |||
|
1 | 1 | #!/usr/bin/env python |
|
2 | 2 | |
|
3 | 3 | """This does HTTP GET requests given a host:port and path and returns |
|
4 | 4 | a subset of the headers plus the body of the result.""" |
|
5 | 5 | |
|
6 | 6 | import httplib, sys |
|
7 | 7 | |
|
8 | 8 | try: |
|
9 | 9 | import msvcrt, os |
|
10 | 10 | msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) |
|
11 | 11 | msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) |
|
12 | 12 | except ImportError: |
|
13 | 13 | pass |
|
14 | 14 | |
|
15 | 15 | twice = False |
|
16 | 16 | if '--twice' in sys.argv: |
|
17 | 17 | sys.argv.remove('--twice') |
|
18 | 18 | twice = True |
|
19 | 19 | headeronly = False |
|
20 | 20 | if '--headeronly' in sys.argv: |
|
21 | 21 | sys.argv.remove('--headeronly') |
|
22 | 22 | headeronly = True |
|
23 | 23 | |
|
24 | 24 | reasons = {'Not modified': 'Not Modified'} # python 2.4 |
|
25 | 25 | |
|
26 | 26 | tag = None |
|
27 | 27 | def request(host, path, show): |
|
28 | 28 | assert not path.startswith('/'), path |
|
29 | 29 | global tag |
|
30 | 30 | headers = {} |
|
31 | 31 | if tag: |
|
32 | 32 | headers['If-None-Match'] = tag |
|
33 | 33 | |
|
34 | 34 | conn = httplib.HTTPConnection(host) |
|
35 | 35 | conn.request("GET", '/' + path, None, headers) |
|
36 | 36 | response = conn.getresponse() |
|
37 | 37 | print response.status, reasons.get(response.reason, response.reason) |
|
38 | 38 | if show[:1] == ['-']: |
|
39 | 39 | show = sorted(h for h, v in response.getheaders() |
|
40 | 40 | if h.lower() not in show) |
|
41 | 41 | for h in [h.lower() for h in show]: |
|
42 | 42 | if response.getheader(h, None) is not None: |
|
43 | 43 | print "%s: %s" % (h, response.getheader(h)) |
|
44 | 44 | if not headeronly: |
|
45 | 45 | |
|
46 | data = response.read() | |
|
47 | sys.stdout.write(data) | |
|
46 | if response.status != 500: | |
|
47 | data = response.read() | |
|
48 | sys.stdout.write(data) | |
|
48 | 49 | |
|
49 | 50 | if twice and response.getheader('ETag', None): |
|
50 | 51 | tag = response.getheader('ETag') |
|
51 | 52 | |
|
52 | 53 | return response.status |
|
53 | 54 | |
|
54 | 55 | status = request(sys.argv[1], sys.argv[2], sys.argv[3:]) |
|
55 | 56 | if twice: |
|
56 | 57 | status = request(sys.argv[1], sys.argv[2], sys.argv[3:]) |
|
57 | 58 | |
|
58 | 59 | if 200 <= status <= 305: |
|
59 | 60 | sys.exit(0) |
|
60 | 61 | sys.exit(1) |
General Comments 0
You need to be logged in to leave comments.
Login now