home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / GrepDialog.py < prev    next >
Text File  |  2003-12-30  |  4KB  |  134 lines
  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 %s 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()
  134.