Mercurial > hg > config
annotate python/dmenu.py @ 498:95ba5770d2f0
file decorator
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Wed, 21 Aug 2013 15:54:09 -0700 |
parents | 618db1b1f34b |
children |
rev | line source |
---|---|
289 | 1 #!/usr/bin/env python |
2 | |
3 import optparse | |
4 import os | |
297 | 5 import shlex |
289 | 6 import subprocess |
7 import sys | |
8 | |
290
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
9 def choose_file(directory, dmenu='dmenu', |
291 | 10 args=('-i', '-nb', 'black', '-nf', 'white', |
11 '-fn', '-*-lucidatypewriter-medium-r-*-*-*-120-*-*-*-*-*-*')): | |
289 | 12 """choose a file in the directory with dmenu""" |
13 directory = os.path.abspath(directory) | |
14 files = os.listdir(directory) | |
15 string = '\n'.join(files) | |
16 | |
290
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
17 if isinstance(dmenu, basestring): |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
18 dmenu = [dmenu] |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
19 dmenu = list(dmenu) |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
20 dmenu.extend(args) |
289 | 21 |
290
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
22 process = subprocess.Popen(dmenu, stdin=subprocess.PIPE, stdout=subprocess.PIPE) |
289 | 23 stdout, _ = process.communicate(input=string) |
24 if process.returncode: | |
25 return | |
26 chosen = os.path.join(directory, stdout) | |
27 if os.path.isdir(chosen): | |
28 return choose_file(chosen) | |
29 return chosen | |
30 | |
31 def main(args=sys.argv[1:]): | |
32 parser = optparse.OptionParser() | |
290
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
33 parser.add_option('-d', '--directory', dest='directory', |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
34 default=os.getcwd(), |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
35 help="call on this directory [Default: current directory]") |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
36 parser.add_option('-e', '--exec', dest='executable', |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
37 help="call this proram with the result") |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
38 options, args = parser.parse_args(args) |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
39 chosen = choose_file(options.directory) |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
40 if chosen: |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
41 if options.executable: |
297 | 42 # TODO: extract this pattern (see ims journal.txt) |
43 command = shlex.split(options.executable) | |
44 executable = command[0] | |
45 command.append(chosen) | |
46 os.execlp(executable, *command) | |
47 else: | |
48 print chosen | |
290
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
49 else: |
03d62a2cbe62
update demnu to almost a real program
Jeff Hammel <jhammel@mozilla.com>
parents:
289
diff
changeset
|
50 sys.exit(1) |
289 | 51 |
52 if __name__ == '__main__': | |
53 main() |