Python code coverage for Lib/idlelib/SearchEngine.py

#	count	content
1	n/a	import re
2	n/a	from tkinter import *
3	n/a	import tkinter.messagebox as tkMessageBox
4	n/a
5	n/a	def get(root):
6	n/a	if not hasattr(root, "_searchengine"):
7	n/a	root._searchengine = SearchEngine(root)
8	n/a	# XXX This will never garbage-collect -- who cares
9	n/a	return root._searchengine
10	n/a
11	n/a	class SearchEngine:
12	n/a
13	n/a	def __init__(self, root):
14	n/a	self.root = root
15	n/a	# State shared by search, replace, and grep;
16	n/a	# the search dialogs bind these to UI elements.
17	n/a	self.patvar = StringVar(root) # search pattern
18	n/a	self.revar = BooleanVar(root) # regular expression?
19	n/a	self.casevar = BooleanVar(root) # match case?
20	n/a	self.wordvar = BooleanVar(root) # match whole word?
21	n/a	self.wrapvar = BooleanVar(root) # wrap around buffer?
22	n/a	self.wrapvar.set(1) # (on by default)
23	n/a	self.backvar = BooleanVar(root) # search backwards?
24	n/a
25	n/a	# Access methods
26	n/a
27	n/a	def getpat(self):
28	n/a	return self.patvar.get()
29	n/a
30	n/a	def setpat(self, pat):
31	n/a	self.patvar.set(pat)
32	n/a
33	n/a	def isre(self):
34	n/a	return self.revar.get()
35	n/a
36	n/a	def iscase(self):
37	n/a	return self.casevar.get()
38	n/a
39	n/a	def isword(self):
40	n/a	return self.wordvar.get()
41	n/a
42	n/a	def iswrap(self):
43	n/a	return self.wrapvar.get()
44	n/a
45	n/a	def isback(self):
46	n/a	return self.backvar.get()
47	n/a
48	n/a	# Higher level access methods
49	n/a
50	n/a	def getcookedpat(self):
51	n/a	pat = self.getpat()
52	n/a	if not self.isre():
53	n/a	pat = re.escape(pat)
54	n/a	if self.isword():
55	n/a	pat = r"\b%s\b" % pat
56	n/a	return pat
57	n/a
58	n/a	def getprog(self):
59	n/a	pat = self.getpat()
60	n/a	if not pat:
61	n/a	self.report_error(pat, "Empty regular expression")
62	n/a	return None
63	n/a	pat = self.getcookedpat()
64	n/a	flags = 0
65	n/a	if not self.iscase():
66	n/a	flags = flags \| re.IGNORECASE
67	n/a	try:
68	n/a	prog = re.compile(pat, flags)
69	n/a	except re.error as what:
70	n/a	try:
71	n/a	msg, col = what
72	n/a	except:
73	n/a	msg = str(what)
74	n/a	col = -1
75	n/a	self.report_error(pat, msg, col)
76	n/a	return None
77	n/a	return prog
78	n/a
79	n/a	def report_error(self, pat, msg, col=-1):
80	n/a	# Derived class could overrid this with something fancier
81	n/a	msg = "Error: " + str(msg)
82	n/a	if pat:
83	n/a	msg = msg + "\np\Pattern: " + str(pat)
84	n/a	if col >= 0:
85	n/a	msg = msg + "\nOffset: " + str(col)
86	n/a	tkMessageBox.showerror("Regular expression error",
87	n/a	msg, master=self.root)
88	n/a
89	n/a	def setcookedpat(self, pat):
90	n/a	if self.isre():
91	n/a	pat = re.escape(pat)
92	n/a	self.setpat(pat)
93	n/a
94	n/a	def search_text(self, text, prog=None, ok=0):
95	n/a	"""Search a text widget for the pattern.
96	n/a
97	n/a	If prog is given, it should be the precompiled pattern.
98	n/a	Return a tuple (lineno, matchobj); None if not found.
99	n/a
100	n/a	This obeys the wrap and direction (back) settings.
101	n/a
102	n/a	The search starts at the selection (if there is one) or
103	n/a	at the insert mark (otherwise). If the search is forward,
104	n/a	it starts at the right of the selection; for a backward
105	n/a	search, it starts at the left end. An empty match exactly
106	n/a	at either end of the selection (or at the insert mark if
107	n/a	there is no selection) is ignored unless the ok flag is true
108	n/a	-- this is done to guarantee progress.
109	n/a
110	n/a	If the search is allowed to wrap around, it will return the
111	n/a	original selection if (and only if) it is the only match.
112	n/a
113	n/a	"""
114	n/a	if not prog:
115	n/a	prog = self.getprog()
116	n/a	if not prog:
117	n/a	return None # Compilation failed -- stop
118	n/a	wrap = self.wrapvar.get()
119	n/a	first, last = get_selection(text)
120	n/a	if self.isback():
121	n/a	if ok:
122	n/a	start = last
123	n/a	else:
124	n/a	start = first
125	n/a	line, col = get_line_col(start)
126	n/a	res = self.search_backward(text, prog, line, col, wrap, ok)
127	n/a	else:
128	n/a	if ok:
129	n/a	start = first
130	n/a	else:
131	n/a	start = last
132	n/a	line, col = get_line_col(start)
133	n/a	res = self.search_forward(text, prog, line, col, wrap, ok)
134	n/a	return res
135	n/a
136	n/a	def search_forward(self, text, prog, line, col, wrap, ok=0):
137	n/a	wrapped = 0
138	n/a	startline = line
139	n/a	chars = text.get("%d.0" % line, "%d.0" % (line+1))
140	n/a	while chars:
141	n/a	m = prog.search(chars[:-1], col)
142	n/a	if m:
143	n/a	if ok or m.end() > col:
144	n/a	return line, m
145	n/a	line = line + 1
146	n/a	if wrapped and line > startline:
147	n/a	break
148	n/a	col = 0
149	n/a	ok = 1
150	n/a	chars = text.get("%d.0" % line, "%d.0" % (line+1))
151	n/a	if not chars and wrap:
152	n/a	wrapped = 1
153	n/a	wrap = 0
154	n/a	line = 1
155	n/a	chars = text.get("1.0", "2.0")
156	n/a	return None
157	n/a
158	n/a	def search_backward(self, text, prog, line, col, wrap, ok=0):
159	n/a	wrapped = 0
160	n/a	startline = line
161	n/a	chars = text.get("%d.0" % line, "%d.0" % (line+1))
162	n/a	while 1:
163	n/a	m = search_reverse(prog, chars[:-1], col)
164	n/a	if m:
165	n/a	if ok or m.start() < col:
166	n/a	return line, m
167	n/a	line = line - 1
168	n/a	if wrapped and line < startline:
169	n/a	break
170	n/a	ok = 1
171	n/a	if line <= 0:
172	n/a	if not wrap:
173	n/a	break
174	n/a	wrapped = 1
175	n/a	wrap = 0
176	n/a	pos = text.index("end-1c")
177	n/a	line, col = map(int, pos.split("."))
178	n/a	chars = text.get("%d.0" % line, "%d.0" % (line+1))
179	n/a	col = len(chars) - 1
180	n/a	return None
181	n/a
182	n/a	# Helper to search backwards in a string.
183	n/a	# (Optimized for the case where the pattern isn't found.)
184	n/a
185	n/a	def search_reverse(prog, chars, col):
186	n/a	m = prog.search(chars)
187	n/a	if not m:
188	n/a	return None
189	n/a	found = None
190	n/a	i, j = m.span()
191	n/a	while i < col and j <= col:
192	n/a	found = m
193	n/a	if i == j:
194	n/a	j = j+1
195	n/a	m = prog.search(chars, j)
196	n/a	if not m:
197	n/a	break
198	n/a	i, j = m.span()
199	n/a	return found
200	n/a
201	n/a	# Helper to get selection end points, defaulting to insert mark.
202	n/a	# Return a tuple of indices ("line.col" strings).
203	n/a
204	n/a	def get_selection(text):
205	n/a	try:
206	n/a	first = text.index("sel.first")
207	n/a	last = text.index("sel.last")
208	n/a	except TclError:
209	n/a	first = last = None
210	n/a	if not first:
211	n/a	first = text.index("insert")
212	n/a	if not last:
213	n/a	last = first
214	n/a	return first, last
215	n/a
216	n/a	# Helper to parse a text index into a (line, col) tuple.
217	n/a
218	n/a	def get_line_col(index):
219	n/a	line, col = map(int, index.split(".")) # Fails on invalid index
220	n/a	return line, col