issue #155: parent: split get_main_kwargs() from get_preamble()

pull/167/head
David Wilson 8 years ago
parent b51365209e
commit c31a177ebe

@ -385,12 +385,10 @@ class Stream(mitogen.core.Stream):
'exec(_(_("%s".encode(),"base64"),"zip"))' % (encoded,) 'exec(_(_("%s".encode(),"base64"),"zip"))' % (encoded,)
] ]
def get_preamble(self): def get_main_kwargs(self):
parent_ids = mitogen.parent_ids[:] parent_ids = mitogen.parent_ids[:]
parent_ids.insert(0, mitogen.context_id) parent_ids.insert(0, mitogen.context_id)
return {
source = inspect.getsource(mitogen.core)
source += '\nExternalContext().main(**%r)\n' % ({
'parent_ids': parent_ids, 'parent_ids': parent_ids,
'context_id': self.remote_id, 'context_id': self.remote_id,
'debug': self.debug, 'debug': self.debug,
@ -398,8 +396,13 @@ class Stream(mitogen.core.Stream):
'log_level': get_log_level(), 'log_level': get_log_level(),
'whitelist': self._router.get_module_whitelist(), 'whitelist': self._router.get_module_whitelist(),
'blacklist': self._router.get_module_blacklist(), '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) return zlib.compress(minimize_source(source), 9)
create_child = staticmethod(create_child) create_child = staticmethod(create_child)

Loading…
Cancel
Save