"""Given a stream of tokens and a screen width plus an optional

lines = 1

pos = cursor

for (token, value) in tokens:

if token is Token.Text and value == '\n':

lines += 1

else:

pos += len(value)

lines += pos // width

pos %= width

"""Provide a fake stdin type for things like raw_input() etc."""

def __init__(self, interface):

"""Take the curses Repl on init and assume it provides a get_key method

self.encoding = getpreferredencoding()

self.interface = interface

self.buffer = list()

def __iter__(self):

return iter(self.readlines())

def flush(self):

"""Flush the internal buffer. This is a no-op. Flushing stdin

def write(self, value):

# XXX IPython expects sys.stdin.write to exist, there will no doubt be

# others, so here's a hack to keep them happy

raise IOError(errno.EBADF, "sys.stdin is read-only")

def isatty(self):

return True

def readline(self, size=-1):

"""I can't think of any reason why anything other than readline would

if not size:

return ''

elif self.buffer:

buffer = self.buffer.pop(0)

else:

buffer = ''

curses.raw(True)

try:

while not buffer.endswith('\n'):

key = self.interface.get_key()

if key in [curses.erasechar(), 'KEY_BACKSPACE']:

y, x = self.interface.scr.getyx()

if buffer:

self.interface.scr.delch(y, x - 1)

buffer = buffer[:-1]

continue

elif key == chr(4) and not buffer:

# C-d

return ''

elif (key != '\n' and

(len(key) > 1 or unicodedata.category(key) == 'Cc')):

continue

sys.stdout.write(key)

buffer += key

finally:

curses.raw(False)

if size > 0:

rest = buffer[size:]

if rest:

self.buffer.append(rest)

buffer = buffer[:size]

if py3:

return buffer

else:

return buffer.encode(getpreferredencoding())

def read(self, size=None):

if size == 0:

return ''

data = list()

while size is None or size > 0:

line = self.readline(size or -1)

if not line:

break

if size is not None:

size -= len(line)

data.append(line)

return ''.join(data)

def readlines(self, size=-1):

"""Init all the colours in curses and bang them into a dictionary"""

# blacK, Red, Green, Yellow, Blue, Magenta, Cyan, White, Default:

c = {

'k': 0,

'r': 1,

'g': 2,

'y': 3,

'b': 4,

'm': 5,

'c': 6,

'w': 7,

'd': -1,

}

for i in range(63):

if i > 7:

j = i // 8

else:

j = c[config.color_scheme['background']]

curses.init_pair(i + 1, i % 8, j)

def __init__(self, scr, interp, statusbar, config, idle=None):

Repl.__init__(self, interp, config)

interp.writetb = self.writetb

self.scr = scr

self.list_win = newwin(get_colpair(config, 'background'), 1, 1, 1, 1)

self.cpos = 0

self.do_exit = False

self.f_string = ''

self.idle = idle

self.in_hist = False

self.paste_mode = False

self.last_key_press = time.time()

self.s = ''

self.statusbar = statusbar

self.formatter = BPythonFormatter(config.color_scheme)

def addstr(self, s):

"""Add a string to the current input line and figure out

if not self.cpos:

self.s += s

else:

l = len(self.s)

self.s = self.s[:l - self.cpos] + s + self.s[l - self.cpos:]

self.complete()

def atbol(self):

"""Return True or False accordingly if the cursor is at the beginning

return not self.s.lstrip()

def back(self):

"""Replace the active line with previous line in history and

self.cpos = 0

self.rl_history.enter(self.s)

self.clear_wrapped_lines()

self.s = self.rl_history.back()

self.print_line(self.s, clr=True)

def bs(self, delete_tabs=True):

"""Process a backspace"""

y, x = self.scr.getyx()

if not self.s:

return

if x == self.ix and y == self.iy:

return

n = 1

self.clear_wrapped_lines()

if not self.cpos:

if self.atbol() and delete_tabs:

n = len(self.s) % self.config.tab_length

if not n:

n = self.config.tab_length

self.s = self.s[:-n]

else:

self.s = self.s[:-self.cpos - 1] + self.s[-self.cpos:]

self.print_line(self.s, clr=True)

return n

def bs_word(self):

pos = len(self.s) - self.cpos - 1

while pos >= 0 and self.s[pos] == ' ':

pos -= self.bs()

while pos >= 0 and self.s[pos] != ' ':

pos -= self.bs()

def check(self):

"""Check if paste mode should still be active and, if not, deactivate

