← 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/ReplaceDialog.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/ReplaceDialog.py
 
1
from Tkinter import *
 
2
import SearchEngine
 
3
from SearchDialogBase import SearchDialogBase
 
4
 
 
5
def replace(text):
 
6
    root = text._root()
 
7
    engine = SearchEngine.get(root)
 
8
    if not hasattr(engine, "_replacedialog"):
 
9
        engine._replacedialog = ReplaceDialog(root, engine)
 
10
    dialog = engine._replacedialog
 
11
    dialog.open(text)
 
12
 
 
13
class ReplaceDialog(SearchDialogBase):
 
14
 
 
15
    title = "Replace Dialog"
 
16
    icon = "Replace"
 
17
 
 
18
    def __init__(self, root, engine):
 
19
        SearchDialogBase.__init__(self, root, engine)
 
20
        self.replvar = StringVar(root)
 
21
 
 
22
    def open(self, text):
 
23
        SearchDialogBase.open(self, text)
 
24
        try:
 
25
            first = text.index("sel.first")
 
26
        except TclError:
 
27
            first = None
 
28
        try:
 
29
            last = text.index("sel.last")
 
30
        except TclError:
 
31
            last = None
 
32
        first = first or text.index("insert")
 
33
        last = last or first
 
34
        self.show_hit(first, last)
 
35
        self.ok = 1
 
36
 
 
37
    def create_entries(self):
 
38
        SearchDialogBase.create_entries(self)
 
39
        self.replent = self.make_entry("Replace with:", self.replvar)
 
40
 
 
41
    def create_command_buttons(self):
 
42
        SearchDialogBase.create_command_buttons(self)
 
43
        self.make_button("Find", self.find_it)
 
44
        self.make_button("Replace", self.replace_it)
 
45
        self.make_button("Replace+Find", self.default_command, 1)
 
46
        self.make_button("Replace All", self.replace_all)
 
47
 
 
48
    def find_it(self, event=None):
 
49
        self.do_find(0)
 
50
 
 
51
    def replace_it(self, event=None):
 
52
        if self.do_find(self.ok):
 
53
            self.do_replace()
 
54
 
 
55
    def default_command(self, event=None):
 
56
        if self.do_find(self.ok):
 
57
            self.do_replace()
 
58
            self.do_find(0)
 
59
 
 
60
    def replace_all(self, event=None):
 
61
        prog = self.engine.getprog()
 
62
        if not prog:
 
63
            return
 
64
        repl = self.replvar.get()
 
65
        text = self.text
 
66
        res = self.engine.search_text(text, prog)
 
67
        if not res:
 
68
            text.bell()
 
69
            return
 
70
        text.tag_remove("sel", "1.0", "end")
 
71
        text.tag_remove("hit", "1.0", "end")
 
72
        line = res[0]
 
73
        col = res[1].start()
 
74
        if self.engine.iswrap():
 
75
            line = 1
 
76
            col = 0
 
77
        ok = 1
 
78
        first = last = None
 
79
        # XXX ought to replace circular instead of top-to-bottom when wrapping
 
80
        text.undo_block_start()
 
81
        while 1:
 
82
            res = self.engine.search_forward(text, prog, line, col, 0, ok)
 
83
            if not res:
 
84
                break
 
85
            line, m = res
 
86
            chars = text.get("%d.0" % line, "%d.0" % (line+1))
 
87
            orig = m.group()
 
88
            new = m.expand(repl)
 
89
            i, j = m.span()
 
90
            first = "%d.%d" % (line, i)
 
91
            last = "%d.%d" % (line, j)
 
92
            if new == orig:
 
93
                text.mark_set("insert", last)
 
94
            else:
 
95
                text.mark_set("insert", first)
 
96
                if first != last:
 
97
                    text.delete(first, last)
 
98
                if new:
 
99
                    text.insert(first, new)
 
100
            col = i + len(new)
 
101
            ok = 0
 
102
        text.undo_block_stop()
 
103
        if first and last:
 
104
            self.show_hit(first, last)
 
105
        self.close()
 
106
 
 
107
    def do_find(self, ok=0):
 
108
        if not self.engine.getprog():
 
109
            return False
 
110
        text = self.text
 
111
        res = self.engine.search_text(text, None, ok)
 
112
        if not res:
 
113
            text.bell()
 
114
            return False
 
115
        line, m = res
 
116
        i, j = m.span()
 
117
        first = "%d.%d" % (line, i)
 
118
        last = "%d.%d" % (line, j)
 
119
        self.show_hit(first, last)
 
120
        self.ok = 1
 
121
        return True
 
122
 
 
123
    def do_replace(self):
 
124
        prog = self.engine.getprog()
 
125
        if not prog:
 
126
            return False
 
127
        text = self.text
 
128
        try:
 
129
            first = pos = text.index("sel.first")
 
130
            last = text.index("sel.last")
 
131
        except TclError:
 
132
            pos = None
 
133
        if not pos:
 
134
            first = last = pos = text.index("insert")
 
135
        line, col = SearchEngine.get_line_col(pos)
 
136
        chars = text.get("%d.0" % line, "%d.0" % (line+1))
 
137
        m = prog.match(chars, col)
 
138
        if not prog:
 
139
            return False
 
140
        new = m.expand(self.replvar.get())
 
141
        text.mark_set("insert", first)
 
142
        text.undo_block_start()
 
143
        if m.group():
 
144
            text.delete(first, last)
 
145
        if new:
 
146
            text.insert(first, new)
 
147
        text.undo_block_stop()
 
148
        self.show_hit(first, text.index("insert"))
 
149
        self.ok = 0
 
150
        return True
 
151
 
 
152
    def show_hit(self, first, last):
 
153
        text = self.text
 
154
        text.mark_set("insert", first)
 
155
        text.tag_remove("sel", "1.0", "end")
 
156
        text.tag_add("sel", first, last)
 
157
        text.tag_remove("hit", "1.0", "end")
 
158
        if first == last:
 
159
            text.tag_add("hit", first)
 
160
        else:
 
161
            text.tag_add("hit", first, last)
 
162
        text.see("insert")
 
163
        text.update_idletasks()
 
164
 
 
165
    def close(self, event=None):
 
166
        SearchDialogBase.close(self, event)
 
167
        self.text.tag_remove("hit", "1.0", "end")