ActorHandler helper module

actors.py

"""    
ActorHandler - class-based dispatch for actors    
    
Usage:    
    
from thespian.actors import Actor    
    
handler = ActorHandler()    
    
class Hello(Actor):    
    
    @handler.handle(SomeMessageClass)    
    def do_someclass(self, message, sender):    
        pass    
    
    receiveMessage = handler.receiveMessage()    
    
"""    
    
REGISTRY = {}    
    
    
class ActorHandler(object):    
    
    def __init__(self):    
        REGISTRY[self] = {}    
        pass    
    
    
    def handle(self, klass):    
        def klass_decorator(func):    
            REGISTRY[self][klass] = func    
            return func    
        return klass_decorator    
    
    
    def receiveMessage(self):    
        registry = REGISTRY[self]    
    
        def receiveMessage(self, message, sender):    
            k = message.__class__    
            if not k in registry:    
                 for klass in registry:    
                    if isinstance(message, klass):    
                        registry[message.__class__] = registry[klass]    
                        break    
            method = registry.get(klass, None)    
            if method is not None:    
                method(self, message, sender)    
            else:    
                self.unhandledMessage(message, sender)    
    
        return receiveMessage