← Back to branch summary

~ubuntu-branches/ubuntu/karmic/pypy/karmic

« back to all changes in this revision

Viewing changes to lib-python/2.4.1/idlelib/GrepDialog.py

  • Committer: Bazaar Package Importer
  • Author(s): Alexandre Fayolle
  • Date: 2007-04-13 09:33:09 UTC
  • Revision ID: james.westby@ubuntu.com-20070413093309-yoojh4jcoocu2krz
Tags: upstream-1.0.0
Import upstream version 1.0.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib-python/2.4.1/idlelib/GrepDialog.py
 
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()