1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# Create hierarchical menus for some volumes.
import os
from Menu import *
# Since we can't (yet) list the mounted volumes, here's a list of some:
my_volumes = ['C:', 'D:', 'E:', 'F:']
def main():
global oldbar
oldbar = GetMenuBar()
ClearMenuBar()
makevolmenus(my_volumes)
DrawMenuBar()
def reset():
oldbar.SetMenuBar()
DrawMenuBar()
id = 1
def nextid():
global id
nid = id
id = id+1
return nid
def makevolmenus(volumes):
for vol in volumes:
makevolmenu(vol)
def makevolmenu(vol):
menu = NewMenu(nextid(), vol)
adddirectory(menu, vol)
menu.InsertMenu(0)
def adddirectory(menu, dir, maxdepth = 1):
print "adddirectory:", `dir`, maxdepth
files = os.listdir(dir)
item = 0
for file in files:
item = item+1
menu.AppendMenu('x') # add a dummy string
menu.SetItem(item, file) # set the actual text
fullname = os.path.join(dir, file)
if os.path.isdir(fullname):
menu.SetItem(item, ':' + file + ':') # append colons
if maxdepth > 0:
id = nextid()
submenu = NewMenu(id, fullname)
adddirectory(submenu, fullname, maxdepth-1)
submenu.InsertMenu(-1)
# If the 'Cmd' is 0x1B, then the 'Mark' is the submenu id
menu.SetItemMark(item, id)
menu.SetItemCmd(item, 0x1B)
if not files:
menu.AppendMenu(':') # dummy item to make it selectable
return menu
if __name__ == '__main__':
main()
|
