mirror of https://github.com/python/cpython
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
"""A directory browser using Ttk Treeview.
|
|
|
|
Based on the demo found in Tk 8.5 library/demos/browse
|
|
"""
|
|
import os
|
|
import glob
|
|
import Tkinter
|
|
import ttk
|
|
|
|
def populate_tree(tree, node):
|
|
if tree.set(node, "type") != 'directory':
|
|
return
|
|
|
|
path = tree.set(node, "fullpath")
|
|
tree.delete(*tree.get_children(node))
|
|
|
|
parent = tree.parent(node)
|
|
special_dirs = [] if parent else glob.glob('.') + glob.glob('..')
|
|
|
|
for p in special_dirs + os.listdir(path):
|
|
ptype = None
|
|
p = os.path.join(path, p).replace('\\', '/')
|
|
if os.path.isdir(p): ptype = "directory"
|
|
elif os.path.isfile(p): ptype = "file"
|
|
|
|
fname = os.path.split(p)[1]
|
|
id = tree.insert(node, "end", text=fname, values=[p, ptype])
|
|
|
|
if ptype == 'directory':
|
|
if fname not in ('.', '..'):
|
|
tree.insert(id, 0, text="dummy")
|
|
tree.item(id, text=fname)
|
|
elif ptype == 'file':
|
|
size = os.stat(p).st_size
|
|
tree.set(id, "size", "%d bytes" % size)
|
|
|
|
|
|
def populate_roots(tree):
|
|
dir = os.path.abspath('.').replace('\\', '/')
|
|
node = tree.insert('', 'end', text=dir, values=[dir, "directory"])
|
|
populate_tree(tree, node)
|
|
|
|
def update_tree(event):
|
|
tree = event.widget
|
|
populate_tree(tree, tree.focus())
|
|
|
|
def change_dir(event):
|
|
tree = event.widget
|
|
node = tree.focus()
|
|
if tree.parent(node):
|
|
path = os.path.abspath(tree.set(node, "fullpath"))
|
|
if os.path.isdir(path):
|
|
os.chdir(path)
|
|
tree.delete(tree.get_children(''))
|
|
populate_roots(tree)
|
|
|
|
def autoscroll(sbar, first, last):
|
|
"""Hide and show scrollbar as needed."""
|
|
first, last = float(first), float(last)
|
|
if first <= 0 and last >= 1:
|
|
sbar.grid_remove()
|
|
else:
|
|
sbar.grid()
|
|
sbar.set(first, last)
|
|
|
|
root = Tkinter.Tk()
|
|
|
|
vsb = ttk.Scrollbar(orient="vertical")
|
|
hsb = ttk.Scrollbar(orient="horizontal")
|
|
|
|
tree = ttk.Treeview(columns=("fullpath", "type", "size"),
|
|
displaycolumns="size", yscrollcommand=lambda f, l: autoscroll(vsb, f, l),
|
|
xscrollcommand=lambda f, l:autoscroll(hsb, f, l))
|
|
|
|
vsb['command'] = tree.yview
|
|
hsb['command'] = tree.xview
|
|
|
|
tree.heading("#0", text="Directory Structure", anchor='w')
|
|
tree.heading("size", text="File Size", anchor='w')
|
|
tree.column("size", stretch=0, width=100)
|
|
|
|
populate_roots(tree)
|
|
tree.bind('<<TreeviewOpen>>', update_tree)
|
|
tree.bind('<Double-Button-1>', change_dir)
|
|
|
|
# Arrange the tree and its scrollbars in the toplevel
|
|
tree.grid(column=0, row=0, sticky='nswe')
|
|
vsb.grid(column=1, row=0, sticky='ns')
|
|
hsb.grid(column=0, row=1, sticky='ew')
|
|
root.grid_columnconfigure(0, weight=1)
|
|
root.grid_rowconfigure(0, weight=1)
|
|
|
|
root.mainloop()
|