home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / GREPDIALOG.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  4.0 KB  |  136 lines
  1. import string
  2. import os
  3. import re
  4. import fnmatch
  5. import sys
  6. from Tkinter import *
  7. import tkMessageBox
  8. import SearchEngine
  9. from SearchDialogBase import SearchDialogBase
  10.  
  11. def grep(text, io=None, flist=None):
  12.     root = text._root()
  13.     engine = SearchEngine.get(root)
  14.     if not hasattr(engine, "_grepdialog"):
  15.         engine._grepdialog = GrepDialog(root, engine, flist)
  16.     dialog = engine._grepdialog
  17.     dialog.open(io)
  18.  
  19. class GrepDialog(SearchDialogBase):
  20.  
  21.     title = "Find in Files Dialog"
  22.     icon = "Grep"
  23.     needwrapbutton = 0
  24.  
  25.     def __init__(self, root, engine, flist):
  26.         SearchDialogBase.__init__(self, root, engine)
  27.         self.flist = flist
  28.         self.globvar = StringVar(root)
  29.         self.recvar = BooleanVar(root)
  30.  
  31.     def open(self, io=None):
  32.         SearchDialogBase.open(self, None)
  33.         if io:
  34.             path = io.filename or ""
  35.         else:
  36.             path = ""
  37.         dir, base = os.path.split(path)
  38.         head, tail = os.path.splitext(base)
  39.         if not tail:
  40.             tail = ".py"
  41.         self.globvar.set(os.path.join(dir, "*" + tail))
  42.  
  43.     def create_entries(self):
  44.         SearchDialogBase.create_entries(self)
  45.         self.globent = self.make_entry("In files:", self.globvar)
  46.  
  47.     def create_other_buttons(self):
  48.         f = self.make_frame()
  49.  
  50.         btn = Checkbutton(f, anchor="w",
  51.                 variable=self.recvar,
  52.                 text="Recurse down subdirectories")
  53.         btn.pack(side="top", fill="both")
  54.         btn.select()
  55.  
  56.     def create_command_buttons(self):
  57.         SearchDialogBase.create_command_buttons(self)
  58.         self.make_button("Search Files", self.default_command, 1)
  59.  
  60.     def default_command(self, event=None):
  61.         prog = self.engine.getprog()
  62.         if not prog:
  63.             return
  64.         path = self.globvar.get()
  65.         if not path:
  66.             self.top.bell()
  67.             return
  68.         from OutputWindow import OutputWindow
  69.         save = sys.stdout
  70.         try:
  71.             sys.stdout = OutputWindow(self.flist)
  72.             self.grep_it(prog, path)
  73.         finally:
  74.             sys.stdout = save
  75.  
  76.     def grep_it(self, prog, path):
  77.         dir, base = os.path.split(path)
  78.         list = self.findfiles(dir, base, self.recvar.get())
  79.         list.sort()
  80.         self.close()
  81.         pat = self.engine.getpat()
  82.         print "Searching %s in %s ..." % (`pat`, path)
  83.         hits = 0
  84.         for fn in list:
  85.             try:
  86.                 f = open(fn)
  87.             except IOError, msg:
  88.                 print msg
  89.                 continue
  90.             lineno = 0
  91.             while 1:
  92.                 block = f.readlines(100000)
  93.                 if not block:
  94.                     break
  95.                 for line in block:
  96.                     lineno = lineno + 1
  97.                     if line[-1:] == '\n':
  98.                         line = line[:-1]
  99.                     if prog.search(line):
  100.                         sys.stdout.write("%s: %s: %s\n" % (fn, lineno, line))
  101.                         hits = hits + 1
  102.         if hits:
  103.             if hits == 1:
  104.                 s = ""
  105.             else:
  106.                 s = "s"
  107.             print "Found", hits, "hit%s." % s
  108.             print "(Hint: right-click to open locations.)"
  109.         else:
  110.             print "No hits."
  111.  
  112.     def findfiles(self, dir, base, rec):
  113.         try:
  114.             names = os.listdir(dir or os.curdir)
  115.         except os.error, msg:
  116.             print msg
  117.             return []
  118.         list = []
  119.         subdirs = []
  120.         for name in names:
  121.             fn = os.path.join(dir, name)
  122.             if os.path.isdir(fn):
  123.                 subdirs.append(fn)
  124.             else:
  125.                 if fnmatch.fnmatch(name, base):
  126.                     list.append(fn)
  127.         if rec:
  128.             for subdir in subdirs:
  129.                 list.extend(self.findfiles(subdir, base, rec))
  130.         return list
  131.  
  132.     def close(self, event=None):
  133.         if self.top:
  134.             self.top.grab_release()
  135.             self.top.withdraw()
  136.