Add show source feature to gtk.

Trundle · Trundle · commit f2e931ab3bd6 · 2010-04-15T00:34:34.000+02:00
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -825,19 +825,14 @@ def p_key(self, key):
             return ''
 
         elif key in key_dispatch[config.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, IOError, NameError, TypeError):
-                self.statusbar.message("Cannot show source.")
-                return ''
-            else:
+            source = self.get_source_of_current_name()
+            if source is not None:
                 if config.highlight_show_source:
                     source = format(PythonLexer().get_tokens(source),
                                     TerminalFormatter())
                 page(source)
+            else:
+                self.statusbar.message('Cannot show source.')
             return ''
 
         elif key == '\n':
diff --git a/bpython/gtk_.py b/bpython/gtk_.py
@@ -50,6 +50,18 @@
 _COLORS = dict(b='blue', c='cyan', g='green', m='magenta', r='red',
                w='white', y='yellow', k='black', d='black')
 
+def add_tags_to_buffer(color_scheme, text_buffer):
+    tags = dict()
+    for (name, value) in color_scheme.iteritems():
+        tag = tags[name] = text_buffer.create_tag(name)
+        for (char, prop) in zip(value, ['foreground', 'background']):
+            if char.lower() == 'd':
+                continue
+            tag.set_property(prop, _COLORS[char.lower()])
+            if char.isupper():
+                tag.set_property('weight', pango.WEIGHT_BOLD)
+    return tags
+
 class ArgspecFormatter(object):
     """
     Format an argspec using Pango markup language.
@@ -335,15 +347,7 @@ def __init__(self, interpreter, config):
 
         self.text_buffer = self.get_buffer()
         self.interact = GTKInteraction(self.config, Statusbar())
-        tags = dict()
-        for (name, value) in self.config.color_gtk_scheme.iteritems():
-            tag = tags[name] = self.text_buffer.create_tag(name)
-            for (char, prop) in zip(value, ['foreground', 'background']):
-                if char.lower() == 'd':
-                    continue
-                tag.set_property(prop, _COLORS[char.lower()])
-                if char.isupper():
-                    tag.set_property('weight', pango.WEIGHT_BOLD)
+        tags = add_tags_to_buffer(self.config.color_gtk_scheme, self.text_buffer)
         tags['prompt'].set_property('editable', False)
 
         self.text_buffer.connect('delete-range', self.on_buf_delete_range)
@@ -467,7 +471,29 @@ def do_key_press_event(self, event):
                                gtk.gdk.MOD4_MASK |
                                gtk.gdk.SHIFT_MASK)
         if not state:
-            if event.keyval == gtk.keysyms.Return:
+            if event.keyval == gtk.keysyms.F2:
+                source = self.get_source_of_current_name()
+                if source is not None:
+                    win = gtk.Window()
+                    sw = gtk.ScrolledWindow()
+                    view = gtk.TextView()
+                    buffer = view.get_buffer()
+                    add_tags_to_buffer(self.config.color_gtk_scheme, buffer)
+                    from pygments.lexers import PythonLexer
+                    tokens = PythonLexer().get_tokens(source)
+                    for (token, value) in tokens:
+                        while token not in theme_map:
+                            token = token.parent
+                        iter_ = buffer.get_end_iter()
+                        buffer.insert_with_tags_by_name(iter_, value,
+                                                        theme_map[token])
+                    sw.add(view)
+                    win.add(sw)
+                    win.show_all()
+                else:
+                    # XXX Error message
+                    pass
+            elif event.keyval == gtk.keysyms.Return:
                 if self.list_win_visible:
                     self.list_win_visible = False
                     self.list_win.hide()
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -488,6 +488,22 @@ def get_args(self):
             return True
         return False
 
+    def get_source_of_current_name(self):
+        """Return the source code of the object which is bound to the
+        current name in the current input line. Return `None` if the
+        source cannot be found."""
+        try:
+            obj = self.current_func
+            if obj is None:
+                line = self.current_line()
+                if inspection.is_eval_safe_name(line):
+                    obj = self.get_object(line)
+            source = inspect.getsource(obj)
+        except (AttributeError, IOError, NameError, TypeError):
+            return None
+        else:
+            return source
+
     def complete(self, tab=False):
         """Construct a full list of possible completions and construct and
         display them in a window. Also check if there's an available argspec
