##// END OF EJS Templates
Fix local variable error.
fptest -
Show More
@@ -1,133 +1,133 b''
1 1 """ Tab completion support for a couple of linux package managers
2 2
3 3 This is also an example of how to write custom completer plugins
4 4 or hooks.
5 5
6 6 Practical use:
7 7
8 8 [ipython]|1> import ipy_linux_package_managers
9 9 [ipython]|2> apt-get u<<< press tab here >>>
10 10 update upgrade
11 11 [ipython]|2> apt-get up
12 12
13 13 """
14 14 import IPython.ipapi
15 15 import glob,os,shlex,sys
16 16
17 17 ip = IPython.ipapi.get()
18 18
19 19 def apt_completers(self, event):
20 20 """ This should return a list of strings with possible completions.
21 21
22 22 Note that all the included strings that don't start with event.symbol
23 23 are removed, in order to not confuse readline.
24 24
25 25 """
26 26 # print event # dbg
27 27
28 28 # commands are only suggested for the 'command' part of package manager
29 29 # invocation
30 30
31 31 cmd = (event.line + "<placeholder>").rsplit(None,1)[0]
32 32 # print cmd
33 33 if cmd.endswith('apt-get') or cmd.endswith('yum'):
34 34 return ['update', 'upgrade', 'install', 'remove']
35 35
36 36 # later on, add dpkg -l / whatever to get list of possible
37 37 # packages, add switches etc. for the rest of command line
38 38 # filling
39 39
40 40 raise IPython.ipapi.TryNext
41 41
42 42
43 43 # re_key specifies the regexp that triggers the specified completer
44 44
45 45 ip.set_hook('complete_command', apt_completers, re_key = '.*apt-get')
46 46 ip.set_hook('complete_command', apt_completers, re_key = '.*yum')
47 47
48 48 pkg_cache = None
49 49
50 50 def module_completer(self,event):
51 51 """ Give completions after user has typed 'import' """
52 52
53 53 # only a local version for py 2.4, pkgutil has no walk_packages() there
54 54 if sys.version_info < (2,5):
55 55 for el in [f[:-3] for f in glob.glob("*.py")]:
56 56 yield el
57 57 return
58 58
59 59 global pkg_cache
60 60 import pkgutil,imp,time
61 61 #current =
62 62 if pkg_cache is None:
63 63 print "\n\n[Standby while scanning modules, this can take a while]\n\n"
64 64 pkg_cache = list(pkgutil.walk_packages())
65 65
66 66 already = set()
67 67 for ld, name, ispkg in pkg_cache:
68 68 if name.count('.') < event.symbol.count('.') + 1:
69 69 if name not in already:
70 70 already.add(name)
71 71 yield name + (ispkg and '.' or '')
72 72 return
73 73
74 74 ip.set_hook('complete_command', module_completer, str_key = 'import')
75 75 ip.set_hook('complete_command', module_completer, str_key = 'from')
76 76
77 77 svn_commands = """\
78 78 add blame praise annotate ann cat checkout co cleanup commit ci copy
79 79 cp delete del remove rm diff di export help ? h import info list ls
80 80 lock log merge mkdir move mv rename ren propdel pdel pd propedit pedit
81 81 pe propget pget pg proplist plist pl propset pset ps resolved revert
82 82 status stat st switch sw unlock update
83 83 """
84 84
85 85 def svn_completer(self,event):
86 86 if len((event.line + 'placeholder').split()) > 2:
87 87 # the rest are probably file names
88 88 return ip.IP.Completer.file_matches(event.symbol)
89 89
90 90 return svn_commands.split()
91 91
92 92 ip.set_hook('complete_command', svn_completer, str_key = 'svn')
93 93
94 94 def runlistpy(self, event):
95 95 comps = shlex.split(event.line)
96 96 relpath = (len(comps) > 1 and comps[-1] or '')
97 97
98 98 #print "rp",relpath # dbg
99 glob = glob.glob
99 lglob = glob.glob
100 100 isdir = os.path.isdir
101 101 if relpath.startswith('~'):
102 102 relpath = os.path.expanduser(relpath)
103 dirs = [f.replace('\\','/') + "/" for f in glob(relpath+'*')
103 dirs = [f.replace('\\','/') + "/" for f in lglob(relpath+'*')
104 104 if isdir(f)]
105 pys = [f.replace('\\','/') for f in glob(relpath+'*.py')]
105 pys = [f.replace('\\','/') for f in lglob(relpath+'*.py')]
106 106 return dirs + pys
107 107
108 108 ip.set_hook('complete_command', runlistpy, str_key = '%run')
109 109
110 110 def cd_completer(self, event):
111 111 relpath = event.symbol
112 112
113 113 if '-b' in event.line:
114 114 # return only bookmark completions
115 115 bkms = self.db.get('bookmarks',{})
116 116 return bkms.keys()
117 117
118 118 if event.symbol == '-':
119 119 # jump in directory history by number
120 120 ents = ['-%d [%s]' % (i,s) for i,s in enumerate(ip.user_ns['_dh'])]
121 121 if len(ents) > 1:
122 122 return ents
123 123 return []
124 124
125 125
126 126 if relpath.startswith('~'):
127 127 relpath = os.path.expanduser(relpath).replace('\\','/')
128 128 found = [f.replace('\\','/')+'/' for f in glob.glob(relpath+'*') if os.path.isdir(f)]
129 129 if not found:
130 130 return [relpath]
131 131 return found
132 132
133 133 ip.set_hook('complete_command', cd_completer, str_key = '%cd')
General Comments 0
You need to be logged in to leave comments. Login now