home *** CD-ROM | disk | FTP | other *** search
# -*- coding: utf-8 -*- # # pythonconsole.py # # Copyright (C) 2006 - Steve Frécinaux # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2, or (at your option) # any later version. # # The Rhythmbox authors hereby grant permission for non-GPL compatible # GStreamer plugins to be used and distributed together with GStreamer # and Rhythmbox. This permission is above and beyond the permissions granted # by the GPL license by which Rhythmbox is covered. If you modify this code # you may extend this exception to your version of the code, but you are not # obligated to do so. If you do not wish to do so, delete this exception # statement from your version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. # Parts from "Interactive Python-GTK Console" (stolen from epiphany's console.py) # Copyright (C), 1998 James Henstridge <james@daa.com.au> # Copyright (C), 2005 Adam Hooper <adamh@densi.com> # Bits from gedit Python Console Plugin # Copyrignt (C), 2005 Raphaël Slinckx import string import sys import re import traceback import gobject import gtk import pango import gconf import rhythmdb, rb try: import rpdb2 have_rpdb2 = True except: have_rpdb2 = False ui_str = """ <ui> <menubar name="MenuBar"> <menu name="ToolsMenu" action="Tools"> <placeholder name="ToolsOps_5"> <menuitem name="PythonConsole" action="PythonConsole"/> <menuitem name="PythonDebugger" action="PythonDebugger"/> </placeholder> </menu> </menubar> </ui> """ class PythonConsolePlugin(rb.Plugin): def __init__(self): rb.Plugin.__init__(self) self.window = None def activate(self, shell): data = dict() manager = shell.get_player().get_property('ui-manager') data['action_group'] = gtk.ActionGroup('PythonConsolePluginActions') action = gtk.Action('PythonConsole', _('_Python Console'), _("Show Rhythmbox's python console"), 'gnome-mime-text-x-python') action.connect('activate', self.show_console, shell) data['action_group'].add_action(action) action = gtk.Action('PythonDebugger', _('Python Debugger'), _("Enable remote python debugging with rpdb2"), None) if have_rpdb2: action.connect('activate', self.enable_debugging, shell) else: action.set_visible(False) data['action_group'].add_action(action) manager.insert_action_group(data['action_group'], 0) data['ui_id'] = manager.add_ui_from_string(ui_str) manager.ensure_update() shell.set_data('PythonConsolePluginInfo', data) def deactivate(self, shell): data = shell.get_data('PythonConsolePluginInfo') manager = shell.get_player().get_property('ui-manager') manager.remove_ui(data['ui_id']) manager.remove_action_group(data['action_group']) manager.ensure_update() shell.set_data('PythonConsolePluginInfo', None) if self.window is not None: self.window.destroy() def show_console(self, action, shell): if not self.window: ns = {'__builtins__' : __builtins__, 'rb' : rb, 'rhythmdb' : rhythmdb, 'shell' : shell} console = PythonConsole(namespace = ns, destroy_cb = self.destroy_console) console.set_size_request(600, 400) console.eval('print "' + \ _('You can access the main window ' \ 'through the \'shell\' variable :') + '\\n%s" % shell', False) self.window = gtk.Window() self.window.set_title('Rhythmbox Python Console') self.window.add(console) self.window.connect('destroy', self.destroy_console) self.window.show_all() else: self.window.show_all() self.window.grab_focus() def enable_debugging(self, action, shell): msg = _("After you press OK, Rhythmbox will wait until you connect to it with winpdb or rpdb2. If you have not set a debugger password in GConf, it will use the default password ('rhythmbox').") dialog = gtk.MessageDialog(None, 0, gtk.MESSAGE_INFO, gtk.BUTTONS_OK_CANCEL, msg) if dialog.run() == gtk.RESPONSE_OK: gconfclient = gconf.client_get_default() password = gconfclient.get_string('/apps/rhythmbox/plugins/pythonconsole/rpdb2_password') or "rhythmbox" def start_debugger(password): rpdb2.start_embedded_debugger(password) return False gobject.idle_add(start_debugger, password) dialog.destroy() def destroy_console(self, *args): self.window.destroy() self.window = None class PythonConsole(gtk.ScrolledWindow): def __init__(self, namespace = {}, destroy_cb = None): gtk.ScrolledWindow.__init__(self) self.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC); self.set_shadow_type(gtk.SHADOW_IN) self.view = gtk.TextView() self.view.modify_font(pango.FontDescription('Monospace')) self.view.set_editable(True) self.view.set_wrap_mode(gtk.WRAP_WORD_CHAR) self.add(self.view) self.view.show() buffer = self.view.get_buffer() self.normal = buffer.create_tag("normal") self.error = buffer.create_tag("error") self.error.set_property("foreground", "red") self.command = buffer.create_tag("command") self.command.set_property("foreground", "blue") self.__spaces_pattern = re.compile(r'^\s+') self.namespace = namespace self.destroy_cb = destroy_cb self.block_command = False # Init first line buffer.create_mark("input-line", buffer.get_end_iter(), True) buffer.insert(buffer.get_end_iter(), ">>> ") buffer.create_mark("input", buffer.get_end_iter(), True) # Init history self.history = [''] self.history_pos = 0 self.current_command = '' self.namespace['__history__'] = self.history # Set up hooks for standard output. self.stdout = gtkoutfile(self, sys.stdout.fileno(), self.normal) self.stderr = gtkoutfile(self, sys.stderr.fileno(), self.error) # Signals self.view.connect("key-press-event", self.__key_press_event_cb) buffer.connect("mark-set", self.__mark_set_cb) def __key_press_event_cb(self, view, event): if event.keyval == gtk.keysyms.d and \ event.state == gtk.gdk.CONTROL_MASK: self.destroy() elif event.keyval == gtk.keysyms.Return and \ event.state == gtk.gdk.CONTROL_MASK: # Get the command buffer = view.get_buffer() inp_mark = buffer.get_mark("input") inp = buffer.get_iter_at_mark(inp_mark) cur = buffer.get_end_iter() line = buffer.get_text(inp, cur) self.current_command = self.current_command + line + "\n" self.history_add(line) # Prepare the new line cur = buffer.get_end_iter() buffer.insert(cur, "\n... ") cur = buffer.get_end_iter() buffer.move_mark(inp_mark, cur) # Keep indentation of precendent line spaces = re.match(self.__spaces_pattern, line) if spaces is not None: buffer.insert(cur, line[spaces.start() : spaces.end()]) cur = buffer.get_end_iter() buffer.place_cursor(cur) gobject.idle_add(self.scroll_to_end) return True elif event.keyval == gtk.keysyms.Return: # Get the marks buffer = view.get_buffer() lin_mark = buffer.get_mark("input-line") inp_mark = buffer.get_mark("input") # Get the command line inp = buffer.get_iter_at_mark(inp_mark) cur = buffer.get_end_iter() line = buffer.get_text(inp, cur) self.current_command = self.current_command + line + "\n" self.history_add(line) # Make the line blue lin = buffer.get_iter_at_mark(lin_mark) buffer.apply_tag(self.command, lin, cur) buffer.insert(cur, "\n") cur_strip = self.current_command.rstrip() if cur_strip.endswith(":") \ or (self.current_command[-2:] != "\n\n" and self.block_command): # Unfinished block command self.block_command = True com_mark = "... " elif cur_strip.endswith("\\"): com_mark = "... " else: # Eval the command self.__run(self.current_command) self.current_command = '' self.block_command = False com_mark = ">>> " # Prepare the new line cur = buffer.get_end_iter() buffer.move_mark(lin_mark, cur) buffer.insert(cur, com_mark) cur = buffer.get_end_iter() buffer.move_mark(inp_mark, cur) buffer.place_cursor(cur) gobject.idle_add(self.scroll_to_end) return True elif event.keyval == gtk.keysyms.KP_Down or \ event.keyval == gtk.keysyms.Down: # Next entry from history view.emit_stop_by_name("key_press_event") self.history_down() gobject.idle_add(self.scroll_to_end) return True elif event.keyval == gtk.keysyms.KP_Up or \ event.keyval == gtk.keysyms.Up: # Previous entry from history view.emit_stop_by_name("key_press_event") self.history_up() gobject.idle_add(self.scroll_to_end) return True elif event.keyval == gtk.keysyms.KP_Left or \ event.keyval == gtk.keysyms.Left or \ event.keyval == gtk.keysyms.BackSpace: buffer = view.get_buffer() inp = buffer.get_iter_at_mark(buffer.get_mark("input")) cur = buffer.get_iter_at_mark(buffer.get_insert()) return inp.compare(cur) == 0 elif event.keyval == gtk.keysyms.Home: # Go to the begin of the command instead of the begin of # the line buffer = view.get_buffer() inp = buffer.get_iter_at_mark(buffer.get_mark("input")) if event.state == gtk.gdk.SHIFT_MASK: buffer.move_mark_by_name("insert", inp) else: buffer.place_cursor(inp) return True def __mark_set_cb(self, buffer, iter, name): input = buffer.get_iter_at_mark(buffer.get_mark("input")) pos = buffer.get_iter_at_mark(buffer.get_insert()) self.view.set_editable(pos.compare(input) != -1) def get_command_line(self): buffer = self.view.get_buffer() inp = buffer.get_iter_at_mark(buffer.get_mark("input")) cur = buffer.get_end_iter() return buffer.get_text(inp, cur) def set_command_line(self, command): buffer = self.view.get_buffer() mark = buffer.get_mark("input") inp = buffer.get_iter_at_mark(mark) cur = buffer.get_end_iter() buffer.delete(inp, cur) buffer.insert(inp, command) buffer.select_range(buffer.get_iter_at_mark(mark), buffer.get_end_iter()) self.view.grab_focus() def history_add(self, line): if line.strip() != '': self.history_pos = len(self.history) self.history[self.history_pos - 1] = line self.history.append('') def history_up(self): if self.history_pos > 0: self.history[self.history_pos] = self.get_command_line() self.history_pos = self.history_pos - 1 self.set_command_line(self.history[self.history_pos]) def history_down(self): if self.history_pos < len(self.history) - 1: self.history[self.history_pos] = self.get_command_line() self.history_pos = self.history_pos + 1 self.set_command_line(self.history[self.history_pos]) def scroll_to_end(self): iter = self.view.get_buffer().get_end_iter() self.view.scroll_to_iter(iter, 0.0) return False def write(self, text, tag = None): buffer = self.view.get_buffer() if tag is None: buffer.insert(buffer.get_end_iter(), text) else: buffer.insert_with_tags(buffer.get_end_iter(), text, tag) gobject.idle_add(self.scroll_to_end) def eval(self, command, display_command = False): buffer = self.view.get_buffer() lin = buffer.get_mark("input-line") buffer.delete(buffer.get_iter_at_mark(lin), buffer.get_end_iter()) if isinstance(command, list) or isinstance(command, tuple): for c in command: if display_command: self.write(">>> " + c + "\n", self.command) self.__run(c) else: if display_command: self.write(">>> " + c + "\n", self.command) self.__run(command) cur = buffer.get_end_iter() buffer.move_mark_by_name("input-line", cur) buffer.insert(cur, ">>> ") cur = buffer.get_end_iter() buffer.move_mark_by_name("input", cur) self.view.scroll_to_iter(buffer.get_end_iter(), 0.0) def __run(self, command): sys.stdout, self.stdout = self.stdout, sys.stdout sys.stderr, self.stderr = self.stderr, sys.stderr try: try: r = eval(command, self.namespace, self.namespace) if r is not None: print `r` except SyntaxError: exec command in self.namespace except: if hasattr(sys, 'last_type') and sys.last_type == SystemExit: self.destroy() else: traceback.print_exc() sys.stdout, self.stdout = self.stdout, sys.stdout sys.stderr, self.stderr = self.stderr, sys.stderr def destroy(self): if self.destroy_cb is not None: self.destroy_cb() class gtkoutfile: """A fake output file object. It sends output to a TK test widget, and if asked for a file number, returns one set on instance creation""" def __init__(self, console, fn, tag): self.fn = fn self.console = console self.tag = tag def close(self): pass def flush(self): pass def fileno(self): return self.fn def isatty(self): return 0 def read(self, a): return '' def readline(self): return '' def readlines(self): return [] def write(self, s): self.console.write(s, self.tag) def writelines(self, l): self.console.write(l, self.tag) def seek(self, a): raise IOError, (29, 'Illegal seek') def tell(self): raise IOError, (29, 'Illegal seek') truncate = tell # ex:noet:ts=8: 
