##// END OF EJS Templates
bash completion: ipython <tab> shows only subcmds...
Paul Ivanov -
Show More
@@ -1,113 +1,113 b''
1 1 # load with: . ipython-completion.bash
2 2
3 3 if [[ -n ${ZSH_VERSION-} ]]; then
4 4 autoload -Uz bashcompinit && bashcompinit
5 5 fi
6 6
7 7 _ipython_get_flags()
8 8 {
9 9 local url=$1
10 10 local var=$2
11 11 local dash=$3
12 12 if [[ "$url $var" == $__ipython_complete_last ]]; then
13 13 opts=$__ipython_complete_last_res
14 14 return
15 15 fi
16 16 # pylab and profile don't need the = and the
17 17 # version without simplifies the special cased completion
18 18 opts=$(ipython ${url} --help-all | grep -E "^-{1,2}[^-]" | sed -e "s/<.*//" -e "s/[^=]$/& /" -e "s/^--pylab=$//" -e "s/^--profile=$/--profile /")
19 19 __ipython_complete_last="$url $var"
20 20 __ipython_complete_last_res="$opts"
21 21 }
22 22
23 23 _ipython()
24 24 {
25 25 local cur=${COMP_WORDS[COMP_CWORD]}
26 26 local prev=${COMP_WORDS[COMP_CWORD - 1]}
27 27 local subcommands="notebook qtconsole console kernel profile locate history nbconvert"
28 28 local opts=""
29 29 if [ -z "$__ipython_complete_baseopts" ]; then
30 30 _ipython_get_flags baseopts
31 31 __ipython_complete_baseopts="${opts}"
32 32 fi
33 33 local baseopts="$__ipython_complete_baseopts"
34 34 local mode=""
35 35 for i in "${COMP_WORDS[@]}"; do
36 36 [ "$cur" = "$i" ] && break
37 37 if [[ ${subcommands} == *${i}* ]]; then
38 38 mode="$i"
39 39 break
40 40 elif [[ ${i} == "--"* ]]; then
41 41 mode="nosubcommand"
42 42 break
43 43 fi
44 44 done
45 45
46 46
47 47 if [[ ${cur} == -* ]]; then
48 48 case $mode in
49 49 "notebook" | "qtconsole" | "console" | "kernel")
50 50 _ipython_get_flags $mode
51 51 opts=$"${opts} ${baseopts}"
52 52 ;;
53 53 "locate" | "history" | "profile")
54 54 _ipython_get_flags $mode
55 55 opts=$"${opts}"
56 56 ;;
57 57 *)
58 58 opts=$baseopts
59 59 esac
60 60 # don't drop the trailing space
61 61 local IFS=$'\t\n'
62 62 COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
63 63 return 0
64 64 elif [[ $mode == "profile" ]]; then
65 65 opts="list create locate"
66 66 COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
67 67 elif [[ $mode == "history" ]]; then
68 68 opts="trim"
69 69 COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
70 70 elif [[ $mode == "locate" ]]; then
71 71 opts="profile"
72 72 COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
73 73 elif [[ ${prev} == "--pylab"* ]] || [[ ${prev} == "--gui"* ]]; then
74 74 if [ -z "$__ipython_complete_pylab" ]; then
75 75 __ipython_complete_pylab=`cat <<EOF | python -
76 76 try:
77 77 import IPython.core.shellapp as mod;
78 78 for k in mod.InteractiveShellApp.pylab.values:
79 79 print "%s " % k
80 80 except:
81 81 pass
82 82 EOF
83 83 `
84 84 fi
85 85 local IFS=$'\t\n'
86 86 COMPREPLY=( $(compgen -W "${__ipython_complete_pylab}" -- ${cur}) )
87 87 elif [[ ${prev} == "--profile"* ]]; then
88 88 if [ -z "$__ipython_complete_profiles" ]; then
89 89 __ipython_complete_profiles=`cat <<EOF | python -
90 90 try:
91 91 import IPython.core.profileapp
92 92 for k in IPython.core.profileapp.list_bundled_profiles():
93 93 print "%s " % k
94 94 p = IPython.core.profileapp.ProfileList()
95 95 for k in IPython.core.profileapp.list_profiles_in(p.ipython_dir):
96 96 print "%s " % k
97 97 except:
98 98 pass
99 99 EOF
100 100 `
101 101 fi
102 102 local IFS=$'\t\n'
103 103 COMPREPLY=( $(compgen -W "${__ipython_complete_profiles}" -- ${cur}) )
104 104 else
105 if [ -z "$mode" ]; then
106 COMPREPLY=( $(compgen -f -W "${subcommands}" -- ${cur}) )
105 if [ "$COMP_CWORD" == 1 ]; then
106 COMPREPLY=( $(compgen -W "${subcommands}" -- ${cur}) )
107 107 else
108 108 COMPREPLY=( $(compgen -f -- ${cur}) )
109 109 fi
110 110 fi
111 111
112 112 }
113 113 complete -o default -o nospace -F _ipython ipython
General Comments 0
You need to be logged in to leave comments. Login now