From fe0373d1b63e21a2f1d04339b30f535e6c969d87 Mon Sep 17 00:00:00 2001 From: Nicolas Schodet Date: Wed, 30 Mar 2011 22:00:52 +0200 Subject: host/mex: accept a string to Node.register, refs #157 --- host/mex/node.py | 2 ++ host/mex/test/test.py | 3 +-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/host/mex/node.py b/host/mex/node.py index a1f226cd..50e77709 100644 --- a/host/mex/node.py +++ b/host/mex/node.py @@ -119,6 +119,8 @@ class Node: def register (self, mtype, handler): """Register an handler for the given message type.""" + if isinstance (mtype, basestring): + mtype = self.reserve (mtype) assert mtype not in self.__handlers self.__handlers[mtype] = handler diff --git a/host/mex/test/test.py b/host/mex/test/test.py index 91e49ee0..dec53b1d 100644 --- a/host/mex/test/test.py +++ b/host/mex/test/test.py @@ -35,7 +35,6 @@ h = Hub (min_clients = 2, log = log) def c1 (): n = Node () - mtype_oucouc = n.reserve ('oucouc') mtype_coucou = n.reserve ('coucou') def a (msg): print 'oucouc' @@ -44,7 +43,7 @@ def c1 (): m = Msg (msg.mtype) m.push ('B', nb) n.response (m) - n.register (mtype_oucouc, a) + n.register ('oucouc', a) def b (): assert False eb = n.schedule (31, b) -- cgit v1.2.3