if (self.paste_mode

and time.time() - self.last_key_press > self.config.paste_time):

self.paste_mode = False

self.print_line(self.s)

def clear_current_line(self):

"""Called when a SyntaxError occured in the interpreter. It is

Repl.clear_current_line(self)

self.s = ''

def clear_wrapped_lines(self):

"""Clear the wrapped lines of the current input."""

# curses does not handle this on its own. Sad.

height, width = self.scr.getmaxyx()

max_y = min(self.iy + (self.ix + len(self.s)) // width + 1, height)

for y in xrange(self.iy + 1, max_y):

self.scr.move(y, 0)

self.scr.clrtoeol()

def complete(self, tab=False):

if self.paste_mode and self.list_win_visible:

self.scr.touchwin()

if self.paste_mode:

return

if self.list_win_visible and not self.config.auto_display_list:

self.scr.touchwin()

self.list_win_visible = False

return

if self.config.auto_display_list or tab:

self.list_win_visible = Repl.complete(self, tab)

if self.list_win_visible:

try:

self.show_list(self.matches, self.argspec)

except curses.error:

# XXX: This is a massive hack, it will go away when I get

# cusswords into a good enough state that we can start

# using it.

self.list_win.border()

self.list_win.refresh()

self.list_win_visible = False

if not self.list_win_visible:

self.scr.redrawwin()

self.scr.refresh()

def clrtobol(self):

"""Clear from cursor to beginning of line; usual C-u behaviour"""

self.clear_wrapped_lines()

if not self.cpos:

self.s = ''

else:

self.s = self.s[-self.cpos:]

self.print_line(self.s, clr=True)

self.scr.redrawwin()

self.scr.refresh()

def current_line(self):

"""Return the current line."""

return self.s

def cut_to_buffer(self):

"""Clear from cursor to end of line, placing into cut buffer"""

self.cut_buffer = self.s[-self.cpos:]

self.s = self.s[:-self.cpos]

self.cpos = 0

self.print_line(self.s, clr=True)

self.scr.redrawwin()

self.scr.refresh()

def cw(self):

"""Return the current word, i.e. the (incomplete) word directly to the

if self.cpos:

return

l = len(self.s)

if (not self.s or

(not self.s[l - 1].isalnum() and

self.s[l - 1] not in ('.', '_'))):

return

i = 1

while i < l + 1:

if not self.s[-i].isalnum() and self.s[-i] not in ('.', '_'):

break

i += 1

return self.s[-i + 1:]

def delete(self):

"""Process a del"""

if not self.s:

return

if self.mvc(-1):

self.bs(False)

def echo(self, s, redraw=True):

"""Parse and echo a formatted string with appropriate attributes. It

if not py3 and isinstance(s, unicode):

s = s.encode(getpreferredencoding())

a = get_colpair(self.config, 'output')

if '\x01' in s:

rx = re.search('\x01([A-Za-z])([A-Za-z]?)', s)

if rx:

fg = rx.groups()[0]

bg = rx.groups()[1]

col_num = self._C[fg.lower()]

if bg and bg != 'I':

col_num *= self._C[bg.lower()]

a = curses.color_pair(int(col_num) + 1)

if bg == 'I':

a = a | curses.A_REVERSE

s = re.sub('\x01[A-Za-z][A-Za-z]?', '', s)

if fg.isupper():

a = a | curses.A_BOLD

s = s.replace('\x03', '')

s = s.replace('\x01', '')

# Replace NUL bytes, as addstr raises an exception otherwise

s = s.replace('\x00', '')

self.scr.addstr(s, a)

if redraw and not self.evaluating:

self.scr.refresh()

def end(self, refresh=True):

self.cpos = 0

h, w = gethw()

y, x = divmod(len(self.s) + self.ix, w)

y += self.iy

self.scr.move(y, x)

if refresh:

self.scr.refresh()

return True

def fwd(self):

"""Same as back() but, well, forward"""

self.cpos = 0

self.clear_wrapped_lines()

self.rl_history.enter(self.s)

self.s = self.rl_history.forward()

self.print_line(self.s, clr=True)

def get_key(self):

key = ''

while True:

try:

key += self.scr.getkey()

if not py3:

key = key.decode(getpreferredencoding())

self.scr.nodelay(False)

except UnicodeDecodeError:

self.scr.nodelay(True)

except curses.error:

self.scr.nodelay(False)

# XXX What to do here? Raise an exception?

if key:

return key

else:

t = time.time()

self.paste_mode = (

t - self.last_key_press <= self.config.paste_time

)

self.last_key_press = t

return key

finally:

if self.idle:

self.idle(self)

def get_line(self):

"""Get a line of text and return it

