l4/pkg/python/contrib/Lib/lib-tk/SimpleDialog.py

   1 """A simple but flexible modal dialog box."""
   2
   3
   4 from Tkinter import *
   5
   6
   7 class SimpleDialog:
   8
   9     def __init__(self, master,
  10                  text='', buttons=[], default=None, cancel=None,
  11                  title=None, class_=None):
  12         if class_:
  13             self.root = Toplevel(master, class_=class_)
  14         else:
  15             self.root = Toplevel(master)
  16         if title:
  17             self.root.title(title)
  18             self.root.iconname(title)
  19         self.message = Message(self.root, text=text, aspect=400)
  20         self.message.pack(expand=1, fill=BOTH)
  21         self.frame = Frame(self.root)
  22         self.frame.pack()
  23         self.num = default
  24         self.cancel = cancel
  25         self.default = default
  26         self.root.bind('<Return>', self.return_event)
  27         for num in range(len(buttons)):
  28             s = buttons[num]
  29             b = Button(self.frame, text=s,
  30                        command=(lambda self=self, num=num: self.done(num)))
  31             if num == default:
  32                 b.config(relief=RIDGE, borderwidth=8)
  33             b.pack(side=LEFT, fill=BOTH, expand=1)
  34         self.root.protocol('WM_DELETE_WINDOW', self.wm_delete_window)
  35         self._set_transient(master)
  36
  37     def _set_transient(self, master, relx=0.5, rely=0.3):
  38         widget = self.root
  39         widget.withdraw() # Remain invisible while we figure out the geometry
  40         widget.transient(master)
  41         widget.update_idletasks() # Actualize geometry information
  42         if master.winfo_ismapped():
  43             m_width = master.winfo_width()
  44             m_height = master.winfo_height()
  45             m_x = master.winfo_rootx()
  46             m_y = master.winfo_rooty()
  47         else:
  48             m_width = master.winfo_screenwidth()
  49             m_height = master.winfo_screenheight()
  50             m_x = m_y = 0
  51         w_width = widget.winfo_reqwidth()
  52         w_height = widget.winfo_reqheight()
  53         x = m_x + (m_width - w_width) * relx
  54         y = m_y + (m_height - w_height) * rely
  55         if x+w_width > master.winfo_screenwidth():
  56             x = master.winfo_screenwidth() - w_width
  57         elif x < 0:
  58             x = 0
  59         if y+w_height > master.winfo_screenheight():
  60             y = master.winfo_screenheight() - w_height
  61         elif y < 0:
  62             y = 0
  63         widget.geometry("+%d+%d" % (x, y))
  64         widget.deiconify() # Become visible at the desired location
  65
  66     def go(self):
  67         self.root.wait_visibility()
  68         self.root.grab_set()
  69         self.root.mainloop()
  70         self.root.destroy()
  71         return self.num
  72
  73     def return_event(self, event):
  74         if self.default is None:
  75             self.root.bell()
  76         else:
  77             self.done(self.default)
  78
  79     def wm_delete_window(self):
  80         if self.cancel is None:
  81             self.root.bell()
  82         else:
  83             self.done(self.cancel)
  84
  85     def done(self, num):
  86         self.num = num
  87         self.root.quit()
  88
  89
  90 if __name__ == '__main__':
  91
  92     def test():
  93         root = Tk()
  94         def doit(root=root):
  95             d = SimpleDialog(root,
  96                          text="This is a test dialog.  "
  97                               "Would this have been an actual dialog, "
  98                               "the buttons below would have been glowing "
  99                               "in soft pink light.\n"
 100                               "Do you believe this?",
 101                          buttons=["Yes", "No", "Cancel"],
 102                          default=0,
 103                          cancel=2,
 104                          title="Test Dialog")
 105             print d.go()
 106         t = Button(root, text='Test', command=doit)
 107         t.pack()
 108         q = Button(root, text='Quit', command=t.quit)
 109         q.pack()
 110         t.mainloop()
 111
 112     test()