Improve python version detection.

9 years ago · 1a20ff7194
parent 845f5f4aa6
commit 1a20ff7194
1 changed files with 3 additions and 2 deletions
--- a/fish_prompt.fish
+++ b/fish_prompt.fish
@ -371,8 +371,9 @@ end

 function __bobthefish_virtualenv_python_version -S -d 'Get current python version'
  set -l python_version (readlink (which python))
-  switch "$python_version"
-    case 'python2*'
+  [ -z "$python_version" ]; and set python_version (which python)
+  switch (basename "$python_version")
+    case 'python' 'python2*'
      echo $__bobthefish_superscript_glyph[2]
    case 'python3*'
      echo $__bobthefish_superscript_glyph[3]