Python code coverage for Lib/lib-tk/tkFileDialog.py

#	count	content
1	n/a	#
2	n/a	# Instant Python
3	n/a	# $Id: tkFileDialog.py 36560 2004-07-18 06:16:08Z tim_one $
4	n/a	#
5	n/a	# tk common file dialogues
6	n/a	#
7	n/a	# this module provides interfaces to the native file dialogues
8	n/a	# available in Tk 4.2 and newer, and the directory dialogue available
9	n/a	# in Tk 8.3 and newer.
10	n/a	#
11	n/a	# written by Fredrik Lundh, May 1997.
12	n/a	#
13	n/a
14	n/a	#
15	n/a	# options (all have default values):
16	n/a	#
17	n/a	# - defaultextension: added to filename if not explicitly given
18	n/a	#
19	n/a	# - filetypes: sequence of (label, pattern) tuples. the same pattern
20	n/a	# may occur with several patterns. use "*" as pattern to indicate
21	n/a	# all files.
22	n/a	#
23	n/a	# - initialdir: initial directory. preserved by dialog instance.
24	n/a	#
25	n/a	# - initialfile: initial file (ignored by the open dialog). preserved
26	n/a	# by dialog instance.
27	n/a	#
28	n/a	# - parent: which window to place the dialog on top of
29	n/a	#
30	n/a	# - title: dialog title
31	n/a	#
32	n/a	# - multiple: if true user may select more than one file
33	n/a	#
34	n/a	# options for the directory chooser:
35	n/a	#
36	n/a	# - initialdir, parent, title: see above
37	n/a	#
38	n/a	# - mustexist: if true, user must pick an existing directory
39	n/a	#
40	n/a	#
41	n/a
42	n/a
43	n/a	from tkCommonDialog import Dialog
44	n/a
45	n/a	class _Dialog(Dialog):
46	n/a
47	n/a	def _fixoptions(self):
48	n/a	try:
49	n/a	# make sure "filetypes" is a tuple
50	n/a	self.options["filetypes"] = tuple(self.options["filetypes"])
51	n/a	except KeyError:
52	n/a	pass
53	n/a
54	n/a	def _fixresult(self, widget, result):
55	n/a	if result:
56	n/a	# keep directory and filename until next time
57	n/a	import os
58	n/a	# convert Tcl path objects to strings
59	n/a	try:
60	n/a	result = result.string
61	n/a	except AttributeError:
62	n/a	# it already is a string
63	n/a	pass
64	n/a	path, file = os.path.split(result)
65	n/a	self.options["initialdir"] = path
66	n/a	self.options["initialfile"] = file
67	n/a	self.filename = result # compatibility
68	n/a	return result
69	n/a
70	n/a
71	n/a	#
72	n/a	# file dialogs
73	n/a
74	n/a	class Open(_Dialog):
75	n/a	"Ask for a filename to open"
76	n/a
77	n/a	command = "tk_getOpenFile"
78	n/a
79	n/a	def _fixresult(self, widget, result):
80	n/a	if isinstance(result, tuple):
81	n/a	# multiple results:
82	n/a	result = tuple([getattr(r, "string", r) for r in result])
83	n/a	if result:
84	n/a	import os
85	n/a	path, file = os.path.split(result[0])
86	n/a	self.options["initialdir"] = path
87	n/a	# don't set initialfile or filename, as we have multiple of these
88	n/a	return result
89	n/a	if not widget.tk.wantobjects() and "multiple" in self.options:
90	n/a	# Need to split result explicitly
91	n/a	return self._fixresult(widget, widget.tk.splitlist(result))
92	n/a	return _Dialog._fixresult(self, widget, result)
93	n/a
94	n/a	class SaveAs(_Dialog):
95	n/a	"Ask for a filename to save as"
96	n/a
97	n/a	command = "tk_getSaveFile"
98	n/a
99	n/a
100	n/a	# the directory dialog has its own _fix routines.
101	n/a	class Directory(Dialog):
102	n/a	"Ask for a directory"
103	n/a
104	n/a	command = "tk_chooseDirectory"
105	n/a
106	n/a	def _fixresult(self, widget, result):
107	n/a	if result:
108	n/a	# convert Tcl path objects to strings
109	n/a	try:
110	n/a	result = result.string
111	n/a	except AttributeError:
112	n/a	# it already is a string
113	n/a	pass
114	n/a	# keep directory until next time
115	n/a	self.options["initialdir"] = result
116	n/a	self.directory = result # compatibility
117	n/a	return result
118	n/a
119	n/a	#
120	n/a	# convenience stuff
121	n/a
122	n/a	def askopenfilename(**options):
123	n/a	"Ask for a filename to open"
124	n/a
125	n/a	return Open(**options).show()
126	n/a
127	n/a	def asksaveasfilename(**options):
128	n/a	"Ask for a filename to save as"
129	n/a
130	n/a	return SaveAs(**options).show()
131	n/a
132	n/a	def askopenfilenames(**options):
133	n/a	"""Ask for multiple filenames to open
134	n/a
135	n/a	Returns a list of filenames or empty list if
136	n/a	cancel button selected
137	n/a	"""
138	n/a	options["multiple"]=1
139	n/a	return Open(**options).show()
140	n/a
141	n/a	# FIXME: are the following perhaps a bit too convenient?
142	n/a
143	n/a	def askopenfile(mode = "r", **options):
144	n/a	"Ask for a filename to open, and returned the opened file"
145	n/a
146	n/a	filename = Open(**options).show()
147	n/a	if filename:
148	n/a	return open(filename, mode)
149	n/a	return None
150	n/a
151	n/a	def askopenfiles(mode = "r", **options):
152	n/a	"""Ask for multiple filenames and return the open file
153	n/a	objects
154	n/a
155	n/a	returns a list of open file objects or an empty list if
156	n/a	cancel selected
157	n/a	"""
158	n/a
159	n/a	files = askopenfilenames(**options)
160	n/a	if files:
161	n/a	ofiles=[]
162	n/a	for filename in files:
163	n/a	ofiles.append(open(filename, mode))
164	n/a	files=ofiles
165	n/a	return files
166	n/a
167	n/a
168	n/a	def asksaveasfile(mode = "w", **options):
169	n/a	"Ask for a filename to save as, and returned the opened file"
170	n/a
171	n/a	filename = SaveAs(**options).show()
172	n/a	if filename:
173	n/a	return open(filename, mode)
174	n/a	return None
175	n/a
176	n/a	def askdirectory (**options):
177	n/a	"Ask for a directory, and return the file name"
178	n/a	return Directory(**options).show()
179	n/a
180	n/a	# --------------------------------------------------------------------
181	n/a	# test stuff
182	n/a
183	n/a	if __name__ == "__main__":
184	n/a	# Since the file name may contain non-ASCII characters, we need
185	n/a	# to find an encoding that likely supports the file name, and
186	n/a	# displays correctly on the terminal.
187	n/a
188	n/a	# Start off with UTF-8
189	n/a	enc = "utf-8"
190	n/a	import sys
191	n/a
192	n/a	# See whether CODESET is defined
193	n/a	try:
194	n/a	import locale
195	n/a	locale.setlocale(locale.LC_ALL,'')
196	n/a	enc = locale.nl_langinfo(locale.CODESET)
197	n/a	except (ImportError, AttributeError):
198	n/a	pass
199	n/a
200	n/a	# dialog for openening files
201	n/a
202	n/a	openfilename=askopenfilename(filetypes=[("all files", "*")])
203	n/a	try:
204	n/a	fp=open(openfilename,"r")
205	n/a	fp.close()
206	n/a	except:
207	n/a	print "Could not open File: "
208	n/a	print sys.exc_info()[1]
209	n/a
210	n/a	print "open", openfilename.encode(enc)
211	n/a
212	n/a	# dialog for saving files
213	n/a
214	n/a	saveasfilename=asksaveasfilename()
215	n/a	print "saveas", saveasfilename.encode(enc)