Rationalize _fixup_package(), fix _setup_master() name.

econtext/core.py

@@ -679,14 +679,17 @@ class Broker(object):
 
 
 class ExternalContext(object):
-    def _fixup_main_module(self):
+    def _fixup_package(self):
-        main = sys.modules['__main__']
+        econtext = imp.new_module('econtext')
-        main.__path__ = []
+        econtext.__package__ = 'econtext'
-        main.core = main
+        econtext.__path__ = []
-
+        econtext.core = sys.modules['__main__']
-        sys.modules['econtext'] = main
+
-        sys.modules['econtext.core'] = main
+        sys.modules['econtext'] = econtext
-        for klass in globals().itervalues():
+        sys.modules['econtext.core'] = econtext.core
+        exec 'from econtext.core import *' in vars(econtext)
+
+        for klass in vars(econtext.core).itervalues():
             if hasattr(klass, '__module__'):
                 klass.__module__ = 'econtext.core'
 
@@ -695,7 +698,7 @@ class ExternalContext(object):
         os.dup2(100, 0)
         os.close(100)
 
-    def _SetupMaster(self, key):
+    def _setup_master(self, key):
         self.broker = Broker()
         self.context = Context(self.broker, 'master', key=key)
         self.channel = Channel(self.context, CALL_FUNCTION)
@@ -743,8 +746,8 @@ class ExternalContext(object):
 
     def main(self, key, log_level):
         self._reap_first_stage()
-        self._fixup_main_module()
+        self._fixup_package()
-        self._SetupMaster(key)
+        self._setup_master(key)
         try:
             self._setup_logging(log_level)
             self._setup_importer()

econtext/master.py

@@ -245,7 +245,7 @@ class Broker(econtext.core.Broker):
         context.stream.connect()
         return self.register(context)
 
-    def get_remote(self, hostname, username, name=None, python_path=None):
+    def get_remote(self, hostname, username=None, name=None, python_path=None):
         """Get the named remote context, creating it if it does not exist."""
         if name is None:
             name = hostname