hacker external pager (hardcoded to less) added for bpython-curtsies

thomasballinger · thomasballinger · commit 96bf950268f4 · 2014-07-11T14:08:37.000-04:00
diff --git a/bpython/curtsies.py b/bpython/curtsies.py
@@ -76,7 +76,8 @@ def event_or_refresh(timeout=None):
                       get_cursor_vertical_diff=window.get_cursor_vertical_diff,
                       banner=banner,
                       interp=interp,
-                      interactive=interactive) as repl:
+                      interactive=interactive,
+                      orig_tcattrs=input_generator.original_stty) as repl:
                 repl.height, repl.width = window.t.height, window.t.width
                 sys.repl = repl
 
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -6,7 +6,9 @@
 import os
 import re
 import signal
+import subprocess
 import sys
+import tempfile
 import threading
 import unicodedata
 
@@ -15,13 +17,15 @@
 from bpython.config import Struct, loadini, default_config_path
 from bpython.formatter import BPythonFormatter
 from pygments import format
+from pygments.lexers import PythonLexer
+from pygments.formatters import TerminalFormatter
 from bpython import importcompletion
 from bpython import translations
 translations.init()
 from bpython.translations import _
 from bpython._py3compat import py3
 
-from curtsies import FSArray, fmtstr, FmtStr
+from curtsies import FSArray, fmtstr, FmtStr, Termmode
 from curtsies.bpythonparse import parse as bpythonparse
 from curtsies.bpythonparse import func_for_letter, color_for_letter
 from curtsies import fmtfuncs
@@ -152,7 +156,8 @@ class Repl(BpythonRepl):
     ## initialization, cleanup
     def __init__(self, locals_=None, config=None,
             request_refresh=lambda: None, get_term_hw=lambda:(50, 10),
-            get_cursor_vertical_diff=lambda: 0, banner=None, interp=None, interactive=True):
+            get_cursor_vertical_diff=lambda: 0, banner=None, interp=None, interactive=True,
+            orig_tcattrs=None):
         """
         locals_ is a mapping of locals to pass into the interpreter
         config is a bpython config.Struct with config attributes
@@ -223,6 +228,7 @@ def smarter_request_refresh():
         self.scroll_offset = 0 # how many times display has been scrolled down
                                # because there wasn't room to display everything
         self.cursor_offset = 0 # from the left, 0 means first char
+        self.orig_tcattrs = orig_tcattrs # useful for shelling out with normal terminal
 
         self.coderunner = CodeRunner(self.interp, self.request_refresh)
         self.stdout = FakeOutput(self.coderunner, self.send_to_stdout)
@@ -360,8 +366,17 @@ def process_event(self, e):
             self.request_paint_to_clear_screen = True
         elif e in key_dispatch[self.config.last_output_key]: #TODO Not Implemented
             pass
-        elif e in key_dispatch[self.config.show_source_key]: #TODO Not Implemented
-            pass
+        elif e in ('\x1f',) + key_dispatch[self.config.show_source_key]: #TODO Not Implemented
+            source = self.get_source_of_current_name()
+            if source is None:
+                self.status_bar.message('whoops!')
+            else:
+                if self.config.highlight_show_source:
+                    source = format(PythonLexer().get_tokens(source), TerminalFormatter())
+                with tempfile.NamedTemporaryFile() as tmp:
+                    tmp.write(source)
+                    tmp.flush()
+                    self.focus_on_subprocess(['less', '-R', tmp.name])
         elif e in key_dispatch[self.config.suspend_key]:
             raise SystemExit()
         elif e in ("",) + key_dispatch[self.config.exit_key]:
@@ -937,6 +952,27 @@ def getstdout(self):
                      ) if lines else ''
         return s
 
+    def focus_on_subprocess(self, args):
+        import blessings
+        prev_sigwinch_handler = signal.getsignal(signal.SIGWINCH)
+        try:
+            signal.signal(signal.SIGWINCH, self.orig_sigwinch_handler)
+            with Termmode(self.orig_stdin, self.orig_tcattrs):
+                terminal = blessings.Terminal(stream=sys.__stdout__)
+                with terminal.fullscreen():
+                    sys.__stdout__.write(terminal.save)
+                    sys.__stdout__.write(terminal.move(0, 0))
+                    sys.__stdout__.flush()
+                    p = subprocess.Popen(args,
+                                         stdin=self.orig_stdin,
+                                         stderr=sys.__stderr__,
+                                         stdout=sys.__stdout__)
+                    p.wait()
+                    sys.__stdout__.write(terminal.restore)
+                    sys.__stdout__.flush()
+        finally:
+            signal.signal(signal.SIGWINCH, prev_sigwinch_handler)
+
 def is_nop(char):
     return unicodedata.category(unicode(char)) == 'Cc'
 
diff --git a/bpython/test/test_autocomplete.py b/bpython/test/test_autocomplete.py
@@ -9,8 +9,8 @@ class TestSimpleComplete(unittest.TestCase):
     complete = partial(autocomplete.complete,
                        namespace={'zabcdef':1, 'zabcqwe':2, 'ze':3},
                        config=simple_config)
-    kwargs = {locals_={'zabcdef':1, 'zabcqwe':2, 'ze':3},
-              argspec=inspect.getargspec(lambda x: x),
+    kwargs = {locals_:{'zabcdef':1, 'zabcqwe':2, 'ze':3},
+              argspec:inspect.getargspec(lambda x: x),
 
     def test_simple_completion(self):
         self.assertEqual(self.complete('zab'), ['zabcdef', 'zabcqwe'])
