Mercurial > code > home > repos > light9
annotate lib/ipython_view.py @ 1425:8796803a1b45
show prog name in my i3 terminals
Ignore-this: f58b7487f90588fd8c82c894bd759bcc
| author | drewp@bigasterisk.com |
|---|---|
| date | Sat, 11 Jun 2016 04:30:40 +0000 |
| parents | 547d65ea9902 |
| children | 3c523c71da29 |
| rev | line source |
|---|---|
| 709 | 1 # this version is adapted from http://wiki.ipython.org/Old_Embedding/GTK |
| 2 | |
| 3 """ | |
| 4 Backend to the console plugin. | |
| 5 | |
| 6 @author: Eitan Isaacson | |
| 7 @organization: IBM Corporation | |
| 8 @copyright: Copyright (c) 2007 IBM Corporation | |
| 9 @license: BSD | |
| 10 | |
| 11 All rights reserved. This program and the accompanying materials are made | |
| 12 available under the terms of the BSD which accompanies this distribution, and | |
| 13 is available at U{http://www.opensource.org/licenses/bsd-license.php} | |
| 14 """ | |
| 15 # this file is a modified version of source code from the Accerciser project | |
| 16 # http://live.gnome.org/accerciser | |
| 17 | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
18 from gi.repository import Gtk |
|
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
19 from gi.repository import Gdk |
| 709 | 20 import re |
| 21 import sys | |
| 22 import os | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
23 from gi.repository import Pango |
| 709 | 24 from StringIO import StringIO |
| 25 | |
| 26 try: | |
| 27 import IPython | |
| 28 except Exception,e: | |
| 29 raise "Error importing IPython (%s)" % str(e) | |
| 30 | |
| 31 ansi_colors = {'0;30': 'Black', | |
| 32 '0;31': 'Red', | |
| 33 '0;32': 'Green', | |
| 34 '0;33': 'Brown', | |
| 35 '0;34': 'Blue', | |
| 36 '0;35': 'Purple', | |
| 37 '0;36': 'Cyan', | |
| 38 '0;37': 'LightGray', | |
| 39 '1;30': 'DarkGray', | |
| 40 '1;31': 'DarkRed', | |
| 41 '1;32': 'SeaGreen', | |
| 42 '1;33': 'Yellow', | |
| 43 '1;34': 'LightBlue', | |
| 44 '1;35': 'MediumPurple', | |
| 45 '1;36': 'LightCyan', | |
| 46 '1;37': 'White'} | |
| 47 | |
| 48 class IterableIPShell: | |
| 49 def __init__(self,argv=None,user_ns=None,user_global_ns=None, | |
| 50 cin=None, cout=None,cerr=None, input_func=None): | |
| 51 if input_func: | |
| 52 IPython.iplib.raw_input_original = input_func | |
| 53 if cin: | |
| 54 IPython.Shell.Term.cin = cin | |
| 55 if cout: | |
| 56 IPython.Shell.Term.cout = cout | |
| 57 if cerr: | |
| 58 IPython.Shell.Term.cerr = cerr | |
| 59 | |
| 60 if argv is None: | |
| 61 argv=[] | |
| 62 | |
| 63 # This is to get rid of the blockage that occurs during | |
| 64 # IPython.Shell.InteractiveShell.user_setup() | |
| 65 IPython.iplib.raw_input = lambda x: None | |
| 66 | |
| 67 self.term = IPython.genutils.IOTerm(cin=cin, cout=cout, cerr=cerr) | |
| 68 os.environ['TERM'] = 'dumb' | |
| 69 excepthook = sys.excepthook | |
| 70 self.IP = IPython.Shell.make_IPython(argv,user_ns=user_ns, | |
| 71 user_global_ns=user_global_ns, | |
| 72 embedded=True, | |
| 73 shell_class=IPython.Shell.InteractiveShell) | |
| 74 self.IP.system = lambda cmd: self.shell(self.IP.var_expand(cmd), | |
| 75 header='IPython system call: ', | |
| 76 verbose=self.IP.rc.system_verbose) | |
| 77 sys.excepthook = excepthook | |
| 78 self.iter_more = 0 | |
| 79 self.history_level = 0 | |
| 80 self.complete_sep = re.compile('[\s\{\}\[\]\(\)]') | |
| 81 | |
| 82 def execute(self): | |
| 83 self.history_level = 0 | |
| 84 orig_stdout = sys.stdout | |
| 85 sys.stdout = IPython.Shell.Term.cout | |
| 86 try: | |
| 87 line = self.IP.raw_input(None, self.iter_more) | |
| 88 if self.IP.autoindent: | |
| 89 self.IP.readline_startup_hook(None) | |
| 90 except KeyboardInterrupt: | |
| 91 self.IP.write('\nKeyboardInterrupt\n') | |
| 92 self.IP.resetbuffer() | |
| 93 # keep cache in sync with the prompt counter: | |
| 94 self.IP.outputcache.prompt_count -= 1 | |
| 95 | |
| 96 if self.IP.autoindent: | |
| 97 self.IP.indent_current_nsp = 0 | |
| 98 self.iter_more = 0 | |
| 99 except: | |
| 100 self.IP.showtraceback() | |
| 101 else: | |
| 102 self.iter_more = self.IP.push(line) | |
| 103 if (self.IP.SyntaxTB.last_syntax_error and | |
| 104 self.IP.rc.autoedit_syntax): | |
| 105 self.IP.edit_syntax_error() | |
| 106 if self.iter_more: | |
| 107 self.prompt = str(self.IP.outputcache.prompt2).strip() | |
| 108 if self.IP.autoindent: | |
| 109 self.IP.readline_startup_hook(self.IP.pre_readline) | |
| 110 else: | |
| 111 self.prompt = str(self.IP.outputcache.prompt1).strip() | |
| 112 sys.stdout = orig_stdout | |
| 113 | |
| 114 def historyBack(self): | |
| 115 self.history_level -= 1 | |
| 116 return self._getHistory() | |
| 117 | |
| 118 def historyForward(self): | |
| 119 self.history_level += 1 | |
| 120 return self._getHistory() | |
| 121 | |
| 122 def _getHistory(self): | |
| 123 try: | |
| 124 rv = self.IP.user_ns['In'][self.history_level].strip('\n') | |
| 125 except IndexError: | |
| 126 self.history_level = 0 | |
| 127 rv = '' | |
| 128 return rv | |
| 129 | |
| 130 def updateNamespace(self, ns_dict): | |
| 131 self.IP.user_ns.update(ns_dict) | |
| 132 | |
| 133 def complete(self, line): | |
| 134 split_line = self.complete_sep.split(line) | |
| 135 possibilities = self.IP.complete(split_line[-1]) | |
| 136 if possibilities: | |
| 137 common_prefix = reduce(self._commonPrefix, possibilities) | |
| 138 completed = line[:-len(split_line[-1])]+common_prefix | |
| 139 else: | |
| 140 completed = line | |
| 141 return completed, possibilities | |
| 142 | |
| 143 def _commonPrefix(self, str1, str2): | |
| 144 for i in range(len(str1)): | |
| 145 if not str2.startswith(str1[:i+1]): | |
| 146 return str1[:i] | |
| 147 return str1 | |
| 148 | |
| 149 def shell(self, cmd,verbose=0,debug=0,header=''): | |
| 150 stat = 0 | |
| 151 if verbose or debug: print header+cmd | |
| 152 # flush stdout so we don't mangle python's buffering | |
| 153 if not debug: | |
| 154 input, output = os.popen4(cmd) | |
| 155 print output.read() | |
| 156 output.close() | |
| 157 input.close() | |
| 158 | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
159 class ConsoleView(Gtk.TextView): |
| 709 | 160 def __init__(self): |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
161 Gtk.TextView.__init__(self) |
|
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
162 self.modify_font(Pango.FontDescription('Mono')) |
| 709 | 163 self.set_cursor_visible(True) |
| 164 self.text_buffer = self.get_buffer() | |
| 165 self.mark = self.text_buffer.create_mark('scroll_mark', | |
| 166 self.text_buffer.get_end_iter(), | |
| 167 False) | |
| 168 for code in ansi_colors: | |
| 169 self.text_buffer.create_tag(code, | |
| 170 foreground=ansi_colors[code], | |
| 171 weight=700) | |
| 172 self.text_buffer.create_tag('0') | |
| 173 self.text_buffer.create_tag('notouch', editable=False) | |
| 174 self.color_pat = re.compile('\x01?\x1b\[(.*?)m\x02?') | |
| 175 self.line_start = \ | |
| 176 self.text_buffer.create_mark('line_start', | |
| 177 self.text_buffer.get_end_iter(), True | |
| 178 ) | |
| 179 self.connect('key-press-event', self._onKeypress) | |
| 180 self.last_cursor_pos = 0 | |
| 181 | |
| 182 def write(self, text, editable=False): | |
| 183 segments = self.color_pat.split(text) | |
| 184 segment = segments.pop(0) | |
| 185 start_mark = self.text_buffer.create_mark(None, | |
| 186 self.text_buffer.get_end_iter(), | |
| 187 True) | |
| 188 self.text_buffer.insert(self.text_buffer.get_end_iter(), segment) | |
| 189 | |
| 190 if segments: | |
| 191 ansi_tags = self.color_pat.findall(text) | |
| 192 for tag in ansi_tags: | |
| 193 i = segments.index(tag) | |
| 194 self.text_buffer.insert_with_tags_by_name(self.text_buffer.get_end_iter(), | |
| 195 segments[i+1], tag) | |
| 196 segments.pop(i) | |
| 197 if not editable: | |
| 198 self.text_buffer.apply_tag_by_name('notouch', | |
| 199 self.text_buffer.get_iter_at_mark(start_mark), | |
| 200 self.text_buffer.get_end_iter()) | |
| 201 self.text_buffer.delete_mark(start_mark) | |
| 202 self.scroll_mark_onscreen(self.mark) | |
| 203 | |
| 204 def showPrompt(self, prompt): | |
| 205 self.write(prompt) | |
| 206 self.text_buffer.move_mark(self.line_start,self.text_buffer.get_end_iter()) | |
| 207 | |
| 208 def changeLine(self, text): | |
| 209 iter = self.text_buffer.get_iter_at_mark(self.line_start) | |
| 210 iter.forward_to_line_end() | |
| 211 self.text_buffer.delete(self.text_buffer.get_iter_at_mark(self.line_start), iter) | |
| 212 self.write(text, True) | |
| 213 | |
| 214 def getCurrentLine(self): | |
| 215 rv = self.text_buffer.get_slice(self.text_buffer.get_iter_at_mark(self.line_start), | |
| 216 self.text_buffer.get_end_iter(), False) | |
| 217 return rv | |
| 218 | |
| 219 def showReturned(self, text): | |
| 220 iter = self.text_buffer.get_iter_at_mark(self.line_start) | |
| 221 iter.forward_to_line_end() | |
| 222 self.text_buffer.apply_tag_by_name('notouch', | |
| 223 self.text_buffer.get_iter_at_mark(self.line_start), | |
| 224 iter) | |
| 225 self.write('\n'+text) | |
| 226 if text: | |
| 227 self.write('\n') | |
| 228 self.showPrompt(self.prompt) | |
| 229 self.text_buffer.move_mark(self.line_start,self.text_buffer.get_end_iter()) | |
| 230 self.text_buffer.place_cursor(self.text_buffer.get_end_iter()) | |
| 231 | |
| 232 def _onKeypress(self, obj, event): | |
| 233 if not event.string: | |
| 234 return | |
| 235 insert_mark = self.text_buffer.get_insert() | |
| 236 insert_iter = self.text_buffer.get_iter_at_mark(insert_mark) | |
| 237 selection_mark = self.text_buffer.get_selection_bound() | |
| 238 selection_iter = self.text_buffer.get_iter_at_mark(selection_mark) | |
| 239 start_iter = self.text_buffer.get_iter_at_mark(self.line_start) | |
| 240 if start_iter.compare(insert_iter) <= 0 and \ | |
| 241 start_iter.compare(selection_iter) <= 0: | |
| 242 return | |
| 243 elif start_iter.compare(insert_iter) > 0 and \ | |
| 244 start_iter.compare(selection_iter) > 0: | |
| 245 self.text_buffer.place_cursor(start_iter) | |
| 246 elif insert_iter.compare(selection_iter) < 0: | |
| 247 self.text_buffer.move_mark(insert_mark, start_iter) | |
| 248 elif insert_iter.compare(selection_iter) > 0: | |
| 249 self.text_buffer.move_mark(selection_mark, start_iter) | |
| 250 | |
| 251 | |
| 252 class IPythonView(ConsoleView, IterableIPShell): | |
| 253 def __init__(self, **kw): | |
| 254 ConsoleView.__init__(self) | |
| 255 self.cout = StringIO() | |
| 256 IterableIPShell.__init__(self, cout=self.cout,cerr=self.cout, | |
| 257 input_func=self.raw_input, **kw) | |
| 258 self.connect('key_press_event', self.keyPress) | |
| 259 self.execute() | |
| 260 self.cout.truncate(0) | |
| 261 self.showPrompt(self.prompt) | |
| 262 self.interrupt = False | |
| 263 | |
| 264 def raw_input(self, prompt=''): | |
| 265 if self.interrupt: | |
| 266 self.interrupt = False | |
| 267 raise KeyboardInterrupt | |
| 268 return self.getCurrentLine() | |
| 269 | |
| 270 def keyPress(self, widget, event): | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
271 if event.state & Gdk.ModifierType.CONTROL_MASK and event.keyval == 99: |
| 709 | 272 self.interrupt = True |
| 273 self._processLine() | |
| 274 return True | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
275 elif event.keyval == Gtk.keysyms.Return: |
| 709 | 276 self._processLine() |
| 277 return True | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
278 elif event.keyval == Gtk.keysyms.Up: |
| 709 | 279 self.changeLine(self.historyBack()) |
| 280 return True | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
281 elif event.keyval == Gtk.keysyms.Down: |
| 709 | 282 self.changeLine(self.historyForward()) |
| 283 return True | |
| 284 # todo: Home needs to advance past the ipython prompt | |
|
1056
547d65ea9902
port curvecalc to gtk3. mostly worked, but there are severe bugs with redraws
Drew Perttula <drewp@bigasterisk.com>
parents:
709
diff
changeset
|
285 elif event.keyval == Gtk.keysyms.Tab: |
| 709 | 286 if not self.getCurrentLine().strip(): |
| 287 return False | |
| 288 completed, possibilities = self.complete(self.getCurrentLine()) | |
| 289 if len(possibilities) > 1: | |
| 290 slice = self.getCurrentLine() | |
| 291 self.write('\n') | |
| 292 for symbol in possibilities: | |
| 293 self.write(symbol+'\n') | |
| 294 self.showPrompt(self.prompt) | |
| 295 self.changeLine(completed or slice) | |
| 296 return True | |
| 297 | |
| 298 def _processLine(self): | |
| 299 self.history_pos = 0 | |
| 300 self.execute() | |
| 301 rv = self.cout.getvalue() | |
| 302 if rv: rv = rv.strip('\n') | |
| 303 self.showReturned(rv) | |
| 304 self.cout.truncate(0) |