From 92a216e798f83d1c3e4cd318bc8e61876980c4d7 2016-04-13 09:01:35
From: Min RK <benjaminrk@gmail.com>
Date: 2016-04-13 09:01:35
Subject: [PATCH] handle failure to get module for globals

I ran into this learning alembic,
which creates dummy modules and apparently doesn't put them in sys.modules.

---

diff --git a/IPython/terminal/embed.py b/IPython/terminal/embed.py
index f78aa2a..aa43ca4 100644
--- a/IPython/terminal/embed.py
+++ b/IPython/terminal/embed.py
@@ -197,7 +197,14 @@ class InteractiveShellEmbed(TerminalInteractiveShell):
                 local_ns = call_frame.f_locals
             if module is None:
                 global_ns = call_frame.f_globals
-                module = sys.modules[global_ns['__name__']]
+                try:
+                    module = sys.modules[global_ns['__name__']]
+                except KeyError:
+                    warnings.warn("Failed to get module %s" % \
+                        global_ns.get('__name__', 'unknown module')
+                    )
+                    module = DummyMod()
+                    module.__dict__ = global_ns
             if compile_flags is None:
                 compile_flags = (call_frame.f_code.co_flags &
                                  compilerop.PyCF_MASK)