Merge pull request #159 from amirmiron/teleport-function-nonexisting-…

…module fixed failed function teleportation when matching module did not exist on the other side
tomerfiliba-org · Feb 25, 2015 · d85d4c5 · d85d4c5
2 parents 85d83e5 + 1839ef9
commit d85d4c5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rpyc/utils/teleportation.py b/rpyc/utils/teleportation.py
@@ -117,7 +117,10 @@ def _import_codetup(codetup):
 
 def import_function(functup):
     name, modname, defaults, codetup = functup
-    mod = __import__(modname, None, None, "*")
+    try:
+        mod = __import__(modname, None, None, "*")
+    except ImportError:
+        mod = __import__("__main__", None, None, "*")
     codeobj = _import_codetup(codetup)
     return FunctionType(codeobj, mod.__dict__, name, defaults)