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

issue #477: polyfill partition() use in mitogen.parent.

This commit is contained in:
David Wilson 2019-01-23 12:44:08 +00:00
parent dd86a157ff
commit 87d2af4c6e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mitogen/parent.py
View File

@ -60,6 +60,7 @@ except ImportError:
import mitogen.core import mitogen.core
from mitogen.core import b from mitogen.core import b
from mitogen.core import bytes_partition
from mitogen.core import LOG from mitogen.core import LOG
from mitogen.core import IOLOG from mitogen.core import IOLOG
@ -1785,7 +1786,7 @@ class RouteMonitor(object):
if msg.is_dead: if msg.is_dead:
return return
target_id_s, _, target_name = msg.data.partition(b(':')) target_id_s, _, target_name = bytes_partition(msg.data, b(':'))
target_name = target_name.decode() target_name = target_name.decode()
target_id = int(target_id_s) target_id = int(target_id_s)
self.router.context_by_id(target_id).name = target_name self.router.context_by_id(target_id).name = target_name
@ -2066,7 +2067,7 @@ class ModuleForwarder(object):
if msg.is_dead: if msg.is_dead:
return return
context_id_s, _, fullname = msg.data.partition(b('\x00')) context_id_s, _, fullname = bytes_partition(msg.data, b('\x00'))
fullname = mitogen.core.to_text(fullname) fullname = mitogen.core.to_text(fullname)
context_id = int(context_id_s) context_id = int(context_id_s)
stream = self.router.stream_by_id(context_id) stream = self.router.stream_by_id(context_id)