324 lines
7.5 KiB
Python
324 lines
7.5 KiB
Python
|
#
|
||
|
|
# An Introduction to Tkinter
|
||
|
|
# tkSimpleDialog.py
|
||
|
|
#
|
||
|
|
# Copyright (c) 1997 by Fredrik Lundh
|
||
|
|
#
|
||
|
|
# fredrik@pythonware.com
|
||
|
|
# http://www.pythonware.com
|
||
|
|
#
|
||
|
|
|
||
|
|
# --------------------------------------------------------------------
|
||
|
|
# dialog base class
|
||
|
|
|
||
|
|
'''Dialog boxes
|
||
|
|
|
||
|
|
This module handles dialog boxes. It contains the following
|
||
|
|
public symbols:
|
||
|
|
|
||
|
|
Dialog -- a base class for dialogs
|
||
|
|
|
||
|
|
askinteger -- get an integer from the user
|
||
|
|
|
||
|
|
askfloat -- get a float from the user
|
||
|
|
|
||
|
|
askstring -- get a string from the user
|
||
|
|
'''
|
||
|
|
|
||
|
|
from Tkinter import *
|
||
|
|
|
||
|
|
class Dialog(Toplevel):
|
||
|
|
|
||
|
|
'''Class to open dialogs.
|
||
|
|
|
||
|
|
This class is intended as a base class for custom dialogs
|
||
|
|
'''
|
||
|
|
|
||
|
|
def __init__(self, parent, title = None):
|
||
|
|
|
||
|
|
'''Initialize a dialog.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
parent -- a parent window (the application window)
|
||
|
|
|
||
|
|
title -- the dialog title
|
||
|
|
'''
|
||
|
|
Toplevel.__init__(self, parent)
|
||
|
|
|
||
|
|
self.withdraw() # remain invisible for now
|
||
|
|
# If the master is not viewable, don't
|
||
|
|
# make the child transient, or else it
|
||
|
|
# would be opened withdrawn
|
||
|
|
if parent.winfo_viewable():
|
||
|
|
self.transient(parent)
|
||
|
|
|
||
|
|
if title:
|
||
|
|
self.title(title)
|
||
|
|
|
||
|
|
self.parent = parent
|
||
|
|
|
||
|
|
self.result = None
|
||
|
|
|
||
|
|
body = Frame(self)
|
||
|
|
self.initial_focus = self.body(body)
|
||
|
|
body.pack(padx=5, pady=5)
|
||
|
|
|
||
|
|
self.buttonbox()
|
||
|
|
|
||
|
|
|
||
|
|
if not self.initial_focus:
|
||
|
|
self.initial_focus = self
|
||
|
|
|
||
|
|
self.protocol("WM_DELETE_WINDOW", self.cancel)
|
||
|
|
|
||
|
|
if self.parent is not None:
|
||
|
|
self.geometry("+%d+%d" % (parent.winfo_rootx()+50,
|
||
|
|
parent.winfo_rooty()+50))
|
||
|
|
|
||
|
|
self.deiconify() # become visibile now
|
||
|
|
|
||
|
|
self.initial_focus.focus_set()
|
||
|
|
|
||
|
|
# wait for window to appear on screen before calling grab_set
|
||
|
|
self.wait_visibility()
|
||
|
|
self.grab_set()
|
||
|
|
self.wait_window(self)
|
||
|
|
|
||
|
|
def destroy(self):
|
||
|
|
'''Destroy the window'''
|
||
|
|
self.initial_focus = None
|
||
|
|
Toplevel.destroy(self)
|
||
|
|
|
||
|
|
#
|
||
|
|
# construction hooks
|
||
|
|
|
||
|
|
def body(self, master):
|
||
|
|
'''create dialog body.
|
||
|
|
|
||
|
|
return widget that should have initial focus.
|
||
|
|
This method should be overridden, and is called
|
||
|
|
by the __init__ method.
|
||
|
|
'''
|
||
|
|
pass
|
||
|
|
|
||
|
|
def buttonbox(self):
|
||
|
|
'''add standard button box.
|
||
|
|
|
||
|
|
override if you do not want the standard buttons
|
||
|
|
'''
|
||
|
|
|
||
|
|
box = Frame(self)
|
||
|
|
|
||
|
|
w = Button(box, text="OK", width=10, command=self.ok, default=ACTIVE)
|
||
|
|
w.pack(side=LEFT, padx=5, pady=5)
|
||
|
|
w = Button(box, text="Cancel", width=10, command=self.cancel)
|
||
|
|
w.pack(side=LEFT, padx=5, pady=5)
|
||
|
|
|
||
|
|
self.bind("<Return>", self.ok)
|
||
|
|
self.bind("<Escape>", self.cancel)
|
||
|
|
|
||
|
|
box.pack()
|
||
|
|
|
||
|
|
#
|
||
|
|
# standard button semantics
|
||
|
|
|
||
|
|
def ok(self, event=None):
|
||
|
|
|
||
|
|
if not self.validate():
|
||
|
|
self.initial_focus.focus_set() # put focus back
|
||
|
|
return
|
||
|
|
|
||
|
|
self.withdraw()
|
||
|
|
self.update_idletasks()
|
||
|
|
|
||
|
|
try:
|
||
|
|
self.apply()
|
||
|
|
finally:
|
||
|
|
self.cancel()
|
||
|
|
|
||
|
|
def cancel(self, event=None):
|
||
|
|
|
||
|
|
# put focus back to the parent window
|
||
|
|
if self.parent is not None:
|
||
|
|
self.parent.focus_set()
|
||
|
|
self.destroy()
|
||
|
|
|
||
|
|
#
|
||
|
|
# command hooks
|
||
|
|
|
||
|
|
def validate(self):
|
||
|
|
'''validate the data
|
||
|
|
|
||
|
|
This method is called automatically to validate the data before the
|
||
|
|
dialog is destroyed. By default, it always validates OK.
|
||
|
|
'''
|
||
|
|
|
||
|
|
return 1 # override
|
||
|
|
|
||
|
|
def apply(self):
|
||
|
|
'''process the data
|
||
|
|
|
||
|
|
This method is called automatically to process the data, *after*
|
||
|
|
the dialog is destroyed. By default, it does nothing.
|
||
|
|
'''
|
||
|
|
|
||
|
|
pass # override
|
||
|
|
|
||
|
|
|
||
|
|
# --------------------------------------------------------------------
|
||
|
|
# convenience dialogues
|
||
|
|
|
||
|
|
class _QueryDialog(Dialog):
|
||
|
|
|
||
|
|
def __init__(self, title, prompt,
|
||
|
|
initialvalue=None,
|
||
|
|
minvalue = None, maxvalue = None,
|
||
|
|
parent = None):
|
||
|
|
|
||
|
|
if not parent:
|
||
|
|
import Tkinter
|
||
|
|
parent = Tkinter._default_root
|
||
|
|
|
||
|
|
self.prompt = prompt
|
||
|
|
self.minvalue = minvalue
|
||
|
|
self.maxvalue = maxvalue
|
||
|
|
|
||
|
|
self.initialvalue = initialvalue
|
||
|
|
|
||
|
|
Dialog.__init__(self, parent, title)
|
||
|
|
|
||
|
|
def destroy(self):
|
||
|
|
self.entry = None
|
||
|
|
Dialog.destroy(self)
|
||
|
|
|
||
|
|
def body(self, master):
|
||
|
|
|
||
|
|
w = Label(master, text=self.prompt, justify=LEFT)
|
||
|
|
w.grid(row=0, padx=5, sticky=W)
|
||
|
|
|
||
|
|
self.entry = Entry(master, name="entry")
|
||
|
|
self.entry.grid(row=1, padx=5, sticky=W+E)
|
||
|
|
|
||
|
|
if self.initialvalue is not None:
|
||
|
|
self.entry.insert(0, self.initialvalue)
|
||
|
|
self.entry.select_range(0, END)
|
||
|
|
|
||
|
|
return self.entry
|
||
|
|
|
||
|
|
def validate(self):
|
||
|
|
|
||
|
|
import tkMessageBox
|
||
|
|
|
||
|
|
try:
|
||
|
|
result = self.getresult()
|
||
|
|
except ValueError:
|
||
|
|
tkMessageBox.showwarning(
|
||
|
|
"Illegal value",
|
||
|
|
self.errormessage + "\nPlease try again",
|
||
|
|
parent = self
|
||
|
|
)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
if self.minvalue is not None and result < self.minvalue:
|
||
|
|
tkMessageBox.showwarning(
|
||
|
|
"Too small",
|
||
|
|
"The allowed minimum value is %s. "
|
||
|
|
"Please try again." % self.minvalue,
|
||
|
|
parent = self
|
||
|
|
)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
if self.maxvalue is not None and result > self.maxvalue:
|
||
|
|
tkMessageBox.showwarning(
|
||
|
|
"Too large",
|
||
|
|
"The allowed maximum value is %s. "
|
||
|
|
"Please try again." % self.maxvalue,
|
||
|
|
parent = self
|
||
|
|
)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
self.result = result
|
||
|
|
|
||
|
|
return 1
|
||
|
|
|
||
|
|
|
||
|
|
class _QueryInteger(_QueryDialog):
|
||
|
|
errormessage = "Not an integer."
|
||
|
|
def getresult(self):
|
||
|
|
return int(self.entry.get())
|
||
|
|
|
||
|
|
def askinteger(title, prompt, **kw):
|
||
|
|
'''get an integer from the user
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
title -- the dialog title
|
||
|
|
prompt -- the label text
|
||
|
|
**kw -- see SimpleDialog class
|
||
|
|
|
||
|
|
Return value is an integer
|
||
|
|
'''
|
||
|
|
d = _QueryInteger(title, prompt, **kw)
|
||
|
|
return d.result
|
||
|
|
|
||
|
|
class _QueryFloat(_QueryDialog):
|
||
|
|
errormessage = "Not a floating point value."
|
||
|
|
def getresult(self):
|
||
|
|
return float(self.entry.get())
|
||
|
|
|
||
|
|
def askfloat(title, prompt, **kw):
|
||
|
|
'''get a float from the user
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
title -- the dialog title
|
||
|
|
prompt -- the label text
|
||
|
|
**kw -- see SimpleDialog class
|
||
|
|
|
||
|
|
Return value is a float
|
||
|
|
'''
|
||
|
|
d = _QueryFloat(title, prompt, **kw)
|
||
|
|
return d.result
|
||
|
|
|
||
|
|
class _QueryString(_QueryDialog):
|
||
|
|
def __init__(self, *args, **kw):
|
||
|
|
if "show" in kw:
|
||
|
|
self.__show = kw["show"]
|
||
|
|
del kw["show"]
|
||
|
|
else:
|
||
|
|
self.__show = None
|
||
|
|
_QueryDialog.__init__(self, *args, **kw)
|
||
|
|
|
||
|
|
def body(self, master):
|
||
|
|
entry = _QueryDialog.body(self, master)
|
||
|
|
if self.__show is not None:
|
||
|
|
entry.configure(show=self.__show)
|
||
|
|
return entry
|
||
|
|
|
||
|
|
def getresult(self):
|
||
|
|
return self.entry.get()
|
||
|
|
|
||
|
|
def askstring(title, prompt, **kw):
|
||
|
|
'''get a string from the user
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
title -- the dialog title
|
||
|
|
prompt -- the label text
|
||
|
|
**kw -- see SimpleDialog class
|
||
|
|
|
||
|
|
Return value is a string
|
||
|
|
'''
|
||
|
|
d = _QueryString(title, prompt, **kw)
|
||
|
|
return d.result
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
root = Tk()
|
||
|
|
root.update()
|
||
|
|
|
||
|
|
print askinteger("Spam", "Egg count", initialvalue=12*12)
|
||
|
|
print askfloat("Spam", "Egg weight\n(in tons)", minvalue=1, maxvalue=100)
|
||
|
|
print askstring("Spam", "Egg label")
|