self.s = ''

self.rl_history.reset()

self.iy, self.ix = self.scr.getyx()

if not self.paste_mode:

for _ in xrange(self.next_indentation()):

self.p_key('\t')

self.cpos = 0

while True:

key = self.get_key()

if self.p_key(key) is None:

return self.s

def home(self, refresh=True):

self.scr.move(self.iy, self.ix)

self.cpos = len(self.s)

if refresh:

self.scr.refresh()

return True

def lf(self):

"""Process a linefeed character; it only needs to check the

if self.cpos:

for _ in range(self.cpos):

self.mvc(-1)

# Reprint the line (as there was maybe a highlighted paren in it)

self.print_line(self.s, newline=True)

self.echo("\n")

def mkargspec(self, topline, down):

"""This figures out what to do with the argspec and puts it nicely into

r = 3

fn = topline[0]

args = topline[1][0]

kwargs = topline[1][3]

_args = topline[1][1]

_kwargs = topline[1][2]

is_bound_method = topline[2]

in_arg = topline[3]

if py3:

kwonly = topline[1][4]

kwonly_defaults = topline[1][5] or dict()

max_w = int(self.scr.getmaxyx()[1] * 0.6)

self.list_win.erase()

self.list_win.resize(3, max_w)

h, w = self.list_win.getmaxyx()

self.list_win.addstr('\n ')

self.list_win.addstr(fn,

get_colpair(self.config, 'name') | curses.A_BOLD)

self.list_win.addstr(': (', get_colpair(self.config, 'name'))

maxh = self.scr.getmaxyx()[0]

if is_bound_method and isinstance(in_arg, int):

in_arg += 1

punctuation_colpair = get_colpair(self.config, 'punctuation')

for k, i in enumerate(args):

y, x = self.list_win.getyx()

ln = len(str(i))

kw = None

if kwargs and k + 1 > len(args) - len(kwargs):

kw = str(kwargs[k - (len(args) - len(kwargs))])

ln += len(kw) + 1

if ln + x >= w:

ty = self.list_win.getbegyx()[0]

if not down and ty > 0:

h += 1

self.list_win.mvwin(ty - 1, 1)

self.list_win.resize(h, w)

elif down and h + r < maxh - ty:

h += 1

self.list_win.resize(h, w)

else:

break

r += 1

self.list_win.addstr('\n\t')

if str(i) == 'self' and k == 0:

color = get_colpair(self.config, 'name')

else:

color = get_colpair(self.config, 'token')

if k == in_arg or i == in_arg:

color |= curses.A_BOLD

self.list_win.addstr(str(i), color)

if kw:

self.list_win.addstr('=', punctuation_colpair)

self.list_win.addstr(kw, get_colpair(self.config, 'token'))

if k != len(args) -1:

self.list_win.addstr(', ', punctuation_colpair)

if _args:

if args:

self.list_win.addstr(', ', punctuation_colpair)

self.list_win.addstr('*%s' % (_args, ),

get_colpair(self.config, 'token'))

if py3 and kwonly:

if not _args:

if args:

self.list_win.addstr(', ', punctuation_colpair)

self.list_win.addstr('*', punctuation_colpair)

marker = object()

for arg in kwonly:

self.list_win.addstr(', ', punctuation_colpair)

color = get_colpair(self.config, 'token')

