Add show source feature.

Trundle · Trundle · commit 4850400b764d · 2009-09-18T16:27:51.000+02:00
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -55,6 +55,7 @@
 
 # These are used for syntax hilighting.
 from pygments import format
+from pygments.formatters import TerminalFormatter
 from pygments.lexers import PythonLexer
 from pygments.token import Token
 from bpython.formatter import BPythonFormatter, Parenthesis
@@ -422,6 +423,7 @@ def __init__(self, scr, interp, statusbar=None, idle=None):
         self.matches = []
         self.matches_iter = MatchesIterator()
         self.argspec = None
+        self.current_func = None
         self.s = ''
         self.inside_string = False
         self.highlighted_paren = None
@@ -523,6 +525,12 @@ def cw(self):
             i += 1
         return self.s[-i +1:]
 
+    def get_object(self, name):
+        if name in self.interp.locals:
+            return self.interp.locals[name]
+        else:
+            return eval(name, self.interp.locals)
+
     def get_args(self):
         """Check if an unclosed parenthesis exists, then attempt to get the
         argspec() for it. On success, update self.argspec and return True,
@@ -533,7 +541,8 @@ def get_args(self):
         if not OPTS.arg_spec:
             return False
 
-        # Find the name of the current function
+        # Get the name of the current function and where we are in
+        # the arguments
         stack = [['', 0, '']]
         try:
             for (token, value) in PythonLexer().get_tokens(self.s):
@@ -560,19 +569,14 @@ def get_args(self):
             func, _, _ = stack.pop()
         except IndexError:
             return False
+        if not func:
+            return False
 
-        # We found a name, now get a function object
         try:
-            if func in self.interp.locals:
-                f = self.interp.locals[func]
-        except TypeError:
-            return None
-        else:
-            try:
-                f = eval(func, self.interp.locals)
-            except Exception:
-                # Same deal with the exceptions :(
-                return None
+            f = self.get_object(func)
+        except (AttributeError, NameError):
+            return False
+
         if inspect.isclass(f):
             try:
                 f = f.__init__
@@ -1565,6 +1569,22 @@ def p_key(self, key):
             page(self.stdout_hist[self.prev_block_finished:-4])
             return ''
 
+        elif key in key_dispatch[OPTS.show_source_key]:
+            try:
+                obj = self.current_func
+                if obj is None and inspection.is_eval_safe_name(self.s):
+                    obj = self.get_object(self.s)
+                source = inspect.getsource(obj)
+            except (AttributeError, NameError, TypeError):
+                self.statusbar.message("Cannot show source.")
+                return ''
+            else:
+                if OPTS.highlight_show_source:
+                    source = format(PythonLexer().get_tokens(source),
+                                    TerminalFormatter())
+                page(source)
+            return ''
+
         elif key == '\n':
             self.lf()
             return None
diff --git a/bpython/config.py b/bpython/config.py
@@ -36,6 +36,7 @@ def loadini(struct, configfile):
             'auto_display_list': True,
             'color_scheme': 'default',
             'flush_output': True,
+            'highlight_show_source': True,
             'hist_file': '~/.pythonhist',
             'hist_length': 100,
             'paste_time': 0.02,
@@ -53,6 +54,7 @@ def loadini(struct, configfile):
             'last_output': 'F9',
             'pastebin': 'F8',
             'save': 'C-s',
+            'show_source': 'F2',
             'undo': 'C-r',
             'up_one_line': 'C-p',
             'yank_from_buffer': 'C-y'
@@ -66,11 +68,14 @@ def loadini(struct, configfile):
     struct.syntax = config.getboolean('general', 'syntax')
     struct.arg_spec = config.getboolean('general', 'arg_spec')
     struct.paste_time = config.getfloat('general', 'paste_time')
+    struct.highlight_show_source = config.getboolean('general',
+                                                     'highlight_show_source')
     struct.hist_file = config.get('general', 'hist_file')
     struct.hist_length = config.getint('general', 'hist_length')
     struct.flush_output = config.getboolean('general', 'flush_output')
     struct.pastebin_key = config.get('keyboard', 'pastebin')
     struct.save_key = config.get('keyboard', 'save')
+    struct.show_source_key = config.get('keyboard', 'show_source')
     struct.undo_key = config.get('keyboard', 'undo')
     struct.up_one_line_key = config.get('keyboard', 'up_one_line')
     struct.down_one_line_key = config.get('keyboard', 'down_one_line')
diff --git a/bpython/inspection.py b/bpython/inspection.py
@@ -34,6 +34,9 @@
 
 py3 = sys.version_info[0] == 3
 
+if not py3:
+    _name = re.compile(r'[a-zA-Z_]\w*')
+
 
 class AttrCleaner(object):
     """A context manager that tries to make an object not exhibit side-effects
@@ -43,7 +46,7 @@ def __init__(self, obj):
 
     def __enter__(self):
         """Try to make an object not exhibit side-effects on attribute
-        lookup.""" 
+        lookup."""
         type_ = type(self.obj)
         __getattribute__ = None
         __getattr__ = None
@@ -198,7 +201,7 @@ def getargspec(func, f):
             argspec = inspect.getfullargspec(f)
         else:
             argspec = inspect.getargspec(f)
-                 
+
         argspec = list(argspec)
         fixlongargs(f, argspec)
         argspec = [func, argspec, is_bound_method]
@@ -211,3 +214,10 @@ def getargspec(func, f):
             argspec[1][0].insert(0, 'obj')
         argspec.append(is_bound_method)
     return argspec
+
+
+def is_eval_safe_name(string):
+    if py3:
+        return all(part.isidentifier() for part in string.split('.'))
+    else:
+        return all(_name.match(part) for part in string.split('.'))
