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

core: fix NameError on disconnect

This commit is contained in:
David Wilson 2018-03-26 11:52:12 +05:45
parent 198bec3320
commit 0f08783330
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mitogen/core.py
View File

@ -863,7 +863,7 @@ class Context(object):
def __reduce__(self): def __reduce__(self):
return _unpickle_context, (self.context_id, self.name) return _unpickle_context, (self.context_id, self.name)
def on_disconnect(self, broker): def on_disconnect(self):
_v and LOG.debug('%r.on_disconnect()', self) _v and LOG.debug('%r.on_disconnect()', self)
fire(self, 'disconnect') fire(self, 'disconnect')
@ -1141,7 +1141,7 @@ class Router(object):
stream_ = self._stream_by_id.get(context.context_id) stream_ = self._stream_by_id.get(context.context_id)
if stream_ is stream: if stream_ is stream:
del self._stream_by_id[context.context_id] del self._stream_by_id[context.context_id]
context.on_disconnect(broker) context.on_disconnect()
def on_broker_shutdown(self): def on_broker_shutdown(self):
for context in self._context_by_id.itervalues(): for context in self._context_by_id.itervalues():
@ -1487,7 +1487,6 @@ class ExternalContext(object):
# Reopen with line buffering. # Reopen with line buffering.
sys.stdout = os.fdopen(1, 'w', 1) sys.stdout = os.fdopen(1, 'w', 1)
def _dispatch_one(self, msg): def _dispatch_one(self, msg):
data = msg.unpickle(throw=False) data = msg.unpickle(throw=False)
_v and LOG.debug('_dispatch_calls(%r)', data) _v and LOG.debug('_dispatch_calls(%r)', data)