adding CrazyLocals for importing stuff automatically

eliasdorneles · eliasdorneles · commit 1b4b2bf51b62 · 2016-09-28T23:28:20.000-03:00
So, this is a really bad idea.

This makes names from sys.modules magically available without
needing to import them.

It does that by replacing the locals dict by a dict subclass that looks
up names in sys.modules when they're not present in current locals.

Please do NOT merge!

But have fun. :)
diff --git a/bpython/curtsiesfrontend/interpreter.py b/bpython/curtsiesfrontend/interpreter.py
@@ -61,6 +61,21 @@ def format(self, tokensource, outfile):
         outfile.write(parse(o.rstrip()))
 
 
+class CrazyLocals(dict):
+    def __getitem__(self, name):
+        try:
+            return super(CrazyLocals, self).__getitem__(name)
+        except KeyError:
+            import sys
+            try:
+                return sys.modules[name]
+            except KeyError:
+                for mod in sys.modules.values():
+                    if hasattr(mod, name):
+                        return getattr(mod, name)
+                raise
+
+
 class Interp(ReplInterpreter):
     def __init__(self, locals=None, encoding=None):
         """Constructor.
@@ -75,6 +90,9 @@ def __init__(self, locals=None, encoding=None):
         """
         if locals is None:
             locals = {"__name__": "__console__", "__doc__": None}
+
+        locals = CrazyLocals(locals)
+
         if encoding is None:
             encoding = getpreferredencoding()
         ReplInterpreter.__init__(self, locals, encoding)
