Python code coverage for Mac/Demo/mlte/mlted.py

#	count	content
1	n/a	# A minimal text editor using MLTE. Based on wed.py.
2	n/a	#
3	n/a	# To be done:
4	n/a	# - Functionality: find, etc.
5	n/a
6	n/a	from Menu import DrawMenuBar
7	n/a	from FrameWork import *
8	n/a	from Carbon import Win
9	n/a	from Carbon import Ctl
10	n/a	from Carbon import Qd
11	n/a	from Carbon import Res
12	n/a	from Carbon import Scrap
13	n/a	import os
14	n/a	from Carbon import MacTextEditor
15	n/a	from Carbon import Mlte
16	n/a
17	n/a	UNDOLABELS = [ # Indexed by MLTECanUndo() value
18	n/a	"Typing", "Cut", "Paste", "Clear", "Font Change", "Color Change", "Size Change",
19	n/a	"Style Change", "Align Left", "Align Center", "Align Right", "Drop", "Move"]
20	n/a
21	n/a	class MlteWindow(Window):
22	n/a	def open(self, path, name, data):
23	n/a	self.path = path
24	n/a	self.name = name
25	n/a	r = windowbounds(400, 400)
26	n/a	w = Win.NewWindow(r, name, 1, 0, -1, 1, 0)
27	n/a	self.wid = w
28	n/a	flags = MacTextEditor.kTXNDrawGrowIconMask\|MacTextEditor.kTXNWantHScrollBarMask\| \
29	n/a	MacTextEditor.kTXNWantVScrollBarMask
30	n/a	self.ted, self.frameid = Mlte.TXNNewObject(None, w, None, flags, MacTextEditor.kTXNTextEditStyleFrameType,
31	n/a	MacTextEditor.kTXNTextFile, MacTextEditor.kTXNMacOSEncoding)
32	n/a	self.ted.TXNSetData(MacTextEditor.kTXNTextData, data, 0, 0x7fffffff)
33	n/a	self.changed = 0
34	n/a	self.do_postopen()
35	n/a	self.do_activate(1, None)
36	n/a
37	n/a	def do_idle(self, event):
38	n/a	self.ted.TXNIdle()
39	n/a	self.ted.TXNAdjustCursor(None)
40	n/a
41	n/a
42	n/a
43	n/a	def do_activate(self, onoff, evt):
44	n/a	if onoff:
45	n/a	## self.ted.TXNActivate(self.frameid, 0)
46	n/a	self.ted.TXNFocus(1)
47	n/a	self.parent.active = self
48	n/a	else:
49	n/a	self.ted.TXNFocus(0)
50	n/a	self.parent.active = None
51	n/a	self.parent.updatemenubar()
52	n/a
53	n/a	def do_update(self, wid, event):
54	n/a	self.ted.TXNDraw(None)
55	n/a
56	n/a	def do_postresize(self, width, height, window):
57	n/a	self.ted.TXNResizeFrame(width, height, self.frameid)
58	n/a
59	n/a	def do_contentclick(self, local, modifiers, evt):
60	n/a	self.ted.TXNClick(evt)
61	n/a	self.parent.updatemenubar()
62	n/a
63	n/a	def do_char(self, ch, event):
64	n/a	self.ted.TXNKeyDown(event)
65	n/a	self.parent.updatemenubar()
66	n/a
67	n/a	def close(self):
68	n/a	if self.changed:
69	n/a	save = EasyDialogs.AskYesNoCancel('Save window "%s" before closing?'%self.name, 1)
70	n/a	if save > 0:
71	n/a	self.menu_save()
72	n/a	elif save < 0:
73	n/a	return
74	n/a	if self.parent.active == self:
75	n/a	self.parent.active = None
76	n/a	self.ted.TXNDeleteObject()
77	n/a	del self.ted
78	n/a	## del self.tedtexthandle
79	n/a	self.do_postclose()
80	n/a
81	n/a	def menu_save(self):
82	n/a	if not self.path:
83	n/a	self.menu_save_as()
84	n/a	return # Will call us recursively
85	n/a	dhandle = self.ted.TXNGetData(0, 0x7fffffff)
86	n/a	data = dhandle.data
87	n/a	fp = open(self.path, 'wb') # NOTE: wb, because data has CR for end-of-line
88	n/a	fp.write(data)
89	n/a	if data[-1] <> '\r': fp.write('\r')
90	n/a	fp.close()
91	n/a	self.changed = 0
92	n/a
93	n/a	def menu_save_as(self):
94	n/a	path = EasyDialogs.AskFileForSave(message='Save as:')
95	n/a	if not path: return
96	n/a	self.path = path
97	n/a	self.name = os.path.split(self.path)[-1]
98	n/a	self.wid.SetWTitle(self.name)
99	n/a	self.menu_save()
100	n/a
101	n/a	def menu_cut(self):
102	n/a	## self.ted.WESelView()
103	n/a	self.ted.TXNCut()
104	n/a	### Mlte.ConvertToPublicScrap()
105	n/a	## Scrap.ZeroScrap()
106	n/a	## self.ted.WECut()
107	n/a	## self.updatescrollbars()
108	n/a	self.parent.updatemenubar()
109	n/a	self.changed = 1
110	n/a
111	n/a	def menu_copy(self):
112	n/a	## Scrap.ZeroScrap()
113	n/a	self.ted.TXNCopy()
114	n/a	### Mlte.ConvertToPublicScrap()
115	n/a	## self.updatescrollbars()
116	n/a	self.parent.updatemenubar()
117	n/a
118	n/a	def menu_paste(self):
119	n/a	### Mlte.ConvertFromPublicScrap()
120	n/a	self.ted.TXNPaste()
121	n/a	## self.updatescrollbars()
122	n/a	self.parent.updatemenubar()
123	n/a	self.changed = 1
124	n/a
125	n/a	def menu_clear(self):
126	n/a	## self.ted.WESelView()
127	n/a	self.ted.TXNClear()
128	n/a	## self.updatescrollbars()
129	n/a	self.parent.updatemenubar()
130	n/a	self.changed = 1
131	n/a
132	n/a	def menu_undo(self):
133	n/a	self.ted.TXNUndo()
134	n/a	## self.updatescrollbars()
135	n/a	self.parent.updatemenubar()
136	n/a
137	n/a	def menu_redo(self):
138	n/a	self.ted.TXNRedo()
139	n/a	## self.updatescrollbars()
140	n/a	self.parent.updatemenubar()
141	n/a
142	n/a	def have_selection(self):
143	n/a	start, stop = self.ted.TXNGetSelection()
144	n/a	return start < stop
145	n/a
146	n/a	def can_paste(self):
147	n/a	return Mlte.TXNIsScrapPastable()
148	n/a
149	n/a	def can_undo(self):
150	n/a	can, which = self.ted.TXNCanUndo()
151	n/a	if not can:
152	n/a	return None
153	n/a	if which >= len(UNDOLABELS):
154	n/a	# Unspecified undo
155	n/a	return "Undo"
156	n/a	which = UNDOLABELS[which]
157	n/a
158	n/a	return "Undo "+which
159	n/a
160	n/a	def can_redo(self):
161	n/a	can, which = self.ted.TXNCanRedo()
162	n/a	if not can:
163	n/a	return None
164	n/a	if which >= len(UNDOLABELS):
165	n/a	# Unspecified undo
166	n/a	return "Redo"
167	n/a	which = UNDOLABELS[which]
168	n/a
169	n/a	return "Redo "+which
170	n/a
171	n/a	class Mlted(Application):
172	n/a	def __init__(self):
173	n/a	Application.__init__(self)
174	n/a	self.num = 0
175	n/a	self.active = None
176	n/a	self.updatemenubar()
177	n/a
178	n/a	def makeusermenus(self):
179	n/a	self.filemenu = m = Menu(self.menubar, "File")
180	n/a	self.newitem = MenuItem(m, "New window", "N", self.open)
181	n/a	self.openitem = MenuItem(m, "Open...", "O", self.openfile)
182	n/a	self.closeitem = MenuItem(m, "Close", "W", self.closewin)
183	n/a	m.addseparator()
184	n/a	self.saveitem = MenuItem(m, "Save", "S", self.save)
185	n/a	self.saveasitem = MenuItem(m, "Save as...", "", self.saveas)
186	n/a	m.addseparator()
187	n/a	self.quititem = MenuItem(m, "Quit", "Q", self.quit)
188	n/a
189	n/a	self.editmenu = m = Menu(self.menubar, "Edit")
190	n/a	self.undoitem = MenuItem(m, "Undo", "Z", self.undo)
191	n/a	self.redoitem = MenuItem(m, "Redo", None, self.redo)
192	n/a	m.addseparator()
193	n/a	self.cutitem = MenuItem(m, "Cut", "X", self.cut)
194	n/a	self.copyitem = MenuItem(m, "Copy", "C", self.copy)
195	n/a	self.pasteitem = MenuItem(m, "Paste", "V", self.paste)
196	n/a	self.clearitem = MenuItem(m, "Clear", "", self.clear)
197	n/a
198	n/a	# Groups of items enabled together:
199	n/a	self.windowgroup = [self.closeitem, self.saveitem, self.saveasitem, self.editmenu]
200	n/a	self.focusgroup = [self.cutitem, self.copyitem, self.clearitem]
201	n/a	self.windowgroup_on = -1
202	n/a	self.focusgroup_on = -1
203	n/a	self.pastegroup_on = -1
204	n/a	self.undo_label = "never"
205	n/a	self.redo_label = "never"
206	n/a
207	n/a	def updatemenubar(self):
208	n/a	changed = 0
209	n/a	on = (self.active <> None)
210	n/a	if on <> self.windowgroup_on:
211	n/a	for m in self.windowgroup:
212	n/a	m.enable(on)
213	n/a	self.windowgroup_on = on
214	n/a	changed = 1
215	n/a	if on:
216	n/a	# only if we have an edit menu
217	n/a	on = self.active.have_selection()
218	n/a	if on <> self.focusgroup_on:
219	n/a	for m in self.focusgroup:
220	n/a	m.enable(on)
221	n/a	self.focusgroup_on = on
222	n/a	changed = 1
223	n/a	on = self.active.can_paste()
224	n/a	if on <> self.pastegroup_on:
225	n/a	self.pasteitem.enable(on)
226	n/a	self.pastegroup_on = on
227	n/a	changed = 1
228	n/a	on = self.active.can_undo()
229	n/a	if on <> self.undo_label:
230	n/a	if on:
231	n/a	self.undoitem.enable(1)
232	n/a	self.undoitem.settext(on)
233	n/a	self.undo_label = on
234	n/a	else:
235	n/a	self.undoitem.settext("Nothing to undo")
236	n/a	self.undoitem.enable(0)
237	n/a	changed = 1
238	n/a	on = self.active.can_redo()
239	n/a	if on <> self.redo_label:
240	n/a	if on:
241	n/a	self.redoitem.enable(1)
242	n/a	self.redoitem.settext(on)
243	n/a	self.redo_label = on
244	n/a	else:
245	n/a	self.redoitem.settext("Nothing to redo")
246	n/a	self.redoitem.enable(0)
247	n/a	changed = 1
248	n/a	if changed:
249	n/a	DrawMenuBar()
250	n/a
251	n/a	#
252	n/a	# Apple menu
253	n/a	#
254	n/a
255	n/a	def do_about(self, id, item, window, event):
256	n/a	EasyDialogs.Message("A simple single-font text editor based on MacTextEditor")
257	n/a
258	n/a	#
259	n/a	# File menu
260	n/a	#
261	n/a
262	n/a	def open(self, *args):
263	n/a	self._open(0)
264	n/a
265	n/a	def openfile(self, *args):
266	n/a	self._open(1)
267	n/a
268	n/a	def _open(self, askfile):
269	n/a	if askfile:
270	n/a	path = EasyDialogs.AskFileForOpen(typeList=('TEXT',))
271	n/a	if not path:
272	n/a	return
273	n/a	name = os.path.split(path)[-1]
274	n/a	try:
275	n/a	fp = open(path, 'rb') # NOTE binary, we need cr as end-of-line
276	n/a	data = fp.read()
277	n/a	fp.close()
278	n/a	except IOError, arg:
279	n/a	EasyDialogs.Message("IOERROR: %r" % (arg,))
280	n/a	return
281	n/a	else:
282	n/a	path = None
283	n/a	name = "Untitled %d"%self.num
284	n/a	data = ''
285	n/a	w = MlteWindow(self)
286	n/a	w.open(path, name, data)
287	n/a	self.num = self.num + 1
288	n/a
289	n/a	def closewin(self, *args):
290	n/a	if self.active:
291	n/a	self.active.close()
292	n/a	else:
293	n/a	EasyDialogs.Message("No active window?")
294	n/a
295	n/a	def save(self, *args):
296	n/a	if self.active:
297	n/a	self.active.menu_save()
298	n/a	else:
299	n/a	EasyDialogs.Message("No active window?")
300	n/a
301	n/a	def saveas(self, *args):
302	n/a	if self.active:
303	n/a	self.active.menu_save_as()
304	n/a	else:
305	n/a	EasyDialogs.Message("No active window?")
306	n/a
307	n/a
308	n/a	def quit(self, *args):
309	n/a	for w in self._windows.values():
310	n/a	w.close()
311	n/a	if self._windows:
312	n/a	return
313	n/a	self._quit()
314	n/a
315	n/a	#
316	n/a	# Edit menu
317	n/a	#
318	n/a
319	n/a	def undo(self, *args):
320	n/a	if self.active:
321	n/a	self.active.menu_undo()
322	n/a	else:
323	n/a	EasyDialogs.Message("No active window?")
324	n/a
325	n/a	def redo(self, *args):
326	n/a	if self.active:
327	n/a	self.active.menu_redo()
328	n/a	else:
329	n/a	EasyDialogs.Message("No active window?")
330	n/a
331	n/a	def cut(self, *args):
332	n/a	if self.active:
333	n/a	self.active.menu_cut()
334	n/a	else:
335	n/a	EasyDialogs.Message("No active window?")
336	n/a
337	n/a	def copy(self, *args):
338	n/a	if self.active:
339	n/a	self.active.menu_copy()
340	n/a	else:
341	n/a	EasyDialogs.Message("No active window?")
342	n/a
343	n/a	def paste(self, *args):
344	n/a	if self.active:
345	n/a	self.active.menu_paste()
346	n/a	else:
347	n/a	EasyDialogs.Message("No active window?")
348	n/a
349	n/a	def clear(self, *args):
350	n/a	if self.active:
351	n/a	self.active.menu_clear()
352	n/a	else:
353	n/a	EasyDialogs.Message("No active window?")
354	n/a
355	n/a	#
356	n/a	# Other stuff
357	n/a	#
358	n/a
359	n/a	def idle(self, event):
360	n/a	if self.active:
361	n/a	self.active.do_idle(event)
362	n/a	else:
363	n/a	Qd.SetCursor(Qd.GetQDGlobalsArrow())
364	n/a
365	n/a	def main():
366	n/a	Mlte.TXNInitTextension(0)
367	n/a	try:
368	n/a	App = Mlted()
369	n/a	App.mainloop()
370	n/a	finally:
371	n/a	Mlte.TXNTerminateTextension()
372	n/a
373	n/a	if __name__ == '__main__':
374	n/a	main()