Mercurial > hg
view tests/get-with-headers.py @ 10716:5f92bde72eef
check-code: Only call check-code if __name__ = "__main__".
This changeset moves the code that actually executes something to the
"if __name__ = '__main__'" section to allow the file to be imported as
a module.
author | Pierre-Yves David <pierre-yves.david@logilab.fr> |
---|---|
date | Tue, 16 Mar 2010 19:52:56 +0100 |
parents | 4d9dea174b84 |
children | 13a1b2fb7ef2 |
line wrap: on
line source
#!/usr/bin/env python """This does HTTP GET requests given a host:port and path and returns a subset of the headers plus the body of the result.""" import httplib, sys, re try: import msvcrt, os msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) except ImportError: pass headers = [h.lower() for h in sys.argv[3:]] conn = httplib.HTTPConnection(sys.argv[1]) conn.request("GET", sys.argv[2]) response = conn.getresponse() print response.status, response.reason for h in headers: if response.getheader(h, None) is not None: print "%s: %s" % (h, response.getheader(h)) print data = response.read() sys.stdout.write(data) if 200 <= response.status <= 299: sys.exit(0) sys.exit(1)