pyasynchat_http_client_with_header_body_terminators.py

1import asynchat
2import asyncore
3import socket
4
5class HTTPClient(asynchat.async_chat):
6
7    def __init__(self, host, path):
8        asynchat.async_chat.__init__(self)
9        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
10        self.connect((host, 80))
11        self.path = path
12        self.set_terminator(b"\r\n\r\n")
13        self.reading_headers = True
14        self.buffer = b""
15
16    def collect_incoming_data(self, data):
17        self.buffer = self.buffer + data
18
19    def found_terminator(self):
20        if self.reading_headers:
21            self.reading_headers = False
22            self.parse_headers(self.buffer)
23            self.buffer = b""
24            # Now determine the content-length
25            # and set the terminator to that length.
26            # (Omitted for brevity in the official snippet)
27            self.set_terminator(None)
28        else:
29            self.process_body(self.buffer)
30            self.buffer = b""
31            self.close_when_done()
32
33    def parse_headers(self, data):
34        # In a real app, you would parse the HTTP headers here
35        print("Headers received.")
36
37    def process_body(self, data):
38        # In a real app, you would process the HTTP body here
39        print("Body received.")
40
41    def handle_connect(self):
42        # Send the HTTP request
43        request = "GET %s HTTP/1.0\r\n\r\n" % self.path
44        self.push(request.encode('ascii'))
45
46if __name__ == '__main__':
47    client = HTTPClient('www.python.org', '/')
48    asyncore.loop()