Make module lookup a little more robust (certain kinds of fiddling to

sys.modules previously produced an exception).
skrah · Mar 28, 2003 · b38bbbd · b38bbbd
1 parent e9638cc
commit b38bbbd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Lib/inspect.py b/Lib/inspect.py
@@ -366,12 +366,12 @@ def getmodule(object):
     except TypeError:
         return None
     if file in modulesbyfile:
-        return sys.modules[modulesbyfile[file]]
+        return sys.modules.get(modulesbyfile[file])
     for module in sys.modules.values():
         if hasattr(module, '__file__'):
             modulesbyfile[getabsfile(module)] = module.__name__
     if file in modulesbyfile:
-        return sys.modules[modulesbyfile[file]]
+        return sys.modules.get(modulesbyfile[file])
     main = sys.modules['__main__']
     if hasattr(main, object.__name__):
         mainobject = getattr(main, object.__name__)