86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
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"
|
|
self.text.mark_set("find", "insert")
|
|
while 1:
|
|
chars = self.text.get("find", "find lineend +1c")
|
|
##print "Searching", `chars`
|
|
if not chars:
|
|
self.text.bell()
|
|
##print "end of buffer"
|
|
break
|
|
m = self.prog.search(chars)
|
|
if m:
|
|
i, j = m.span()
|
|
self.text.mark_set("insert", "find +%dc" % j)
|
|
self.text.mark_set("find", "find +%dc" % i)
|
|
self.text.tag_remove("sel", "1.0", "end")
|
|
self.text.tag_add("sel", "find", "insert")
|
|
self.text.see("insert")
|
|
break
|
|
self.text.mark_set("find", "find lineend +1c")
|
|
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")
|