if arg == in_arg:

color |= curses.A_BOLD

self.list_win.addstr(arg, color)

default = kwonly_defaults.get(arg, marker)

if default is not marker:

self.list_win.addstr('=', punctuation_colpair)

self.list_win.addstr(default,

get_colpair(self.config, 'token'))

if _kwargs:

if args or _args or (py3 and kwonly):

self.list_win.addstr(', ', punctuation_colpair)

self.list_win.addstr('**%s' % (_kwargs, ),

get_colpair(self.config, 'token'))

self.list_win.addstr(')', punctuation_colpair)

return r

def mvc(self, i, refresh=True):

"""This method moves the cursor relatively from the current

y, x = self.scr.getyx()

if self.cpos == 0 and i < 0:

return False

if x == self.ix and y == self.iy and i >= 1:

return False

h, w = gethw()

if x - i < 0:

y -= 1

x = w

if x - i >= w:

y += 1

x = 0 + i

self.cpos += i

self.scr.move(y, x - i)

if refresh:

self.scr.refresh()

return True

def p_key(self, key):

"""Process a keypress"""

if key is None:

return ''

config = self.config

if key == chr(8): # C-Backspace (on my computer anyway!)

self.clrtobol()

key = '\n'

# Don't return; let it get handled

if key == chr(27):

return ''

if key in (chr(127), 'KEY_BACKSPACE'):

self.bs()

self.complete()

return ''

elif key in key_dispatch[config.delete_key] and not self.s:

# Delete on empty line exits

self.do_exit = True

return None

elif key in ('KEY_DC', ) + key_dispatch[config.delete_key]:

self.delete()

self.complete()

# Redraw (as there might have been highlighted parens)

self.print_line(self.s)

return ''

elif key in key_dispatch[config.undo_key]: # C-r

self.undo()

return ''

elif key in ('KEY_UP', ) + key_dispatch[config.up_one_line_key]:

# Cursor Up/C-p

self.back()

return ''

elif key in ('KEY_DOWN', ) + key_dispatch[config.down_one_line_key]:

# Cursor Down/C-n

self.fwd()

return ''

elif key in ("KEY_LEFT",' ^B', chr(2)): # Cursor Left or ^B

self.mvc(1)

# Redraw (as there might have been highlighted parens)

self.print_line(self.s)

elif key in ("KEY_RIGHT", '^F', chr(6)): # Cursor Right or ^F

self.mvc(-1)

# Redraw (as there might have been highlighted parens)

self.print_line(self.s)

elif key in ("KEY_HOME", '^A', chr(1)): # home or ^A

self.home()

# Redraw (as there might have been highlighted parens)

self.print_line(self.s)

elif key in ("KEY_END", '^E', chr(5)): # end or ^E

self.end()

# Redraw (as there might have been highlighted parens)

self.print_line(self.s)

elif key in key_dispatch[config.cut_to_buffer_key]: # cut to buffer

self.cut_to_buffer()

return ''

elif key in key_dispatch[config.yank_from_buffer_key]:

# yank from buffer

self.yank_from_buffer()

return ''

elif key in key_dispatch[config.clear_word_key]:

self.bs_word()

self.complete()

return ''

elif key in key_dispatch[config.clear_line_key]:

self.clrtobol()

return ''

elif key in key_dispatch[config.clear_screen_key]:

self.s_hist = [self.s_hist[-1]]

self.highlighted_paren = None

self.redraw()

return ''

elif key in key_dispatch[config.exit_key]:

if not self.s:

self.do_exit = True

return None

else:

return ''

elif key in key_dispatch[config.save_key]:

self.write2file()

return ''

elif key in key_dispatch[config.pastebin_key]:

self.pastebin()

return ''

elif key in key_dispatch[config.last_output_key]:

page(self.stdout_hist[self.prev_block_finished:-4])

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:

if config.highlight_show_source:

source = format(PythonLexer().get_tokens(source),

TerminalFormatter())

page(source)

return ''

elif key == '\n':

self.lf()

return None

elif key == '\t':

return self.tab()

elif key == 'KEY_BTAB':

return self.tab(back=True)

elif len(key) == 1 and not unicodedata.category(key) == 'Cc':

self.addstr(key)

self.print_line(self.s)

else:

return ''

return True

def print_line(self, s, clr=False, newline=False):

"""Chuck a line of text through the highlighter, move the cursor

