Untitled

#!/usr/bin/env python
"""An externally-accessible toy server whith several response strategies"""

import socket
import sys
from multiprocessing import Pool

def web_scrape(site):
    s = socket.socket()
    s.connect((site, 1333))
    s.send('GET / \r\n\r\n')
    response_characters = []
    while True:
        print 'starting loop'
        o = s.recv(1000)
        if not o:
            break
        response_characters.append(o)
    return "".join(response_characters)

def get_local_address_hack():
    s = socket.socket()
    s.connect(('google.com', 80))
    address, port = s.getsockname()
    return address

def long_running_request(client_socket):
    print 'calculating complicated response...'
    for x in xrange(10**8):
        pass
    print '...finished'
    s = client_socket.recv(1000)
    open('output.txt', 'a').write(s)
    # I'm confused about what makes a valid http response
    test_response = "HTTP/1.0 200 OK\r\n\r\n<html><body>This is a response!</body></html>"
    client_socket.send(test_response)
    client_socket.close()

class Server(object):
    def __init__(self, port=8000):
        self.address = get_local_address_hack()
        self.port = port
        self.socket = socket.socket()
        self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.socket.bind((self.address, self.port))
        print 'ip to hit:'
        print self.address,':', self.port
        self.socket.listen(1)

    def blocking_listen(self, response_function):
        (client_socket, client_address) = self.socket.accept()
        print 'client connected at ', client_address, 'connected'
        response_function(client_socket)

    def multiprocessing_listen(self, response_function, pool_size=3):
        """Takes a function(client_socket) to use for responses."""

        self.pool = Pool(pool_size)
        print 'starting listen loop'
        while True:
            (client_socket, client_address) = self.socket.accept()
            print client_socket, 'at', client_address, 'connected'
            self.pool.apply_async(response_function, [client_socket])

if __name__ == '__main__':
    #print web_scrape('localhost')
    #raw_input('wait while setup server again')
    #s = '10.242.11.247'
    #print web_scrape(s)
    s = Server(8000)
    s.blocking_listen(long_running_request)