diff options
Diffstat (limited to 'examples')
| -rwxr-xr-x | examples/paged_output.py | 42 |
1 files changed, 30 insertions, 12 deletions
diff --git a/examples/paged_output.py b/examples/paged_output.py index 72efd4f5..d1b1b2c2 100755 --- a/examples/paged_output.py +++ b/examples/paged_output.py @@ -14,25 +14,43 @@ class PagedOutput(cmd2.Cmd): def __init__(self): super().__init__() + def page_file(self, file_path: str, chop: bool=False): + """Helper method to prevent having too much duplicated code.""" + filename = os.path.expanduser(file_path) + try: + with open(filename, 'r') as f: + text = f.read() + self.ppaged(text, chop=chop) + except FileNotFoundError as ex: + self.perror('ERROR: file {!r} not found'.format(filename), traceback_war=False) + @cmd2.with_argument_list - def do_page_file(self, args: List[str]): - """Read in a text file and display its output in a pager. + def do_page_wrap(self, args: List[str]): + """Read in a text file and display its output in a pager, wrapping long lines if they don't fit. - Usage: page_file <file_path> + Usage: page_wrap <file_path> """ if not args: - self.perror('page_file requires a path to a file as an argument', traceback_war=False) + self.perror('page_wrap requires a path to a file as an argument', traceback_war=False) return + self.page_file(args[0], chop=False) - filename = os.path.expanduser(args[0]) - try: - with open(filename, 'r') as f: - text = f.read() - self.ppaged(text) - except FileNotFoundError as ex: - self.perror('ERROR: file {!r} not found'.format(filename), traceback_war=False) + complete_page_wrap = cmd2.Cmd.path_complete + + @cmd2.with_argument_list + def do_page_truncate(self, args: List[str]): + """Read in a text file and display its output in a pager, truncating long lines if they don't fit. + + Truncated lines can still be accessed by scrolling to the right using the arrow keys. + + Usage: page_chop <file_path> + """ + if not args: + self.perror('page_truncate requires a path to a file as an argument', traceback_war=False) + return + self.page_file(args[0], chop=True) - complete_page_file = cmd2.Cmd.path_complete + complete_page_truncate = cmd2.Cmd.path_complete if __name__ == '__main__': |