if not s:

clr = True

if self.highlighted_paren is not None:

# Clear previous highlighted paren

self.reprint_line(*self.highlighted_paren)

self.highlighted_paren = None

if self.config.syntax and (not self.paste_mode or newline):

o = format(self.tokenize(s, newline), self.formatter)

else:

o = s

self.f_string = o

self.scr.move(self.iy, self.ix)

if clr:

self.scr.clrtoeol()

if clr and not s:

self.scr.refresh()

if o:

for t in o.split('\x04'):

self.echo(t.rstrip('\n'))

if self.cpos:

t = self.cpos

for _ in range(self.cpos):

self.mvc(1)

self.cpos = t

def prompt(self, more):

"""Show the appropriate Python prompt"""

if not more:

self.echo("\x01%s\x03>>> " % (self.config.color_scheme['prompt'],))

self.stdout_hist += '>>> '

self.s_hist.append('\x01%s\x03>>> \x04' %

(self.config.color_scheme['prompt'],))

else:

prompt_more_color = self.config.color_scheme['prompt_more']

self.echo("\x01%s\x03... " % (prompt_more_color, ))

self.stdout_hist += '... '

self.s_hist.append('\x01%s\x03... \x04' % (prompt_more_color, ))

def push(self, s, insert_into_history=True):

# curses.raw(True) prevents C-c from causing a SIGINT

curses.raw(False)

try:

return Repl.push(self, s, insert_into_history)

except SystemExit:

# Avoid a traceback on e.g. quit()

self.do_exit = True

return False

finally:

curses.raw(True)

def redraw(self):

"""Redraw the screen."""

self.scr.erase()

for k, s in enumerate(self.s_hist):

if not s:

continue

self.iy, self.ix = self.scr.getyx()

for i in s.split('\x04'):

self.echo(i, redraw=False)

if k < len(self.s_hist) -1:

self.scr.addstr('\n')

self.iy, self.ix = self.scr.getyx()

self.print_line(self.s)

self.scr.refresh()

self.statusbar.refresh()

def repl(self):

"""Initialise the repl and jump into the loop. This method also has to

self.push('from bpython._internal import _help as help\n', False)

self.iy, self.ix = self.scr.getyx()

more = False

while not self.do_exit:

self.f_string = ''

self.prompt(more)

try:

inp = self.get_line()

except KeyboardInterrupt:

self.statusbar.message('KeyboardInterrupt')

self.scr.addstr('\n')

self.scr.touchwin()

self.scr.refresh()

continue

self.scr.redrawwin()

if self.do_exit:

return

self.history.append(inp)

self.s_hist[-1] += self.f_string

if py3:

self.stdout_hist += inp + '\n'

else:

self.stdout_hist += inp.encode(getpreferredencoding()) + '\n'

stdout_position = len(self.stdout_hist)

more = self.push(inp)

if not more:

self.prev_block_finished = stdout_position

self.s = ''

def reprint_line(self, lineno, tokens):

"""Helper function for paren highlighting: Reprint line at offset

if not self.buffer or lineno == len(self.buffer):

return

real_lineno = self.iy

height, width = self.scr.getmaxyx()

for i in xrange(lineno, len(self.buffer)):

string = self.buffer[i]

# 4 = length of prompt

length = len(string.encode(getpreferredencoding())) + 4

real_lineno -= int(math.ceil(length / width))

if real_lineno < 0:

return

self.scr.move(real_lineno, 4)

line = format(tokens, BPythonFormatter(self.config.color_scheme))

for string in line.split('\x04'):

self.echo(string)

def resize(self):

"""This method exists simply to keep it straight forward when

self.size()

self.scr.erase()

self.scr.resize(self.h, self.w)

self.scr.mvwin(self.y, self.x)

self.statusbar.resize(refresh=False)

self.redraw()

def show_list(self, items, topline=None, current_item=None):

shared = Struct()

shared.cols = 0

shared.rows = 0