mirror of
https://github.com/python/cpython.git
synced 2025-07-28 21:55:21 +00:00
aaab30e00c
(with one small bugfix in bgen/bgen/scantools.py) This replaces string module functions with string methods for the stuff in the Tools directory. Several uses of string.letters etc. are still remaining.
134 lines
3.9 KiB
Python
134 lines
3.9 KiB
Python
import os
|
|
import re
|
|
import fnmatch
|
|
import sys
|
|
from Tkinter import *
|
|
import tkMessageBox
|
|
import SearchEngine
|
|
from SearchDialogBase import SearchDialogBase
|
|
|
|
def grep(text, io=None, flist=None):
|
|
root = text._root()
|
|
engine = SearchEngine.get(root)
|
|
if not hasattr(engine, "_grepdialog"):
|
|
engine._grepdialog = GrepDialog(root, engine, flist)
|
|
dialog = engine._grepdialog
|
|
dialog.open(io)
|
|
|
|
class GrepDialog(SearchDialogBase):
|
|
|
|
title = "Find in Files Dialog"
|
|
icon = "Grep"
|
|
needwrapbutton = 0
|
|
|
|
def __init__(self, root, engine, flist):
|
|
SearchDialogBase.__init__(self, root, engine)
|
|
self.flist = flist
|
|
self.globvar = StringVar(root)
|
|
self.recvar = BooleanVar(root)
|
|
|
|
def open(self, io=None):
|
|
SearchDialogBase.open(self, None)
|
|
if io:
|
|
path = io.filename or ""
|
|
else:
|
|
path = ""
|
|
dir, base = os.path.split(path)
|
|
head, tail = os.path.splitext(base)
|
|
if not tail:
|
|
tail = ".py"
|
|
self.globvar.set(os.path.join(dir, "*" + tail))
|
|
|
|
def create_entries(self):
|
|
SearchDialogBase.create_entries(self)
|
|
self.globent = self.make_entry("In files:", self.globvar)
|
|
|
|
def create_other_buttons(self):
|
|
f = self.make_frame()
|
|
|
|
btn = Checkbutton(f, anchor="w",
|
|
variable=self.recvar,
|
|
text="Recurse down subdirectories")
|
|
btn.pack(side="top", fill="both")
|
|
btn.select()
|
|
|
|
def create_command_buttons(self):
|
|
SearchDialogBase.create_command_buttons(self)
|
|
self.make_button("Search Files", self.default_command, 1)
|
|
|
|
def default_command(self, event=None):
|
|
prog = self.engine.getprog()
|
|
if not prog:
|
|
return
|
|
path = self.globvar.get()
|
|
if not path:
|
|
self.top.bell()
|
|
return
|
|
from OutputWindow import OutputWindow
|
|
save = sys.stdout
|
|
try:
|
|
sys.stdout = OutputWindow(self.flist)
|
|
self.grep_it(prog, path)
|
|
finally:
|
|
sys.stdout = save
|
|
|
|
def grep_it(self, prog, path):
|
|
dir, base = os.path.split(path)
|
|
list = self.findfiles(dir, base, self.recvar.get())
|
|
list.sort()
|
|
self.close()
|
|
pat = self.engine.getpat()
|
|
print "Searching %s in %s ..." % (`pat`, path)
|
|
hits = 0
|
|
for fn in list:
|
|
try:
|
|
f = open(fn)
|
|
except IOError, msg:
|
|
print msg
|
|
continue
|
|
lineno = 0
|
|
while 1:
|
|
block = f.readlines(100000)
|
|
if not block:
|
|
break
|
|
for line in block:
|
|
lineno = lineno + 1
|
|
if line[-1:] == '\n':
|
|
line = line[:-1]
|
|
if prog.search(line):
|
|
sys.stdout.write("%s: %s: %s\n" % (fn, lineno, line))
|
|
hits = hits + 1
|
|
if hits:
|
|
if hits == 1:
|
|
s = ""
|
|
else:
|
|
s = "s"
|
|
print "Found", hits, "hit%s." % s
|
|
print "(Hint: right-click to open locations.)"
|
|
else:
|
|
print "No hits."
|
|
|
|
def findfiles(self, dir, base, rec):
|
|
try:
|
|
names = os.listdir(dir or os.curdir)
|
|
except os.error, msg:
|
|
print msg
|
|
return []
|
|
list = []
|
|
subdirs = []
|
|
for name in names:
|
|
fn = os.path.join(dir, name)
|
|
if os.path.isdir(fn):
|
|
subdirs.append(fn)
|
|
else:
|
|
if fnmatch.fnmatch(name, base):
|
|
list.append(fn)
|
|
if rec:
|
|
for subdir in subdirs:
|
|
list.extend(self.findfiles(subdir, base, rec))
|
|
return list
|
|
|
|
def close(self, event=None):
|
|
if self.top:
|
|
self.top.grab_release()
|
|
self.top.withdraw()
|