Added REQUIRE capabilities for Routing

This commit is contained in:
PhilipTrauner 2016-03-08 08:43:52 +01:00
parent 14c28e4fd8
commit befca7cdf8

View file

@ -1,3 +1,11 @@
BROADCAST = 0
ROUTE = 1
LAST_STOP = 2
class InvalidRouteSetup(AttributeError):
def __init__(self, msg):
super(AttributeError, self).__init__(msg)
class ServerRoute:
def __init__(self, **kwargs):
self.setup(**kwargs)
@ -12,9 +20,24 @@ class ClientRoute:
def run(self, data, handler):
pass
def create_routes(routes):
def create_routes(routes, handler):
routes = routes.copy()
reverse_routes = {}
for prefix in routes:
if type(routes[prefix]) is tuple or type(routes[prefix]) is list:
routes[prefix] = routes[prefix][0](**routes[prefix][1])
reverse_routes[routes[prefix]] = prefix
for prefix in routes:
attrs = dir(routes[prefix])
if "REQUIRED" in attrs:
if not "start" in attrs:
raise InvalidRouteSetup("method named 'start' required if 'REQUIRE' is defined")
for required in type(routes[prefix]).REQUIRED:
if required == BROADCAST:
routes[prefix].broadcast = handler.broadcast
elif required == ROUTE:
routes[prefix].route = reverse_routes[routes[prefix]]
elif required == LAST_STOP:
routes[prefix].last_stop = handler.last_stop
routes[prefix].start()
return routes