Python code coverage for Mac/Demo/textedit/ped.py

#	count	content
1	n/a	# A minimal text editor.
2	n/a	#
3	n/a	# To be done:
4	n/a	# - Update viewrect after resize
5	n/a	# - Handle horizontal scrollbar correctly
6	n/a	# - Functionality: find, etc.
7	n/a
8	n/a	from Carbon.Menu import DrawMenuBar
9	n/a	from FrameWork import *
10	n/a	from Carbon import Win
11	n/a	from Carbon import Qd
12	n/a	from Carbon import TE
13	n/a	from Carbon import Scrap
14	n/a	import os
15	n/a	import macfs
16	n/a
17	n/a	class TEWindow(ScrolledWindow):
18	n/a	def open(self, path, name, data):
19	n/a	self.path = path
20	n/a	self.name = name
21	n/a	r = windowbounds(400, 400)
22	n/a	w = Win.NewWindow(r, name, 1, 0, -1, 1, 0)
23	n/a	self.wid = w
24	n/a	x0, y0, x1, y1 = self.wid.GetWindowPort().GetPortBounds()
25	n/a	x0 = x0 + 4
26	n/a	y0 = y0 + 4
27	n/a	x1 = x1 - 20
28	n/a	y1 = y1 - 20
29	n/a	vr = dr = x0, y0, x1, y1
30	n/a	##vr = 4, 0, r[2]-r[0]-15, r[3]-r[1]-15
31	n/a	##dr = (0, 0, vr[2], 0)
32	n/a	Qd.SetPort(w)
33	n/a	Qd.TextFont(4)
34	n/a	Qd.TextSize(9)
35	n/a	self.ted = TE.TENew(dr, vr)
36	n/a	self.ted.TEAutoView(1)
37	n/a	self.ted.TESetText(data)
38	n/a	w.DrawGrowIcon()
39	n/a	self.scrollbars()
40	n/a	self.changed = 0
41	n/a	self.do_postopen()
42	n/a	self.do_activate(1, None)
43	n/a
44	n/a	def do_idle(self):
45	n/a	self.ted.TEIdle()
46	n/a
47	n/a	def getscrollbarvalues(self):
48	n/a	dr = self.ted.destRect
49	n/a	vr = self.ted.viewRect
50	n/a	height = self.ted.nLines * self.ted.lineHeight
51	n/a	vx = self.scalebarvalue(dr[0], dr[2]-dr[0], vr[0], vr[2])
52	n/a	vy = self.scalebarvalue(dr[1], dr[1]+height, vr[1], vr[3])
53	n/a	print dr, vr, height, vx, vy
54	n/a	return None, vy
55	n/a
56	n/a	def scrollbar_callback(self, which, what, value):
57	n/a	if which == 'y':
58	n/a	if what == 'set':
59	n/a	height = self.ted.nLines * self.ted.lineHeight
60	n/a	cur = self.getscrollbarvalues()[1]
61	n/a	delta = (cur-value)*height/32767
62	n/a	if what == '-':
63	n/a	delta = self.ted.lineHeight
64	n/a	elif what == '--':
65	n/a	delta = (self.ted.viewRect[3]-self.ted.lineHeight)
66	n/a	if delta <= 0:
67	n/a	delta = self.ted.lineHeight
68	n/a	elif what == '+':
69	n/a	delta = -self.ted.lineHeight
70	n/a	elif what == '++':
71	n/a	delta = -(self.ted.viewRect[3]-self.ted.lineHeight)
72	n/a	if delta >= 0:
73	n/a	delta = -self.ted.lineHeight
74	n/a	self.ted.TEPinScroll(0, delta)
75	n/a	print 'SCROLL Y', delta
76	n/a	else:
77	n/a	pass # No horizontal scrolling
78	n/a
79	n/a	def do_activate(self, onoff, evt):
80	n/a	print "ACTIVATE", onoff
81	n/a	ScrolledWindow.do_activate(self, onoff, evt)
82	n/a	if onoff:
83	n/a	self.ted.TEActivate()
84	n/a	self.parent.active = self
85	n/a	self.parent.updatemenubar()
86	n/a	else:
87	n/a	self.ted.TEDeactivate()
88	n/a
89	n/a	def do_update(self, wid, event):
90	n/a	Qd.EraseRect(wid.GetWindowPort().GetPortBounds())
91	n/a	self.ted.TEUpdate(wid.GetWindowPort().GetPortBounds())
92	n/a	self.updatescrollbars()
93	n/a
94	n/a	def do_contentclick(self, local, modifiers, evt):
95	n/a	shifted = (modifiers & 0x200)
96	n/a	self.ted.TEClick(local, shifted)
97	n/a	self.updatescrollbars()
98	n/a	self.parent.updatemenubar()
99	n/a
100	n/a	def do_char(self, ch, event):
101	n/a	self.ted.TESelView()
102	n/a	self.ted.TEKey(ord(ch))
103	n/a	self.changed = 1
104	n/a	self.updatescrollbars()
105	n/a	self.parent.updatemenubar()
106	n/a
107	n/a	def close(self):
108	n/a	if self.changed:
109	n/a	save = EasyDialogs.AskYesNoCancel('Save window "%s" before closing?'%self.name, 1)
110	n/a	if save > 0:
111	n/a	self.menu_save()
112	n/a	elif save < 0:
113	n/a	return
114	n/a	if self.parent.active == self:
115	n/a	self.parent.active = None
116	n/a	self.parent.updatemenubar()
117	n/a	del self.ted
118	n/a	self.do_postclose()
119	n/a
120	n/a	def menu_save(self):
121	n/a	if not self.path:
122	n/a	self.menu_save_as()
123	n/a	return # Will call us recursively
124	n/a	print 'Saving to ', self.path
125	n/a	dhandle = self.ted.TEGetText()
126	n/a	data = dhandle.data
127	n/a	fp = open(self.path, 'wb') # NOTE: wb, because data has CR for end-of-line
128	n/a	fp.write(data)
129	n/a	if data[-1] <> '\r': fp.write('\r')
130	n/a	fp.close()
131	n/a	self.changed = 0
132	n/a
133	n/a	def menu_save_as(self):
134	n/a	path = EasyDialogs.AskFileForSave(message='Save as:')
135	n/a	if not path: return
136	n/a	self.path = path
137	n/a	self.name = os.path.split(self.path)[-1]
138	n/a	self.wid.SetWTitle(self.name)
139	n/a	self.menu_save()
140	n/a
141	n/a	def menu_cut(self):
142	n/a	self.ted.TESelView()
143	n/a	self.ted.TECut()
144	n/a	if hasattr(Scrap, 'ZeroScrap'):
145	n/a	Scrap.ZeroScrap()
146	n/a	else:
147	n/a	Scrap.ClearCurrentScrap()
148	n/a	TE.TEToScrap()
149	n/a	self.updatescrollbars()
150	n/a	self.parent.updatemenubar()
151	n/a	self.changed = 1
152	n/a
153	n/a	def menu_copy(self):
154	n/a	self.ted.TECopy()
155	n/a	if hasattr(Scrap, 'ZeroScrap'):
156	n/a	Scrap.ZeroScrap()
157	n/a	else:
158	n/a	Scrap.ClearCurrentScrap()
159	n/a	TE.TEToScrap()
160	n/a	self.updatescrollbars()
161	n/a	self.parent.updatemenubar()
162	n/a
163	n/a	def menu_paste(self):
164	n/a	TE.TEFromScrap()
165	n/a	self.ted.TESelView()
166	n/a	self.ted.TEPaste()
167	n/a	self.updatescrollbars()
168	n/a	self.parent.updatemenubar()
169	n/a	self.changed = 1
170	n/a
171	n/a	def menu_clear(self):
172	n/a	self.ted.TESelView()
173	n/a	self.ted.TEDelete()
174	n/a	self.updatescrollbars()
175	n/a	self.parent.updatemenubar()
176	n/a	self.changed = 1
177	n/a
178	n/a	def have_selection(self):
179	n/a	return (self.ted.selStart < self.ted.selEnd)
180	n/a
181	n/a	class Ped(Application):
182	n/a	def __init__(self):
183	n/a	Application.__init__(self)
184	n/a	self.num = 0
185	n/a	self.active = None
186	n/a	self.updatemenubar()
187	n/a
188	n/a	def makeusermenus(self):
189	n/a	self.filemenu = m = Menu(self.menubar, "File")
190	n/a	self.newitem = MenuItem(m, "New window", "N", self.open)
191	n/a	self.openitem = MenuItem(m, "Open...", "O", self.openfile)
192	n/a	self.closeitem = MenuItem(m, "Close", "W", self.closewin)
193	n/a	m.addseparator()
194	n/a	self.saveitem = MenuItem(m, "Save", "S", self.save)
195	n/a	self.saveasitem = MenuItem(m, "Save as...", "", self.saveas)
196	n/a	m.addseparator()
197	n/a	self.quititem = MenuItem(m, "Quit", "Q", self.quit)
198	n/a
199	n/a	self.editmenu = m = Menu(self.menubar, "Edit")
200	n/a	self.undoitem = MenuItem(m, "Undo", "Z", self.undo)
201	n/a	self.cutitem = MenuItem(m, "Cut", "X", self.cut)
202	n/a	self.copyitem = MenuItem(m, "Copy", "C", self.copy)
203	n/a	self.pasteitem = MenuItem(m, "Paste", "V", self.paste)
204	n/a	self.clearitem = MenuItem(m, "Clear", "", self.clear)
205	n/a
206	n/a	# Not yet implemented:
207	n/a	self.undoitem.enable(0)
208	n/a
209	n/a	# Groups of items enabled together:
210	n/a	self.windowgroup = [self.closeitem, self.saveitem, self.saveasitem, self.editmenu]
211	n/a	self.focusgroup = [self.cutitem, self.copyitem, self.clearitem]
212	n/a	self.windowgroup_on = -1
213	n/a	self.focusgroup_on = -1
214	n/a	self.pastegroup_on = -1
215	n/a
216	n/a	def updatemenubar(self):
217	n/a	changed = 0
218	n/a	on = (self.active <> None)
219	n/a	if on <> self.windowgroup_on:
220	n/a	for m in self.windowgroup:
221	n/a	m.enable(on)
222	n/a	self.windowgroup_on = on
223	n/a	changed = 1
224	n/a	if on:
225	n/a	# only if we have an edit menu
226	n/a	on = self.active.have_selection()
227	n/a	if on <> self.focusgroup_on:
228	n/a	for m in self.focusgroup:
229	n/a	m.enable(on)
230	n/a	self.focusgroup_on = on
231	n/a	changed = 1
232	n/a	if hasattr(Scrap, 'InfoScrap'):
233	n/a	on = (Scrap.InfoScrap()[0] <> 0)
234	n/a	else:
235	n/a	flavors = Scrap.GetCurrentScrap().GetScrapFlavorInfoList()
236	n/a	for tp, info in flavors:
237	n/a	if tp == 'TEXT':
238	n/a	on = 1
239	n/a	break
240	n/a	else:
241	n/a	on = 0
242	n/a	if on <> self.pastegroup_on:
243	n/a	self.pasteitem.enable(on)
244	n/a	self.pastegroup_on = on
245	n/a	changed = 1
246	n/a	if changed:
247	n/a	DrawMenuBar()
248	n/a
249	n/a	#
250	n/a	# Apple menu
251	n/a	#
252	n/a
253	n/a	def do_about(self, id, item, window, event):
254	n/a	EasyDialogs.Message("A simple single-font text editor")
255	n/a
256	n/a	#
257	n/a	# File menu
258	n/a	#
259	n/a
260	n/a	def open(self, *args):
261	n/a	self._open(0)
262	n/a
263	n/a	def openfile(self, *args):
264	n/a	self._open(1)
265	n/a
266	n/a	def _open(self, askfile):
267	n/a	if askfile:
268	n/a	path = EasyDialogs.AskFileForOpen(typeList=('TEXT',))
269	n/a	if not path:
270	n/a	return
271	n/a	name = os.path.split(path)[-1]
272	n/a	try:
273	n/a	fp = open(path, 'rb') # NOTE binary, we need cr as end-of-line
274	n/a	data = fp.read()
275	n/a	fp.close()
276	n/a	except IOError, arg:
277	n/a	EasyDialogs.Message("IOERROR: %r" % (arg,))
278	n/a	return
279	n/a	else:
280	n/a	path = None
281	n/a	name = "Untitled %d"%self.num
282	n/a	data = ''
283	n/a	w = TEWindow(self)
284	n/a	w.open(path, name, data)
285	n/a	self.num = self.num + 1
286	n/a
287	n/a	def closewin(self, *args):
288	n/a	if self.active:
289	n/a	self.active.close()
290	n/a	else:
291	n/a	EasyDialogs.Message("No active window?")
292	n/a
293	n/a	def save(self, *args):
294	n/a	if self.active:
295	n/a	self.active.menu_save()
296	n/a	else:
297	n/a	EasyDialogs.Message("No active window?")
298	n/a
299	n/a	def saveas(self, *args):
300	n/a	if self.active:
301	n/a	self.active.menu_save_as()
302	n/a	else:
303	n/a	EasyDialogs.Message("No active window?")
304	n/a
305	n/a
306	n/a	def quit(self, *args):
307	n/a	for w in self._windows.values():
308	n/a	w.close()
309	n/a	if self._windows:
310	n/a	return
311	n/a	self._quit()
312	n/a
313	n/a	#
314	n/a	# Edit menu
315	n/a	#
316	n/a
317	n/a	def undo(self, *args):
318	n/a	pass
319	n/a
320	n/a	def cut(self, *args):
321	n/a	if self.active:
322	n/a	self.active.menu_cut()
323	n/a	else:
324	n/a	EasyDialogs.Message("No active window?")
325	n/a
326	n/a	def copy(self, *args):
327	n/a	if self.active:
328	n/a	self.active.menu_copy()
329	n/a	else:
330	n/a	EasyDialogs.Message("No active window?")
331	n/a
332	n/a	def paste(self, *args):
333	n/a	if self.active:
334	n/a	self.active.menu_paste()
335	n/a	else:
336	n/a	EasyDialogs.Message("No active window?")
337	n/a
338	n/a	def clear(self, *args):
339	n/a	if self.active:
340	n/a	self.active.menu_clear()
341	n/a	else:
342	n/a	EasyDialogs.Message("No active window?")
343	n/a
344	n/a	#
345	n/a	# Other stuff
346	n/a	#
347	n/a
348	n/a	def idle(self, *args):
349	n/a	if self.active:
350	n/a	self.active.do_idle()
351	n/a	else:
352	n/a	Qd.SetCursor(Qd.GetQDGlobalsArrow())
353	n/a
354	n/a	def main():
355	n/a	App = Ped()
356	n/a	App.mainloop()
357	n/a
358	n/a	if __name__ == '__main__':
359	n/a	main()