Lib/idlelib/SearchDialog.py

   1 from Tkinter import *
   2 import SearchEngine
   3 from SearchDialogBase import SearchDialogBase
   4
   5
   6 def _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
  13 def find(text):
  14     pat = text.get("sel.first", "sel.last")
  15     return _setup(text).open(text,pat)
  16
  17 def find_again(text):
  18     return _setup(text).find_again(text)
  19
  20 def find_selection(text):
  21     return _setup(text).find_selection(text)
  22
  23 class 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)