Added support for the nifty ptpython shell.

inducer · Feb 13, 2015 · dc3a6c2 · dc3a6c2
1 parent e9c8dd3
commit dc3a6c2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/pudb/debugger.py b/pudb/debugger.py
@@ -1672,6 +1672,8 @@ def run_external_cmdline(w, size, key):
                 runner = shell.run_ipython_shell
             elif shell.HAVE_BPYTHON and CONFIG["shell"] == "bpython":
                 runner = shell.run_bpython_shell
+            elif shell.HAVE_PTPYTHON and CONFIG["shell"] == "ptpython":
+                runner = shell.run_ptpython_shell
             else:
                 runner = shell.run_classic_shell
 

diff --git a/pudb/settings.py b/pudb/settings.py
@@ -203,7 +203,7 @@ def _update_config(check_box, new_state, option_newvalue):
 
     shell_info = urwid.Text("This is the shell that will be "
             "used when you hit '!'.\n")
-    shells = ["internal", "classic", "ipython", "bpython"]
+    shells = ["internal", "classic", "ipython", "bpython", "ptpython"]
 
     shell_rb_group = []
     shell_rbs = [

diff --git a/pudb/shell.py b/pudb/shell.py
@@ -12,6 +12,13 @@
 else:
     HAVE_BPYTHON = True
 
+try:
+    from prompt_toolkit.contrib.repl import embed as ptpython_embed
+except ImportError:
+    HAVE_PTPYTHON = False
+else:
+    HAVE_PTPYTHON = True
+
 
 # {{{ readline wrangling
 
@@ -141,6 +148,14 @@ def run_ipython_shell_v11(locals, globals, first_time):
     _update_ns(shell, old_locals, old_globals)
 
 
+def run_ptpython_shell(locals, globals, first_time):
+    ## Use the default ptpython history:
+    import os
+    history_filename = os.path.expanduser('~/.ptpython_history')
+    ptpython_embed(globals.copy(), locals.copy(),
+                   history_filename=history_filename)
+
+
 def _update_ns(shell, locals, globals):
     '''Update the IPython 0.11 namespace at every visit'''