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

Allow importing master's __main__.

This commit is contained in:
David Wilson 2016-08-13 23:20:25 +01:00
parent a174c32d22
commit 052c96d7ee
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
econtext/__init__.py
View File

@ -1 +1,2 @@
slave = False
from econtext.core import * # NOQA from econtext.core import * # NOQA

6
econtext/core.py
View File

@ -681,7 +681,9 @@ class ExternalContext(object):
econtext = imp.new_module('econtext') econtext = imp.new_module('econtext')
econtext.__package__ = 'econtext' econtext.__package__ = 'econtext'
econtext.__path__ = [] econtext.__path__ = []
econtext.slave = True
econtext.core = sys.modules['__main__'] econtext.core = sys.modules['__main__']
del sys.modules['__main__']
sys.modules['econtext'] = econtext sys.modules['econtext'] = econtext
sys.modules['econtext.core'] = econtext.core sys.modules['econtext.core'] = econtext.core
@ -692,13 +694,13 @@ class ExternalContext(object):
klass.__module__ = 'econtext.core' klass.__module__ = 'econtext.core'
def _setup_master(self, key): def _setup_master(self, key):
os.wait() # Reap first stage.
self.broker = Broker() self.broker = Broker()
self.context = Context(self.broker, 'master', key=key) self.context = Context(self.broker, 'master', key=key)
self.channel = Channel(self.context, CALL_FUNCTION) self.channel = Channel(self.context, CALL_FUNCTION)
self.context.stream = Stream(self.context) self.context.stream = Stream(self.context)
self.context.stream.accept(100, 1) self.context.stream.accept(100, 1)
os.wait() # Reap first stage.
os.close(100) os.close(100)
def _setup_logging(self, log_level): def _setup_logging(self, log_level):