121 lines
4.4 KiB
Python
121 lines
4.4 KiB
Python
import code, re, string, io, sys
|
|
import buffer, color, completer, lex, method, mode, mode.mini, mode.consolemini, window
|
|
from subprocess import Popen, PIPE, STDOUT
|
|
from mode.python import PythonGrammar
|
|
from point import Point
|
|
|
|
class IpythonExec(method.Method):
|
|
def _execute(self, w, **vargs):
|
|
if w.application.completion_window_is_open():
|
|
w.application.close_completion_buffer()
|
|
s = w.buffer.make_string()
|
|
w.mode.history[-1] = s
|
|
w.mode.history.append('')
|
|
w.buffer.set_data('')
|
|
w.mode.hindex = len(w.mode.history) - 1
|
|
|
|
b = w.mode.get_ipython()
|
|
a = w.application
|
|
if a.window().buffer is not b:
|
|
a.switch_buffer(b)
|
|
p = a.get_mini_buffer_prompt()
|
|
b.insert_string(b.get_buffer_end(), p + s + '\n', force=True)
|
|
|
|
b.pipe_write("ENTER:%s" % s)
|
|
w.mode.read_sync()
|
|
|
|
class IpythonTab(method.Method):
|
|
def execute(self, w, **vargs):
|
|
a = w.application
|
|
s = w.buffer.make_string()
|
|
|
|
x2 = w.logical_cursor().x
|
|
if not s or s[:x2].isspace():
|
|
w.insert_string_at_cursor(' ' * w.mode.tabwidth)
|
|
return
|
|
|
|
r = re.compile('^[a-zA-Z0-9_.]$')
|
|
line = s
|
|
x1 = x2
|
|
while x1 > 0 and r.match(s[x1 - 1]):
|
|
x1 -= 1
|
|
word = line[x1:x2]
|
|
|
|
b = w.mode.get_ipython()
|
|
candidates = b.completions(word)
|
|
if candidates:
|
|
s = completer.find_common_string(candidates)
|
|
w.buffer.delete(Point(x1, 0), Point(x2, 0), force=True)
|
|
w.insert_string_at_cursor(s)
|
|
mode.mini.use_completion_window(a, s, candidates)
|
|
elif a.completion_window_is_open():
|
|
a.close_completion_buffer()
|
|
|
|
class IpythonPathStart(method.Method):
|
|
'''Interactively run python statements in the context of a buffer'''
|
|
def _start(self, w, parent):
|
|
a = w.application
|
|
if w.buffer.btype == 'ipython':
|
|
b = w.buffer
|
|
else:
|
|
name = buffer.IpythonBuffer.create_name(parent)
|
|
if not a.has_buffer_name(name):
|
|
b = buffer.IpythonBuffer(parent, a)
|
|
a.add_buffer(b)
|
|
window.Window(b, a)
|
|
else:
|
|
b = a.get_buffer_by_name(name)
|
|
self.main_buffer = b
|
|
if a.window().buffer is not b:
|
|
a.switch_buffer(b)
|
|
f = lambda x: None
|
|
w.application.open_mini_buffer('*** ', f, self, None, 'ipythonmini')
|
|
def execute(self, w, **vargs):
|
|
self._start(w, w.buffer)
|
|
|
|
class IpythonStart(IpythonPathStart):
|
|
'''Interactively run python statements'''
|
|
def execute(self, w, **vargs):
|
|
self._start(w, None)
|
|
|
|
class IpythonPageUp(mode.consolemini.ConsolePageUp):
|
|
subbuf = '*IPython*'
|
|
class IpythonPageDown(mode.consolemini.ConsolePageDown):
|
|
subbuf = '*IPython*'
|
|
class IpythonGotoBeginning(mode.consolemini.ConsoleGotoBeginning):
|
|
subbuf = '*IPython*'
|
|
class IpythonGotoEnd(mode.consolemini.ConsoleGotoEnd):
|
|
subbuf = '*IPython*'
|
|
|
|
class IpythonMini(mode.Fundamental):
|
|
name = 'IpythonMini'
|
|
actions = [IpythonExec, IpythonTab, IpythonStart, IpythonPathStart,
|
|
IpythonPageUp, IpythonPageDown, IpythonGotoBeginning, IpythonGotoEnd]
|
|
def get_ipython(self):
|
|
return self.window.buffer.method.main_buffer
|
|
def read_sync(self):
|
|
b = self.get_ipython()
|
|
b.pipe_read()
|
|
self.window.application.set_mini_buffer_prompt(b.prompt)
|
|
def __init__(self, w):
|
|
mode.Fundamental.__init__(self, w)
|
|
self.history = ['']
|
|
self.hindex = 0
|
|
b = self.window.buffer.method.main_buffer
|
|
assert hasattr(b, 'pipe')
|
|
self.window.application.set_mini_buffer_prompt(b.prompt)
|
|
self.add_bindings('ipython-exec', ('RETURN',))
|
|
self.add_bindings('console-clear', ('C-l',))
|
|
self.add_bindings('console-cancel', ('C-]', 'C-g'))
|
|
self.add_bindings('console-history-prev', ('C-p', 'UP'))
|
|
self.add_bindings('console-history-next', ('C-n', 'DOWN'))
|
|
self.add_bindings('ipython-tab', ('TAB',))
|
|
self.add_bindings('ipython-page-up', ('M-v',))
|
|
self.add_bindings('ipython-page-down', ('C-v',))
|
|
self.add_bindings('ipython-goto-beginning', ('M-<',))
|
|
self.add_bindings('ipython-goto-end', ('M->',))
|
|
for c in string.ascii_letters + string.digits + string.punctuation:
|
|
self.add_binding('insert-string-%s' % c, c)
|
|
|
|
install = IpythonMini.install
|