KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

DRYer

Browse Source
This commit is contained in:
Kovid Goyal 2022-02-23 18:14:40 +05:30
parent f62acab715
commit 03f9f29ce4
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
kitty/window.py
View File

@ -301,6 +301,16 @@ def text_sanitizer(as_ansi: bool, add_wrap_markers: bool) -> Callable[[str], str
return remove_wrap_markers return remove_wrap_markers
def cmd_output(screen: Screen, which: CommandOutput = CommandOutput.last_run, as_ansi: bool = False, add_wrap_markers: bool = False) -> str:
lines: List[str] = []
search_in_pager_hist = screen.cmd_output(which, lines.append, as_ansi, add_wrap_markers)
if search_in_pager_hist:
pht = pagerhist(screen, as_ansi, add_wrap_markers, True)
if pht:
lines.insert(0, pht)
return ''.join(lines)
def process_remote_print(msg: bytes) -> str: def process_remote_print(msg: bytes) -> str:
from base64 import standard_b64decode from base64 import standard_b64decode
from .cli import green from .cli import green
@ -1069,13 +1079,7 @@ class Window:
return as_text(self.screen, as_ansi, add_history, add_wrap_markers, alternate_screen, add_cursor) return as_text(self.screen, as_ansi, add_history, add_wrap_markers, alternate_screen, add_cursor)
def cmd_output(self, which: CommandOutput = CommandOutput.last_run, as_ansi: bool = False, add_wrap_markers: bool = False) -> str: def cmd_output(self, which: CommandOutput = CommandOutput.last_run, as_ansi: bool = False, add_wrap_markers: bool = False) -> str:
lines: List[str] = [] return cmd_output(self.screen, which, as_ansi, add_wrap_markers)
search_in_pager_hist = self.screen.cmd_output(which, lines.append, as_ansi, add_wrap_markers)
if search_in_pager_hist:
pht = pagerhist(self.screen, as_ansi, add_wrap_markers, True)
if pht:
lines.insert(0, pht)
return ''.join(lines)
@property @property
def cwd_of_child(self) -> Optional[str]: def cwd_of_child(self) -> Optional[str]:

6
kitty_tests/__init__.py
View File

@ -240,7 +240,5 @@ class PTY:
return '\n'.join(lines) return '\n'.join(lines)
def last_cmd_output(self, as_ansi=False, add_wrap_markers=False): def last_cmd_output(self, as_ansi=False, add_wrap_markers=False):
lines = [] from kitty.window import cmd_output
from kitty.window import CommandOutput return cmd_output(self.screen, as_ansi=as_ansi, add_wrap_markers=add_wrap_markers)
self.screen.cmd_output(CommandOutput.last_run, lines.append, as_ansi, add_wrap_markers)
return ''.join(lines)