asyncore, asynchat and pickle

A little demo I did for spike using asycore and asynchat that sends pickles around.

(yes the delimiter sucks, and i could really not of used asynchat at all... :)

easymsg.py ------------------------------------------------------------------
import asyncore, asynchat
import os, socket, string
import pickle

PORT = 8000

class EasyMsgRequest(asyncore.dispatcher):

    def __init__(self, host, obj, port=PORT):
        asyncore.dispatcher.__init__(self)
        self.obj = obj
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connect((host, port))

    def handle_connect(self):
        self.send(pickle.dumps(self.obj)+'\r\n')
        self.close()

    def handle_expt(self):
        self.close()

    def handle_close(self):
        self.close()


class EasyMsgChannel(asynchat.async_chat):

    def __init__(self, server, sock, addr):
        asynchat.async_chat.__init__(self, sock)
        self.set_terminator("\r\n")
        self.data = ""
        self.server = server

    def collect_incoming_data(self, data):
        self.data = self.data + data

    def found_terminator(self):
        obj = pickle.loads(self.data)
        self.route(obj)

    def route(self, obj):
        print "dont know how to route %s" % str(obj)


class EasyMsgServer(asyncore.dispatcher):

    channel_class = EasyMsgChannel

    def __init__(self, port=PORT):
        asyncore.dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.bind(("", port))
        self.listen(5)

    def handle_accept(self):
        con, addr = self.accept()
        self.create_channel(con, addr)
    
    def create_channel(self, con, addr):
        self.channel_class(self, con, addr)


client.py ------------------------------------------------------------------
import asyncore
from easymsg import EasyMsgRequest, PORT
from code import InteractiveConsole

def sender(host, port=PORT):
    def send(obj):
        req = EasyMsgRequest(host, obj, port=port)
        asyncore.loop()
    return send

if __name__ == '__main__':
    import sys
    import readline

    if len(sys.argv) < 2:
        print "usage: client.py  []"
        sys.exit(1)

    try:
        port = int(sys.argv[2])
    except (IndexError, ValueError):
        port = PORT

    v = globals().copy()
    v['s'] = v['send'] = sender(sys.argv[1], port)
    console = InteractiveConsole(v)
    console.interact()

server.py ------------------------------------------------------------------
import asyncore
from easymsg import EasyMsgServer, PORT

if __name__ == '__main__':
    import sys
    try:
        port = int(sys.argv[1])
    except (IndexError, ValueError):
        port = PORT
    s = EasyMsgServer(port)
    print "serving..."
    asyncore.loop()

forwarder.py ------------------------------------------------------------------
from easymsg import EasyMsgServer, EasyMsgChannel, EasyMsgRequest, PORT

class ForwarderChannel(EasyMsgChannel):
    def route(self, obj):
        server = self.server
        print "forwarding %s" % str(obj)
        EasyMsgRequest(server.dest_host, obj, port=server.dest_port)


class ForwarderServer(EasyMsgServer):
    
    channel_class = ForwarderChannel

    def __init__(self, dest_host, dest_port=PORT, port=PORT):
        self.dest_host = dest_host
        self.dest_port = dest_port
        EasyMsgServer.__init__(self, port)


if __name__ == '__main__':
    import sys
    import asyncore

    try:
        dest_host = sys.argv[1]
        dest_port = int(sys.argv[2])
    except (IndexError, ValueError):
        print "usage: forwarder.py   []"
        sys.exit(1)
    try:
        port = int(sys.argv[3])
    except (IndexError, ValueError):
        port = PORT
    s = ForwarderServer(dest_host, dest_port, port)
    print "serving..."
    asyncore.loop()