Rooba
/
mitogen
mirror of https://github.com/mitogen-hq/mitogen.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

core: Restore behaviour of unpickling Router-specific Context subclass

This commit is contained in:
David Wilson 2018-02-14 19:45:46 +05:45
parent dd088908df
commit 9238e09ae8
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mitogen/core.py
View File

@ -794,7 +794,7 @@ def _unpickle_context(router, context_id, name):
assert isinstance(router, Router) assert isinstance(router, Router)
assert isinstance(context_id, (int, long)) and context_id > 0 assert isinstance(context_id, (int, long)) and context_id > 0
assert isinstance(name, basestring) and len(name) < 100 assert isinstance(name, basestring) and len(name) < 100
return Context(router, context_id, name) return router.context_class(router, context_id, name)
class Waker(BasicStream): class Waker(BasicStream):
@ -883,6 +883,8 @@ class IoLogger(BasicStream):
class Router(object): class Router(object):
context_class = Context
def __init__(self, broker): def __init__(self, broker):
self.broker = broker self.broker = broker
listen(broker, 'shutdown', self.on_broker_shutdown) listen(broker, 'shutdown', self.on_broker_shutdown)