cpython/Tools/idle/SearchBinding.py

90 lines
2.4 KiB
Python

import string
import re
import tkSimpleDialog
import tkMessageBox
class SearchBinding:
def __init__(self, text):
self.text = text
self.pat = ""
self.prog = None
self.text.bind("<<find>>", self.find_event)
self.text.bind("<<find-next>>", self.find_next_event)
self.text.bind("<<find-same>>", self.find_same_event)
self.text.bind("<<goto-line>>", self.goto_line_event)
def find_event(self, event):
default = self.text.get("self.first", "sel.last") or self.pat
new = tkSimpleDialog.askstring("Find",
"Regular Expression:",
initialvalue=default,
parent=self.text)
if not new:
return "break"
self.pat = new
try:
self.prog = re.compile(self.pat)
except re.error, msg:
tkMessageBox.showerror("RE error", str(msg),
master=self.text)
return "break"
return self.find_next_event(event)
def find_same_event(self, event):
pat = self.text.get("sel.first", "sel.last")
if not pat:
return self.find_event(event)
self.pat = re.escape(pat)
self.prog = None
try:
self.prog = re.compile(self.pat)
except re.error, msg:
tkMessageBox.showerror("RE error", str(message),
master=self.text)
return "break"
self.text.mark_set("insert", "sel.last")
return self.find_next_event(event)
def find_next_event(self, event):
if not self.pat:
return self.find_event(event)
if not self.prog:
self.text.bell()
##print "No program"
return "break"
line, col = map(int,
string.split(self.text.index("insert"), "."))
chars = self.text.get("%d.0" % line, "%d.0" % (line+1))
while chars:
m = self.prog.search(chars, col)
if m:
i, j = m.span()
self.text.mark_set("insert",
"%d.%d" % (line, j))
self.text.tag_remove("sel", "1.0", "end")
self.text.tag_add("sel",
"%d.%d" % (line, i),
"%d.%d" % (line, j))
self.text.see("insert")
break
line = line + 1
col = 0
chars = self.text.get("%d.0" % line, "%d.0" % (line+1))
else:
# Not found
self.text.bell()
return "break"
def goto_line_event(self, event):
lineno = tkSimpleDialog.askinteger("Goto",
"Go to line number:",
parent=self.text)
if lineno is None:
return "break"
if lineno <= 0:
self.text.bell()
return "break"
self.text.mark_set("insert", "%d.0" % lineno)
self.text.see("insert")