raw socket http - keep alive

from socket import socket
import sys

def readlines(sock, recv_buffer=4096, delim='\n', wait=False):
        buffer = ''
        data = True
        while data:
                data = sock.recv(recv_buffer)
                buffer += data

                while buffer.find(delim) != -1:
                        line, buffer = buffer.split('\n', 1)
                        yield line
                if not wait: break
        return

sock = socket()
host = sys.argv[1]
port = int(sys.argv[2])
print host
print port
sock.connect((host, port))
sock.sendall('GET / HTTP/1.1\r\nHOST: %s\r\nConnection: keep-alive\r\n\r\n' % host)
for line in readlines(sock):
   print line


sock.sendall('GET / HTTP/1.1\r\nHOST: %s\r\nConnection: keep-alive\r\n\r\n' % host)
for line in readlines(sock):
   print line
sock.sendall('GET / HTTP/1.1\r\nHOST: %s\r\nConnection: keep-alive\r\n\r\n' % host)
for line in readlines(sock,wait=True):
   print line
sock.close()