Home | History | Annotate | Download | only in lib-tk 
      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()
    113