| 1 | """A directory browser using Ttk Treeview.
|
|---|
| 2 |
|
|---|
| 3 | Based on the demo found in Tk 8.5 library/demos/browse
|
|---|
| 4 | """
|
|---|
| 5 | import os
|
|---|
| 6 | import glob
|
|---|
| 7 | import Tkinter
|
|---|
| 8 | import ttk
|
|---|
| 9 |
|
|---|
| 10 | def populate_tree(tree, node):
|
|---|
| 11 | if tree.set(node, "type") != 'directory':
|
|---|
| 12 | return
|
|---|
| 13 |
|
|---|
| 14 | path = tree.set(node, "fullpath")
|
|---|
| 15 | tree.delete(*tree.get_children(node))
|
|---|
| 16 |
|
|---|
| 17 | parent = tree.parent(node)
|
|---|
| 18 | special_dirs = [] if parent else glob.glob('.') + glob.glob('..')
|
|---|
| 19 |
|
|---|
| 20 | for p in special_dirs + os.listdir(path):
|
|---|
| 21 | ptype = None
|
|---|
| 22 | p = os.path.join(path, p).replace('\\', '/')
|
|---|
| 23 | if os.path.isdir(p): ptype = "directory"
|
|---|
| 24 | elif os.path.isfile(p): ptype = "file"
|
|---|
| 25 |
|
|---|
| 26 | fname = os.path.split(p)[1]
|
|---|
| 27 | id = tree.insert(node, "end", text=fname, values=[p, ptype])
|
|---|
| 28 |
|
|---|
| 29 | if ptype == 'directory':
|
|---|
| 30 | if fname not in ('.', '..'):
|
|---|
| 31 | tree.insert(id, 0, text="dummy")
|
|---|
| 32 | tree.item(id, text=fname)
|
|---|
| 33 | elif ptype == 'file':
|
|---|
| 34 | size = os.stat(p).st_size
|
|---|
| 35 | tree.set(id, "size", "%d bytes" % size)
|
|---|
| 36 |
|
|---|
| 37 |
|
|---|
| 38 | def populate_roots(tree):
|
|---|
| 39 | dir = os.path.abspath('.').replace('\\', '/')
|
|---|
| 40 | node = tree.insert('', 'end', text=dir, values=[dir, "directory"])
|
|---|
| 41 | populate_tree(tree, node)
|
|---|
| 42 |
|
|---|
| 43 | def update_tree(event):
|
|---|
| 44 | tree = event.widget
|
|---|
| 45 | populate_tree(tree, tree.focus())
|
|---|
| 46 |
|
|---|
| 47 | def change_dir(event):
|
|---|
| 48 | tree = event.widget
|
|---|
| 49 | node = tree.focus()
|
|---|
| 50 | if tree.parent(node):
|
|---|
| 51 | path = os.path.abspath(tree.set(node, "fullpath"))
|
|---|
| 52 | if os.path.isdir(path):
|
|---|
| 53 | os.chdir(path)
|
|---|
| 54 | tree.delete(tree.get_children(''))
|
|---|
| 55 | populate_roots(tree)
|
|---|
| 56 |
|
|---|
| 57 | def autoscroll(sbar, first, last):
|
|---|
| 58 | """Hide and show scrollbar as needed."""
|
|---|
| 59 | first, last = float(first), float(last)
|
|---|
| 60 | if first <= 0 and last >= 1:
|
|---|
| 61 | sbar.grid_remove()
|
|---|
| 62 | else:
|
|---|
| 63 | sbar.grid()
|
|---|
| 64 | sbar.set(first, last)
|
|---|
| 65 |
|
|---|
| 66 | root = Tkinter.Tk()
|
|---|
| 67 |
|
|---|
| 68 | vsb = ttk.Scrollbar(orient="vertical")
|
|---|
| 69 | hsb = ttk.Scrollbar(orient="horizontal")
|
|---|
| 70 |
|
|---|
| 71 | tree = ttk.Treeview(columns=("fullpath", "type", "size"),
|
|---|
| 72 | displaycolumns="size", yscrollcommand=lambda f, l: autoscroll(vsb, f, l),
|
|---|
| 73 | xscrollcommand=lambda f, l:autoscroll(hsb, f, l))
|
|---|
| 74 |
|
|---|
| 75 | vsb['command'] = tree.yview
|
|---|
| 76 | hsb['command'] = tree.xview
|
|---|
| 77 |
|
|---|
| 78 | tree.heading("#0", text="Directory Structure", anchor='w')
|
|---|
| 79 | tree.heading("size", text="File Size", anchor='w')
|
|---|
| 80 | tree.column("size", stretch=0, width=100)
|
|---|
| 81 |
|
|---|
| 82 | populate_roots(tree)
|
|---|
| 83 | tree.bind('<<TreeviewOpen>>', update_tree)
|
|---|
| 84 | tree.bind('<Double-Button-1>', change_dir)
|
|---|
| 85 |
|
|---|
| 86 | # Arrange the tree and its scrollbars in the toplevel
|
|---|
| 87 | tree.grid(column=0, row=0, sticky='nswe')
|
|---|
| 88 | vsb.grid(column=1, row=0, sticky='ns')
|
|---|
| 89 | hsb.grid(column=0, row=1, sticky='ew')
|
|---|
| 90 | root.grid_columnconfigure(0, weight=1)
|
|---|
| 91 | root.grid_rowconfigure(0, weight=1)
|
|---|
| 92 |
|
|---|
| 93 | root.mainloop()
|
|---|
