Python code coverage for Lib/idlelib/GrepDialog.py

#	count	content
1	n/a	import os
2	n/a	import fnmatch
3	n/a	import sys
4	n/a	from tkinter import *
5	n/a	from idlelib import SearchEngine
6	n/a	from idlelib.SearchDialogBase import SearchDialogBase
7	n/a
8	n/a	def grep(text, io=None, flist=None):
9	n/a	root = text._root()
10	n/a	engine = SearchEngine.get(root)
11	n/a	if not hasattr(engine, "_grepdialog"):
12	n/a	engine._grepdialog = GrepDialog(root, engine, flist)
13	n/a	dialog = engine._grepdialog
14	n/a	searchphrase = text.get("sel.first", "sel.last")
15	n/a	dialog.open(text, searchphrase, io)
16	n/a
17	n/a	class GrepDialog(SearchDialogBase):
18	n/a
19	n/a	title = "Find in Files Dialog"
20	n/a	icon = "Grep"
21	n/a	needwrapbutton = 0
22	n/a
23	n/a	def __init__(self, root, engine, flist):
24	n/a	SearchDialogBase.__init__(self, root, engine)
25	n/a	self.flist = flist
26	n/a	self.globvar = StringVar(root)
27	n/a	self.recvar = BooleanVar(root)
28	n/a
29	n/a	def open(self, text, searchphrase, io=None):
30	n/a	SearchDialogBase.open(self, text, searchphrase)
31	n/a	if io:
32	n/a	path = io.filename or ""
33	n/a	else:
34	n/a	path = ""
35	n/a	dir, base = os.path.split(path)
36	n/a	head, tail = os.path.splitext(base)
37	n/a	if not tail:
38	n/a	tail = ".py"
39	n/a	self.globvar.set(os.path.join(dir, "*" + tail))
40	n/a
41	n/a	def create_entries(self):
42	n/a	SearchDialogBase.create_entries(self)
43	n/a	self.globent = self.make_entry("In files:", self.globvar)
44	n/a
45	n/a	def create_other_buttons(self):
46	n/a	f = self.make_frame()
47	n/a
48	n/a	btn = Checkbutton(f, anchor="w",
49	n/a	variable=self.recvar,
50	n/a	text="Recurse down subdirectories")
51	n/a	btn.pack(side="top", fill="both")
52	n/a	btn.select()
53	n/a
54	n/a	def create_command_buttons(self):
55	n/a	SearchDialogBase.create_command_buttons(self)
56	n/a	self.make_button("Search Files", self.default_command, 1)
57	n/a
58	n/a	def default_command(self, event=None):
59	n/a	prog = self.engine.getprog()
60	n/a	if not prog:
61	n/a	return
62	n/a	path = self.globvar.get()
63	n/a	if not path:
64	n/a	self.top.bell()
65	n/a	return
66	n/a	from idlelib.OutputWindow import OutputWindow
67	n/a	save = sys.stdout
68	n/a	try:
69	n/a	sys.stdout = OutputWindow(self.flist)
70	n/a	self.grep_it(prog, path)
71	n/a	finally:
72	n/a	sys.stdout = save
73	n/a
74	n/a	def grep_it(self, prog, path):
75	n/a	dir, base = os.path.split(path)
76	n/a	list = self.findfiles(dir, base, self.recvar.get())
77	n/a	list.sort()
78	n/a	self.close()
79	n/a	pat = self.engine.getpat()
80	n/a	print("Searching %r in %s ..." % (pat, path))
81	n/a	hits = 0
82	n/a	for fn in list:
83	n/a	try:
84	n/a	with open(fn, errors='replace') as f:
85	n/a	for lineno, line in enumerate(f, 1):
86	n/a	if line[-1:] == '\n':
87	n/a	line = line[:-1]
88	n/a	if prog.search(line):
89	n/a	sys.stdout.write("%s: %s: %s\n" %
90	n/a	(fn, lineno, line))
91	n/a	hits += 1
92	n/a	except OSError as msg:
93	n/a	print(msg)
94	n/a	print(("Hits found: %s\n"
95	n/a	"(Hint: right-click to open locations.)"
96	n/a	% hits) if hits else "No hits.")
97	n/a
98	n/a	def findfiles(self, dir, base, rec):
99	n/a	try:
100	n/a	names = os.listdir(dir or os.curdir)
101	n/a	except OSerror as msg:
102	n/a	print(msg)
103	n/a	return []
104	n/a	list = []
105	n/a	subdirs = []
106	n/a	for name in names:
107	n/a	fn = os.path.join(dir, name)
108	n/a	if os.path.isdir(fn):
109	n/a	subdirs.append(fn)
110	n/a	else:
111	n/a	if fnmatch.fnmatch(name, base):
112	n/a	list.append(fn)
113	n/a	if rec:
114	n/a	for subdir in subdirs:
115	n/a	list.extend(self.findfiles(subdir, base, rec))
116	n/a	return list
117	n/a
118	n/a	def close(self, event=None):
119	n/a	if self.top:
120	n/a	self.top.grab_release()
121	n/a	self.top.withdraw()
122	n/a
123	n/a	if __name__ == "__main__":
124	n/a	# A human test is a bit tricky since EditorWindow() imports this module.
125	n/a	# Hence Idle must be restarted after editing this file for a live test.
126	n/a	import unittest
127	n/a	unittest.main('idlelib.idle_test.test_grep', verbosity=2, exit=False)