Fixed how we complete ~

2 files changed, 24 insertions, 12 deletions

diff --git a/cmd2.py b/cmd2.py
index fe55cd93..1c5a867e 100755
--- a/cmd2.py
+++ b/cmd2.py
@@ -323,8 +323,6 @@ def path_complete(text, line, begidx, endidx, dir_exe_only=False, dir_only=False
     if endidx == len(line) or (endidx < len(line) and line[endidx] != os.path.sep):
         add_trailing_sep_if_dir = True
 
-    add_sep_after_tilde = False
-
     # Readline places begidx after ~ and path separators (/) so we need to extract any directory
     # path that appears before the search text
     dirname = line[prev_space_index + 1:begidx]
@@ -342,8 +340,12 @@ def path_complete(text, line, begidx, endidx, dir_exe_only=False, dir_only=False
         if not dirname:
             dirname = os.getcwd()
         elif dirname == '~':
-            # If tilde was used without separator, add a separator after the tilde in the completions
-            add_sep_after_tilde = True
+            # If a ~ was used without a separator between text, then this is invalid
+            if text:
+                return []
+            # If only a ~ was entered, then complete it with a slash
+            else:
+                return [os.path.sep]
 
         # Build the search string
         search_str = os.path.join(dirname, text + '*')
@@ -376,9 +378,6 @@ def path_complete(text, line, begidx, endidx, dir_exe_only=False, dir_only=False
         # If it is a file and we are at the end of the line, then add a space
         if os.path.isfile(path_completions[0]) and endidx == len(line):
             completions[0] += ' '
-        # If tilde was expanded without a separator, prepend one
-        elif os.path.isdir(path_completions[0]) and add_sep_after_tilde:
-            completions[0] = os.path.sep + completions[0]
 
     completions.sort()
     return completions
diff --git a/tests/test_completion.py b/tests/test_completion.py
index a51a4e1f..697e68d4 100644
--- a/tests/test_completion.py
+++ b/tests/test_completion.py
@@ -382,16 +382,29 @@ def test_path_completion_doesnt_match_wildcards(request):
     # Currently path completion doesn't accept wildcards, so will always return empty results
     assert path_complete(text, line, begidx, endidx) == []
 
-def test_path_completion_user_expansion():
+def test_path_completion_just_tilde():
     # Run path with just a tilde
     text = ''
+    line = 'shell fake ~'
+    endidx = len(line)
+    begidx = endidx - len(text)
+    completions_tilde = path_complete(text, line, begidx, endidx)
+
+    # Path complete should return a slash
+    assert completions_tilde == [os.path.sep]
+
+def test_path_completion_user_expansion():
+    # Run path with a tilde and a slash
+    text = ''
     if sys.platform.startswith('win'):
-        line = 'shell dir ~{}'.format(text)
+        cmd = 'dir'
     else:
-        line = 'shell ls ~{}'.format(text)
+        cmd = 'ls'
+
+    line = 'shell {} ~{}'.format(cmd, os.path.sep)
     endidx = len(line)
     begidx = endidx - len(text)
-    completions_tilde = path_complete(text, line, begidx, endidx)
+    completions_tilde_slash = path_complete(text, line, begidx, endidx)
 
     # Run path complete on the user's home directory
     user_dir = os.path.expanduser('~')
@@ -404,7 +417,7 @@ def test_path_completion_user_expansion():
     completions_home = path_complete(text, line, begidx, endidx)
 
     # Verify that the results are the same in both cases
-    assert completions_tilde == completions_home
+    assert completions_tilde_slash == completions_home
 
 def test_path_completion_directories_only(request):
     test_dir = os.path.dirname(request.module.__file__)