1 files changed, 211 insertions, 0 deletions
diff --git a/Lib/lib-tk/FileDialog.py b/Lib/lib-tk/FileDialog.py
new file mode 100644
index 0000000..fb2ccd7
--- /dev/null
+++ b/Lib/lib-tk/FileDialog.py
@@ -0,0 +1,211 @@
+"""File selection dialog classes.
+
+Classes:
+
+- FileDialog
+- LoadFileDialog
+- SaveFileDialog
+
+XXX Bugs:
+
+- The fields are not labeled
+- Default doesn't have absolute pathname
+- Each FileDialog instance can be used only once
+- There is no easy way for an application to add widgets of its own
+
+"""
+
+from Tkinter import *
+from Dialog import Dialog
+
+ANCHOR = 'anchor'
+
+import os
+import fnmatch
+
+
+class FileDialog:
+
+    """Standard file selection dialog -- no checks on selected file.
+
+    Usage:
+
+        d = FileDialog(master)
+        file = d.go(directory, pattern, default)
+        if file is None: ...canceled...
+
+    """
+
+    title = "File Selection Dialog"
+
+    def __init__(self, master):
+	self.master = master
+	self.directory = None
+	self.top = Toplevel(master)
+	self.top.title(self.title)
+	self.filter = Entry(self.top)
+	self.filter.pack(fill=X)
+	self.filter.bind('<Return>', self.filter_command)
+	self.midframe = Frame(self.top)
+	self.midframe.pack(expand=YES, fill=BOTH)
+	self.dirs = Listbox(self.midframe)
+	self.dirs.pack(side=LEFT, expand=YES, fill=BOTH)
+	self.dirs.bind('<ButtonRelease-1>', self.dirs_select_event)
+	self.dirs.bind('<Double-ButtonRelease-1>', self.dirs_double_event)
+	self.files = Listbox(self.midframe)
+	self.files.pack(side=RIGHT, expand=YES, fill=BOTH)
+	self.files.bind('<ButtonRelease-1>', self.files_select_event)
+	self.files.bind('<Double-ButtonRelease-1>', self.files_double_event)
+	self.selection = Entry(self.top)
+	self.selection.pack(fill=X)
+	self.selection.bind('<Return>', self.ok_event)
+	self.botframe = Frame(self.top)
+	self.botframe.pack(fill=X)
+	self.ok_button = Button(self.botframe,
+				 text="OK",
+				 command=self.ok_command)
+	self.ok_button.pack(side=LEFT)
+	self.filter_button = Button(self.botframe,
+				    text="Filter",
+				    command=self.filter_command)
+	self.filter_button.pack(side=LEFT, expand=YES)
+	self.cancel_button = Button(self.botframe,
+				    text="Cancel",
+				    command=self.cancel_command)
+	self.cancel_button.pack(side=RIGHT)
+
+    def go(self, directory=os.curdir, pattern="*", default=""):
+	self.directory = directory
+	self.set_filter(directory, pattern)
+	self.set_selection(default)
+	self.selection.focus_set()
+	self.filter_command()
+	self.top.grab_set()
+	try:
+	    self.master.mainloop()
+	except SystemExit, how:
+	    self.top.destroy()
+	    return how
+
+    def dirs_double_event(self, event):
+##	self.dirs_select_event(event)
+	self.filter_command()
+
+    def dirs_select_event(self, event):
+	dir, pat = self.get_filter()
+	subdir = self.dirs.get(ANCHOR)
+	dir = os.path.normpath(os.path.join(self.directory, subdir))
+	self.set_filter(dir, pat)
+
+    def files_double_event(self, event):
+##	self.files_select_event(event)
+##	self.master.update_idletasks()
+	self.ok_command()
+
+    def files_select_event(self, event):
+	file = self.files.get(ANCHOR)
+	self.set_selection(file)
+
+    def ok_event(self, event):
+	self.ok_command()
+
+    def ok_command(self):
+	raise SystemExit, self.selection.get()
+
+    def filter_command(self, event=None):
+	dir, pat = self.get_filter()
+	try:
+	    names = os.listdir(dir)
+	except os.error:
+	    self.master.bell()
+	    return
+	self.directory = dir
+	self.set_filter(dir, pat)
+	names.sort()
+	subdirs = [os.pardir]
+	matchingfiles = []
+	for name in names:
+	    fullname = os.path.join(dir, name)
+	    if os.path.isdir(fullname):
+		subdirs.append(name)
+	    elif fnmatch.fnmatch(name, pat):
+		matchingfiles.append(name)
+	self.dirs.delete(0, END)
+	for name in subdirs:
+	    self.dirs.insert(END, name)
+	self.files.delete(0, END)
+	for name in matchingfiles:
+	    self.files.insert(END, name)
+
+    def get_filter(self):
+	filter = self.filter.get()
+	if filter[-1:] == os.sep:
+	    filter = filter + "*"
+	return os.path.split(filter)
+
+    def cancel_command(self):
+	raise SystemExit, None
+
+    def set_filter(self, dir, pat):
+	self.filter.delete(0, END)
+	self.filter.insert(END, os.path.join(dir or os.curdir, pat or "*"))
+
+    def set_selection(self, file):
+	self.selection.delete(0, END)
+	self.selection.insert(END, os.path.join(self.directory, file))
+
+
+class LoadFileDialog(FileDialog):
+
+    """File selection dialog which checks that the file exists."""
+
+    title = "Load File Selection Dialog"
+
+    def ok_command(self):
+	file = self.selection.get()
+	if not os.path.isfile(file):
+	    self.master.bell()
+	else:
+	    raise SystemExit, file
+
+
+class SaveFileDialog(FileDialog):
+
+    """File selection dialog which checks that the file may be created."""
+
+    title = "Save File Selection Dialog"
+
+    def ok_command(self):
+	file = self.selection.get()
+	if os.path.exists(file):
+	    if os.path.isdir(file):
+		self.master.bell()
+		return
+	    d = Dialog(self.master,
+		       title="Overwrite Existing File Question",
+		       text="Overwrite existing file %s?" % `file`,
+		       bitmap='questhead',
+		       default=0,
+		       strings=("Yes", "Cancel"))
+	    if d.num != 0: file = None
+	else:
+	    head, tail = os.path.split(file)
+	    if not os.path.isdir(head):
+		self.master.bell()
+		return
+	raise SystemExit, file
+
+
+def test():
+    """Simple test program."""
+    root = Tk()
+    root.withdraw()
+    fd = LoadFileDialog(root)
+    loadfile = fd.go()
+    fd = SaveFileDialog(root)
+    savefile = fd.go()
+    print loadfile, savefile
+
+
+if __name__ == '__main__':
+    test()