##// END OF EJS Templates
get-with-headers: don't block indefinitely if the server had an internal error...
Javi Merino -
r19865:ba6577a1 default
parent child Browse files
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 print
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