Skip to content

Use python-prompt-toolkit for ipdb. #9481

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Carreau wants to merge 1 commit into from
Closed

Use python-prompt-toolkit for ipdb. #9481

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 42 additions & 7 deletions IPython/core/debugger.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@
from IPython.core.excolors import exception_colors
from IPython.testing.skipdoctest import skip_doctest

from prompt_toolkit import prompt as ptk_prompt

# See if we can use pydb.
has_pydb = False
prompt = 'ipdb> '
Expand Down Expand Up @@ -285,13 +287,46 @@ def __init__(self,color_scheme='NoColor',completekey=None,
self.parser = PyColorize.Parser()

# Set the prompt - the default prompt is '(Pdb)'
Colors = cst.active_colors
if color_scheme == 'NoColor':
self.prompt = prompt
else:
# The colour markers are wrapped by bytes 01 and 02 so that readline
# can calculate the width.
self.prompt = u'\x01%s\x02%s\x01%s\x02' % (Colors.prompt, prompt, Colors.Normal)
self.prompt = prompt


def cmdloop(self, intro=None):
"""Repeatedly issue a prompt, accept input, parse an initial prefix
off the received input, and dispatch to action methods, passing them
the remainder of the line as argument.

override the same methods from cmd.Cmd to provide prompt toolkit replacement.
"""
if not self.use_rawinput:
raise ValueError('Sorry ipdb does not support raw_input=False')

def get_prompt_tokens(cli):
from pygments.token import Token
return [(Token.Prompt, self.prompt)]

self.preloop()
try:
if intro is not None:
self.intro = intro
if self.intro:
self.stdout.write(str(self.intro)+"\n")
stop = None
while not stop:
if self.cmdqueue:
line = self.cmdqueue.pop(0)
else:
try:
line = ptk_prompt(get_prompt_tokens=get_prompt_tokens)
except EOFError:
line = 'EOF'
line = self.precmd(line)
stop = self.onecmd(line)
stop = self.postcmd(stop, line)
self.postloop()
except Exception:
pass



def set_colors(self, scheme):
"""Shorthand access to the color table scheme selector method."""
Expand Down