1 import os

     2 import fnmatch

     3 import sys

     4 from Tkinter import *

     5 import SearchEngine

     6 from SearchDialogBase import SearchDialogBase

     7 

     8 def grep(text, io=None, flist=None):

     9     root = text._root()

    10     engine = SearchEngine.get(root)

    11     if not hasattr(engine, "_grepdialog"):

    12         engine._grepdialog = GrepDialog(root, engine, flist)

    13     dialog = engine._grepdialog

    14     searchphrase = text.get("sel.first", "sel.last")

    15     dialog.open(text, searchphrase, io)

    16 

    17 class GrepDialog(SearchDialogBase):

    18 

    19     title = "Find in Files Dialog"

    20     icon = "Grep"

    21     needwrapbutton = 0

    22 

    23     def __init__(self, root, engine, flist):

    24         SearchDialogBase.__init__(self, root, engine)

    25         self.flist = flist

    26         self.globvar = StringVar(root)

    27         self.recvar = BooleanVar(root)

    28 

    29     def open(self, text, searchphrase, io=None):

    30         SearchDialogBase.open(self, text, searchphrase)

    31         if io:

    32             path = io.filename or ""

    33         else:

    34             path = ""

    35         dir, base = os.path.split(path)

    36         head, tail = os.path.splitext(base)

    37         if not tail:

    38             tail = ".py"

    39         self.globvar.set(os.path.join(dir, "*" + tail))

    40 

    41     def create_entries(self):

    42         SearchDialogBase.create_entries(self)

    43         self.globent = self.make_entry("In files:", self.globvar)

    44 

    45     def create_other_buttons(self):

    46         f = self.make_frame()

    47 

    48         btn = Checkbutton(f, anchor="w",

    49                 variable=self.recvar,

    50                 text="Recurse down subdirectories")

    51         btn.pack(side="top", fill="both")

    52         btn.select()

    53 

    54     def create_command_buttons(self):

    55         SearchDialogBase.create_command_buttons(self)

    56         self.make_button("Search Files", self.default_command, 1)

    57 

    58     def default_command(self, event=None):

    59         prog = self.engine.getprog()

    60         if not prog:

    61             return

    62         path = self.globvar.get()

    63         if not path:

    64             self.top.bell()

    65             return

    66         from OutputWindow import OutputWindow

    67         save = sys.stdout

    68         try:

    69             sys.stdout = OutputWindow(self.flist)

    70             self.grep_it(prog, path)

    71         finally:

    72             sys.stdout = save

    73 

    74     def grep_it(self, prog, path):

    75         dir, base = os.path.split(path)

    76         list = self.findfiles(dir, base, self.recvar.get())

    77         list.sort()

    78         self.close()

    79         pat = self.engine.getpat()

    80         print "Searching %r in %s ..." % (pat, path)

    81         hits = 0

    82         for fn in list:

    83             try:

    84                 f = open(fn)

    85             except IOError, msg:

    86                 print msg

    87                 continue

    88             lineno = 0

    89             while 1:

    90                 block = f.readlines(100000)

    91                 if not block:

    92                     break

    93                 for line in block:

    94                     lineno = lineno + 1

    95                     if line[-1:] == '\n':

    96                         line = line[:-1]

    97                     if prog.search(line):

    98                         sys.stdout.write("%s: %s: %s\n" % (fn, lineno, line))

    99                         hits = hits + 1

   100         if hits:

   101             if hits == 1:

   102                 s = ""

   103             else:

   104                 s = "s"

   105             print "Found", hits, "hit%s." % s

   106             print "(Hint: right-click to open locations.)"

   107         else:

   108             print "No hits."

   109 

   110     def findfiles(self, dir, base, rec):

   111         try:

   112             names = os.listdir(dir or os.curdir)

   113         except os.error, msg:

   114             print msg

   115             return []

   116         list = []

   117         subdirs = []

   118         for name in names:

   119             fn = os.path.join(dir, name)

   120             if os.path.isdir(fn):

   121                 subdirs.append(fn)

   122             else:

   123                 if fnmatch.fnmatch(name, base):

   124                     list.append(fn)

   125         if rec:

   126             for subdir in subdirs:

   127                 list.extend(self.findfiles(subdir, base, rec))

   128         return list

   129 

   130     def close(self, event=None):

   131         if self.top:

   132             self.top.grab_release()

   133             self.top.withdraw()