From c31a177ebe66ca28a4c5d0e5f385134344ea73c0 Mon Sep 17 00:00:00 2001 From: David Wilson Date: Sat, 24 Mar 2018 15:43:57 +0545 Subject: [PATCH] issue #155: parent: split get_main_kwargs() from get_preamble() --- mitogen/parent.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/mitogen/parent.py b/mitogen/parent.py index 29d7e16e..3b905977 100644 --- a/mitogen/parent.py +++ b/mitogen/parent.py @@ -385,12 +385,10 @@ class Stream(mitogen.core.Stream): 'exec(_(_("%s".encode(),"base64"),"zip"))' % (encoded,) ] - def get_preamble(self): + def get_main_kwargs(self): parent_ids = mitogen.parent_ids[:] parent_ids.insert(0, mitogen.context_id) - - source = inspect.getsource(mitogen.core) - source += '\nExternalContext().main(**%r)\n' % ({ + return { 'parent_ids': parent_ids, 'context_id': self.remote_id, 'debug': self.debug, @@ -398,8 +396,13 @@ class Stream(mitogen.core.Stream): 'log_level': get_log_level(), 'whitelist': self._router.get_module_whitelist(), 'blacklist': self._router.get_module_blacklist(), - },) + } + def get_preamble(self): + source = inspect.getsource(mitogen.core) + source += '\nExternalContext().main(**%r)\n' % ( + self.get_main_kwargs(), + ) return zlib.compress(minimize_source(source), 9) create_child = staticmethod(create_child)