Python code coverage for Lib/idlelib/UndoDelegator.py

#	count	content
1	n/a	import string
2	n/a	from tkinter import *
3	n/a
4	n/a	from idlelib.Delegator import Delegator
5	n/a
6	n/a	#$ event <<redo>>
7	n/a	#$ win <Control-y>
8	n/a	#$ unix <Alt-z>
9	n/a
10	n/a	#$ event <<undo>>
11	n/a	#$ win <Control-z>
12	n/a	#$ unix <Control-z>
13	n/a
14	n/a	#$ event <<dump-undo-state>>
15	n/a	#$ win <Control-backslash>
16	n/a	#$ unix <Control-backslash>
17	n/a
18	n/a
19	n/a	class UndoDelegator(Delegator):
20	n/a
21	n/a	max_undo = 1000
22	n/a
23	n/a	def __init__(self):
24	n/a	Delegator.__init__(self)
25	n/a	self.reset_undo()
26	n/a
27	n/a	def setdelegate(self, delegate):
28	n/a	if self.delegate is not None:
29	n/a	self.unbind("<<undo>>")
30	n/a	self.unbind("<<redo>>")
31	n/a	self.unbind("<<dump-undo-state>>")
32	n/a	Delegator.setdelegate(self, delegate)
33	n/a	if delegate is not None:
34	n/a	self.bind("<<undo>>", self.undo_event)
35	n/a	self.bind("<<redo>>", self.redo_event)
36	n/a	self.bind("<<dump-undo-state>>", self.dump_event)
37	n/a
38	n/a	def dump_event(self, event):
39	n/a	from pprint import pprint
40	n/a	pprint(self.undolist[:self.pointer])
41	n/a	print("pointer:", self.pointer, end=' ')
42	n/a	print("saved:", self.saved, end=' ')
43	n/a	print("can_merge:", self.can_merge, end=' ')
44	n/a	print("get_saved():", self.get_saved())
45	n/a	pprint(self.undolist[self.pointer:])
46	n/a	return "break"
47	n/a
48	n/a	def reset_undo(self):
49	n/a	self.was_saved = -1
50	n/a	self.pointer = 0
51	n/a	self.undolist = []
52	n/a	self.undoblock = 0 # or a CommandSequence instance
53	n/a	self.set_saved(1)
54	n/a
55	n/a	def set_saved(self, flag):
56	n/a	if flag:
57	n/a	self.saved = self.pointer
58	n/a	else:
59	n/a	self.saved = -1
60	n/a	self.can_merge = False
61	n/a	self.check_saved()
62	n/a
63	n/a	def get_saved(self):
64	n/a	return self.saved == self.pointer
65	n/a
66	n/a	saved_change_hook = None
67	n/a
68	n/a	def set_saved_change_hook(self, hook):
69	n/a	self.saved_change_hook = hook
70	n/a
71	n/a	was_saved = -1
72	n/a
73	n/a	def check_saved(self):
74	n/a	is_saved = self.get_saved()
75	n/a	if is_saved != self.was_saved:
76	n/a	self.was_saved = is_saved
77	n/a	if self.saved_change_hook:
78	n/a	self.saved_change_hook()
79	n/a
80	n/a	def insert(self, index, chars, tags=None):
81	n/a	self.addcmd(InsertCommand(index, chars, tags))
82	n/a
83	n/a	def delete(self, index1, index2=None):
84	n/a	self.addcmd(DeleteCommand(index1, index2))
85	n/a
86	n/a	# Clients should call undo_block_start() and undo_block_stop()
87	n/a	# around a sequence of editing cmds to be treated as a unit by
88	n/a	# undo & redo. Nested matching calls are OK, and the inner calls
89	n/a	# then act like nops. OK too if no editing cmds, or only one
90	n/a	# editing cmd, is issued in between: if no cmds, the whole
91	n/a	# sequence has no effect; and if only one cmd, that cmd is entered
92	n/a	# directly into the undo list, as if undo_block_xxx hadn't been
93	n/a	# called. The intent of all that is to make this scheme easy
94	n/a	# to use: all the client has to worry about is making sure each
95	n/a	# _start() call is matched by a _stop() call.
96	n/a
97	n/a	def undo_block_start(self):
98	n/a	if self.undoblock == 0:
99	n/a	self.undoblock = CommandSequence()
100	n/a	self.undoblock.bump_depth()
101	n/a
102	n/a	def undo_block_stop(self):
103	n/a	if self.undoblock.bump_depth(-1) == 0:
104	n/a	cmd = self.undoblock
105	n/a	self.undoblock = 0
106	n/a	if len(cmd) > 0:
107	n/a	if len(cmd) == 1:
108	n/a	# no need to wrap a single cmd
109	n/a	cmd = cmd.getcmd(0)
110	n/a	# this blk of cmds, or single cmd, has already
111	n/a	# been done, so don't execute it again
112	n/a	self.addcmd(cmd, 0)
113	n/a
114	n/a	def addcmd(self, cmd, execute=True):
115	n/a	if execute:
116	n/a	cmd.do(self.delegate)
117	n/a	if self.undoblock != 0:
118	n/a	self.undoblock.append(cmd)
119	n/a	return
120	n/a	if self.can_merge and self.pointer > 0:
121	n/a	lastcmd = self.undolist[self.pointer-1]
122	n/a	if lastcmd.merge(cmd):
123	n/a	return
124	n/a	self.undolist[self.pointer:] = [cmd]
125	n/a	if self.saved > self.pointer:
126	n/a	self.saved = -1
127	n/a	self.pointer = self.pointer + 1
128	n/a	if len(self.undolist) > self.max_undo:
129	n/a	##print "truncating undo list"
130	n/a	del self.undolist[0]
131	n/a	self.pointer = self.pointer - 1
132	n/a	if self.saved >= 0:
133	n/a	self.saved = self.saved - 1
134	n/a	self.can_merge = True
135	n/a	self.check_saved()
136	n/a
137	n/a	def undo_event(self, event):
138	n/a	if self.pointer == 0:
139	n/a	self.bell()
140	n/a	return "break"
141	n/a	cmd = self.undolist[self.pointer - 1]
142	n/a	cmd.undo(self.delegate)
143	n/a	self.pointer = self.pointer - 1
144	n/a	self.can_merge = False
145	n/a	self.check_saved()
146	n/a	return "break"
147	n/a
148	n/a	def redo_event(self, event):
149	n/a	if self.pointer >= len(self.undolist):
150	n/a	self.bell()
151	n/a	return "break"
152	n/a	cmd = self.undolist[self.pointer]
153	n/a	cmd.redo(self.delegate)
154	n/a	self.pointer = self.pointer + 1
155	n/a	self.can_merge = False
156	n/a	self.check_saved()
157	n/a	return "break"
158	n/a
159	n/a
160	n/a	class Command:
161	n/a
162	n/a	# Base class for Undoable commands
163	n/a
164	n/a	tags = None
165	n/a
166	n/a	def __init__(self, index1, index2, chars, tags=None):
167	n/a	self.marks_before = {}
168	n/a	self.marks_after = {}
169	n/a	self.index1 = index1
170	n/a	self.index2 = index2
171	n/a	self.chars = chars
172	n/a	if tags:
173	n/a	self.tags = tags
174	n/a
175	n/a	def __repr__(self):
176	n/a	s = self.__class__.__name__
177	n/a	t = (self.index1, self.index2, self.chars, self.tags)
178	n/a	if self.tags is None:
179	n/a	t = t[:-1]
180	n/a	return s + repr(t)
181	n/a
182	n/a	def do(self, text):
183	n/a	pass
184	n/a
185	n/a	def redo(self, text):
186	n/a	pass
187	n/a
188	n/a	def undo(self, text):
189	n/a	pass
190	n/a
191	n/a	def merge(self, cmd):
192	n/a	return 0
193	n/a
194	n/a	def save_marks(self, text):
195	n/a	marks = {}
196	n/a	for name in text.mark_names():
197	n/a	if name != "insert" and name != "current":
198	n/a	marks[name] = text.index(name)
199	n/a	return marks
200	n/a
201	n/a	def set_marks(self, text, marks):
202	n/a	for name, index in marks.items():
203	n/a	text.mark_set(name, index)
204	n/a
205	n/a
206	n/a	class InsertCommand(Command):
207	n/a
208	n/a	# Undoable insert command
209	n/a
210	n/a	def __init__(self, index1, chars, tags=None):
211	n/a	Command.__init__(self, index1, None, chars, tags)
212	n/a
213	n/a	def do(self, text):
214	n/a	self.marks_before = self.save_marks(text)
215	n/a	self.index1 = text.index(self.index1)
216	n/a	if text.compare(self.index1, ">", "end-1c"):
217	n/a	# Insert before the final newline
218	n/a	self.index1 = text.index("end-1c")
219	n/a	text.insert(self.index1, self.chars, self.tags)
220	n/a	self.index2 = text.index("%s+%dc" % (self.index1, len(self.chars)))
221	n/a	self.marks_after = self.save_marks(text)
222	n/a	##sys.__stderr__.write("do: %s\n" % self)
223	n/a
224	n/a	def redo(self, text):
225	n/a	text.mark_set('insert', self.index1)
226	n/a	text.insert(self.index1, self.chars, self.tags)
227	n/a	self.set_marks(text, self.marks_after)
228	n/a	text.see('insert')
229	n/a	##sys.__stderr__.write("redo: %s\n" % self)
230	n/a
231	n/a	def undo(self, text):
232	n/a	text.mark_set('insert', self.index1)
233	n/a	text.delete(self.index1, self.index2)
234	n/a	self.set_marks(text, self.marks_before)
235	n/a	text.see('insert')
236	n/a	##sys.__stderr__.write("undo: %s\n" % self)
237	n/a
238	n/a	def merge(self, cmd):
239	n/a	if self.__class__ is not cmd.__class__:
240	n/a	return False
241	n/a	if self.index2 != cmd.index1:
242	n/a	return False
243	n/a	if self.tags != cmd.tags:
244	n/a	return False
245	n/a	if len(cmd.chars) != 1:
246	n/a	return False
247	n/a	if self.chars and \
248	n/a	self.classify(self.chars[-1]) != self.classify(cmd.chars):
249	n/a	return False
250	n/a	self.index2 = cmd.index2
251	n/a	self.chars = self.chars + cmd.chars
252	n/a	return True
253	n/a
254	n/a	alphanumeric = string.ascii_letters + string.digits + "_"
255	n/a
256	n/a	def classify(self, c):
257	n/a	if c in self.alphanumeric:
258	n/a	return "alphanumeric"
259	n/a	if c == "\n":
260	n/a	return "newline"
261	n/a	return "punctuation"
262	n/a
263	n/a
264	n/a	class DeleteCommand(Command):
265	n/a
266	n/a	# Undoable delete command
267	n/a
268	n/a	def __init__(self, index1, index2=None):
269	n/a	Command.__init__(self, index1, index2, None, None)
270	n/a
271	n/a	def do(self, text):
272	n/a	self.marks_before = self.save_marks(text)
273	n/a	self.index1 = text.index(self.index1)
274	n/a	if self.index2:
275	n/a	self.index2 = text.index(self.index2)
276	n/a	else:
277	n/a	self.index2 = text.index(self.index1 + " +1c")
278	n/a	if text.compare(self.index2, ">", "end-1c"):
279	n/a	# Don't delete the final newline
280	n/a	self.index2 = text.index("end-1c")
281	n/a	self.chars = text.get(self.index1, self.index2)
282	n/a	text.delete(self.index1, self.index2)
283	n/a	self.marks_after = self.save_marks(text)
284	n/a	##sys.__stderr__.write("do: %s\n" % self)
285	n/a
286	n/a	def redo(self, text):
287	n/a	text.mark_set('insert', self.index1)
288	n/a	text.delete(self.index1, self.index2)
289	n/a	self.set_marks(text, self.marks_after)
290	n/a	text.see('insert')
291	n/a	##sys.__stderr__.write("redo: %s\n" % self)
292	n/a
293	n/a	def undo(self, text):
294	n/a	text.mark_set('insert', self.index1)
295	n/a	text.insert(self.index1, self.chars)
296	n/a	self.set_marks(text, self.marks_before)
297	n/a	text.see('insert')
298	n/a	##sys.__stderr__.write("undo: %s\n" % self)
299	n/a
300	n/a	class CommandSequence(Command):
301	n/a
302	n/a	# Wrapper for a sequence of undoable cmds to be undone/redone
303	n/a	# as a unit
304	n/a
305	n/a	def __init__(self):
306	n/a	self.cmds = []
307	n/a	self.depth = 0
308	n/a
309	n/a	def __repr__(self):
310	n/a	s = self.__class__.__name__
311	n/a	strs = []
312	n/a	for cmd in self.cmds:
313	n/a	strs.append(" %r" % (cmd,))
314	n/a	return s + "(\n" + ",\n".join(strs) + "\n)"
315	n/a
316	n/a	def __len__(self):
317	n/a	return len(self.cmds)
318	n/a
319	n/a	def append(self, cmd):
320	n/a	self.cmds.append(cmd)
321	n/a
322	n/a	def getcmd(self, i):
323	n/a	return self.cmds[i]
324	n/a
325	n/a	def redo(self, text):
326	n/a	for cmd in self.cmds:
327	n/a	cmd.redo(text)
328	n/a
329	n/a	def undo(self, text):
330	n/a	cmds = self.cmds[:]
331	n/a	cmds.reverse()
332	n/a	for cmd in cmds:
333	n/a	cmd.undo(text)
334	n/a
335	n/a	def bump_depth(self, incr=1):
336	n/a	self.depth = self.depth + incr
337	n/a	return self.depth
338	n/a
339	n/a	def main():
340	n/a	from idlelib.Percolator import Percolator
341	n/a	root = Tk()
342	n/a	root.wm_protocol("WM_DELETE_WINDOW", root.quit)
343	n/a	text = Text()
344	n/a	text.pack()
345	n/a	text.focus_set()
346	n/a	p = Percolator(text)
347	n/a	d = UndoDelegator()
348	n/a	p.insertfilter(d)
349	n/a	root.mainloop()
350	n/a
351	n/a	if __name__ == "__main__":
352	n/a	main()