1from Tkinter import * 2 3from idlelib import SearchEngine 4from idlelib.SearchDialogBase import SearchDialogBase 5 6def _setup(text): 7 root = text._root() 8 engine = SearchEngine.get(root) 9 if not hasattr(engine, "_searchdialog"): 10 engine._searchdialog = SearchDialog(root, engine) 11 return engine._searchdialog 12 13def find(text): 14 pat = text.get("sel.first", "sel.last") 15 return _setup(text).open(text,pat) 16 17def find_again(text): 18 return _setup(text).find_again(text) 19 20def find_selection(text): 21 return _setup(text).find_selection(text) 22 23class SearchDialog(SearchDialogBase): 24 25 def create_widgets(self): 26 f = SearchDialogBase.create_widgets(self) 27 self.make_button("Find", self.default_command, 1) 28 29 def default_command(self, event=None): 30 if not self.engine.getprog(): 31 return 32 if self.find_again(self.text): 33 self.close() 34 35 def find_again(self, text): 36 if not self.engine.getpat(): 37 self.open(text) 38 return False 39 if not self.engine.getprog(): 40 return False 41 res = self.engine.search_text(text) 42 if res: 43 line, m = res 44 i, j = m.span() 45 first = "%d.%d" % (line, i) 46 last = "%d.%d" % (line, j) 47 try: 48 selfirst = text.index("sel.first") 49 sellast = text.index("sel.last") 50 if selfirst == first and sellast == last: 51 text.bell() 52 return False 53 except TclError: 54 pass 55 text.tag_remove("sel", "1.0", "end") 56 text.tag_add("sel", first, last) 57 text.mark_set("insert", self.engine.isback() and first or last) 58 text.see("insert") 59 return True 60 else: 61 text.bell() 62 return False 63 64 def find_selection(self, text): 65 pat = text.get("sel.first", "sel.last") 66 if pat: 67 self.engine.setcookedpat(pat) 68 return self.find_again(text) 69