From 272864e850e6dc3be353ca940510b0b4040b2291 Mon Sep 17 00:00:00 2001
From: Todd Leonhardt <todd.leonhardt@gmail.com>
Date: Tue, 16 May 2017 09:08:40 -0400
Subject: Added path_complete() static method for local file system path
 completion

The edit, load, and shell commands now all use this for their tab completion.

Additionally, end users can easily wire it up for use with any of their custom commands.
---
 cmd2.py | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

(limited to 'cmd2.py')

diff --git a/cmd2.py b/cmd2.py
index 3504d7d4..e49fe15d 100755
--- a/cmd2.py
+++ b/cmd2.py
@@ -1327,6 +1327,38 @@ class Cmd(cmd.Cmd):
     Usage:  shell cmd"""
         self.stdout.write("{}\n".format(help_str))
 
+    @staticmethod
+    def path_complete(line):
+        """Method called to complete an input line by local file system path completion.
+
+        :param line: str - the current input line with leading whitespace removed
+        :return: List[str] - a list of possible tab completions
+        """
+        path = line.split()[-1]
+        if not path:
+            path = '.'
+
+        dirname, rest = os.path.split(path)
+        real_dir = os.path.expanduser(dirname)
+
+        path_completions = glob.glob(os.path.join(real_dir, rest) + '*')
+
+        # Strip off everything but the final part of the completion
+        completions = [os.path.basename(c) for c in path_completions]
+        return completions
+
+    # noinspection PyUnusedLocal
+    def complete_shell(self, text, line, begidx, endidx):
+        """Handles completion of arguments for the shell command.
+
+        :param text: str - the string prefix we are attempting to match (all returned matches must begin with it)
+        :param line: str - the current input line with leading whitespace removed
+        :param begidx: str - the beginning indexe of the prefix text
+        :param endidx: str - the ending index of the prefix text
+        :return: List[str] - a list of possible tab completions
+        """
+        return self.path_complete(line)
+
     def do_py(self, arg):
         """
         py <command>: Executes a Python command.
@@ -1551,6 +1583,18 @@ Edited files are run on close if the `autorun_on_edit` settable parameter is Tru
                   pyparsing.Optional(pyparsing.Word(legalChars + '/\\'))("fname") +
                   pyparsing.stringEnd)
 
+    # noinspection PyUnusedLocal
+    def complete_edit(self, text, line, begidx, endidx):
+        """Handles completion of arguments for the edit command.
+
+        :param text: str - the string prefix we are attempting to match (all returned matches must begin with it)
+        :param line: str - the current input line with leading whitespace removed
+        :param begidx: str - the beginning indexe of the prefix text
+        :param endidx: str - the ending index of the prefix text
+        :return: List[str] - a list of possible tab completions
+        """
+        return self.path_complete(line)
+
     def do_save(self, arg):
         """Saves command(s) from history to file.
 
@@ -1688,6 +1732,18 @@ relative to the already-running script's directory.
 Script should contain one command per line, just like command would be typed in console."""
         self.stdout.write("{}\n".format(help_str))
 
+    # noinspection PyUnusedLocal
+    def complete_load(self, text, line, begidx, endidx):
+        """Handles completion of arguments for the load command.
+
+        :param text: str - the string prefix we are attempting to match (all returned matches must begin with it)
+        :param line: str - the current input line with leading whitespace removed
+        :param begidx: str - the beginning indexe of the prefix text
+        :param endidx: str - the ending index of the prefix text
+        :return: List[str] - a list of possible tab completions
+        """
+        return self.path_complete(line)
+
     def do_run(self, arg):
         """run [arg]: re-runs an earlier command
 
-- 
cgit v1.2.1