1 files changed, 102 insertions, 0 deletions
diff --git a/Lib/idlelib/search.py b/Lib/idlelib/search.py
new file mode 100644
index 0000000..4b90659
--- /dev/null
+++ b/Lib/idlelib/search.py
@@ -0,0 +1,102 @@
+from tkinter import TclError
+
+from idlelib import searchengine
+from idlelib.searchbase import SearchDialogBase
+
+def _setup(text):
+    "Create or find the singleton SearchDialog instance."
+    root = text._root()
+    engine = searchengine.get(root)
+    if not hasattr(engine, "_searchdialog"):
+        engine._searchdialog = SearchDialog(root, engine)
+    return engine._searchdialog
+
+def find(text):
+    "Handle the editor edit menu item and corresponding event."
+    pat = text.get("sel.first", "sel.last")
+    return _setup(text).open(text, pat)  # Open is inherited from SDBase.
+
+def find_again(text):
+    "Handle the editor edit menu item and corresponding event."
+    return _setup(text).find_again(text)
+
+def find_selection(text):
+    "Handle the editor edit menu item and corresponding event."
+    return _setup(text).find_selection(text)
+
+
+class SearchDialog(SearchDialogBase):
+
+    def create_widgets(self):
+        SearchDialogBase.create_widgets(self)
+        self.make_button("Find Next", self.default_command, 1)
+
+    def default_command(self, event=None):
+        if not self.engine.getprog():
+            return
+        self.find_again(self.text)
+
+    def find_again(self, text):
+        if not self.engine.getpat():
+            self.open(text)
+            return False
+        if not self.engine.getprog():
+            return False
+        res = self.engine.search_text(text)
+        if res:
+            line, m = res
+            i, j = m.span()
+            first = "%d.%d" % (line, i)
+            last = "%d.%d" % (line, j)
+            try:
+                selfirst = text.index("sel.first")
+                sellast = text.index("sel.last")
+                if selfirst == first and sellast == last:
+                    self.bell()
+                    return False
+            except TclError:
+                pass
+            text.tag_remove("sel", "1.0", "end")
+            text.tag_add("sel", first, last)
+            text.mark_set("insert", self.engine.isback() and first or last)
+            text.see("insert")
+            return True
+        else:
+            self.bell()
+            return False
+
+    def find_selection(self, text):
+        pat = text.get("sel.first", "sel.last")
+        if pat:
+            self.engine.setcookedpat(pat)
+        return self.find_again(text)
+
+
+def _search_dialog(parent):  # htest #
+    "Display search test box."
+    from tkinter import Toplevel, Text
+    from tkinter.ttk import Button
+
+    box = Toplevel(parent)
+    box.title("Test SearchDialog")
+    x, y = map(int, parent.geometry().split('+')[1:])
+    box.geometry("+%d+%d" % (x, y + 175))
+    text = Text(box, inactiveselectbackground='gray')
+    text.pack()
+    text.insert("insert","This is a sample string.\n"*5)
+
+    def show_find():
+        text.tag_add('sel', '1.0', 'end')
+        _setup(text).open(text)
+        text.tag_remove('sel', '1.0', 'end')
+
+    button = Button(box, text="Search (selection ignored)", command=show_find)
+    button.pack()
+
+if __name__ == '__main__':
+    import unittest
+    unittest.main('idlelib.idle_test.test_search',
+                  verbosity=2, exit=False)
+
+    from idlelib.idle_test.htest import run
+    run(_search_dialog)