""" \
Base class for Dialogs. Also contains a few useful utility functions
"""
# dialog.py
# Python class for Dialog Boxes in PythonWin.
import win32ui
import win32con
# sob - 2to3 doesn't see this as a relative import :(
from pywin.mfc import window
def dllFromDll(dllid):
" given a 'dll' (maybe a dll, filename, etc), return a DLL object "
if dllid==None:
return None
elif type('')==type(dllid):
return win32ui.LoadLibrary(dllid)
else:
try:
dllid.GetFileName()
except AttributeError:
raise TypeError("DLL parameter must be None, a filename or a dll object")
return dllid
class Dialog(window.Wnd):
" Base class for a dialog"
def __init__( self, id, dllid=None ):
""" id is the resource ID, or a template
dllid may be None, a dll object, or a string with a dll name """
# must take a reference to the DLL until InitDialog.
self.dll=dllFromDll(dllid)
if type(id)==type([]): # a template
dlg=win32ui.CreateDialogIndirect(id)
else:
dlg=win32ui.CreateDialog(id, self.dll)
window.Wnd.__init__(self, dlg)
self.HookCommands()
self.bHaveInit = None
def HookCommands(self):
pass
def OnAttachedObjectDeath(self):
self.data = self._obj_.data
window.Wnd.OnAttachedObjectDeath(self)
# provide virtuals.
def OnOK(self):
self._obj_.OnOK()
def OnCancel(self):
self._obj_.OnCancel()
def OnInitDialog(self):
self.bHaveInit = 1
if self._obj_.data:
self._obj_.UpdateData(0)
return 1 # I did NOT set focus to a child window.
def OnDestroy(self,msg):
self.dll = None # theoretically not needed if object destructs normally.
# DDX support
def AddDDX( self, *args ):
self._obj_.datalist.append(args)
# Make a dialog object look like a dictionary for the DDX support
def __nonzero__(self):
return True
def __len__(self): return len(self.data)
def __getitem__(self, key): return self.data[key]
def __setitem__(self, key, item): self._obj_.data[key] = item# self.UpdateData(0)
def keys(self): return self.data.keys()
def items(self): return self.data.items()
def values(self): return self.data.values()
# XXX - needs py3k work!
def has_key(self, key): return key in self.data
class PrintDialog(Dialog):
" Base class for a print dialog"
def __init__(self, pInfo, dlgID,
printSetupOnly = 0,
flags=(win32ui.PD_ALLPAGES|
win32ui.PD_USEDEVMODECOPIES|
win32ui.PD_NOPAGENUMS|
win32ui.PD_HIDEPRINTTOFILE|
win32ui.PD_NOSELECTION),
parent=None,
dllid=None):
self.dll=dllFromDll(dllid)
if type(dlgID)==type([]): # a template
raise TypeError("dlgID parameter must be an integer resource ID")
dlg=win32ui.CreatePrintDialog(dlgID, printSetupOnly,
flags, parent,
self.dll)
window.Wnd.__init__(self, dlg)
self.HookCommands()
self.bHaveInit = None
self.pInfo = pInfo
# init values (if PrintSetup is called, values still available)
flags = pInfo.GetFlags()
self['toFile'] = (flags&win32ui.PD_PRINTTOFILE != 0)
self['direct'] = pInfo.GetDirect()
self['preview'] = pInfo.GetPreview()
self['continuePrinting'] = pInfo.GetContinuePrinting()
self['curPage'] = pInfo.GetCurPage()
self['numPreviewPages'] = pInfo.GetNumPreviewPages()
self['userData'] = pInfo.GetUserData()
self['draw'] = pInfo.GetDraw()
self['pageDesc'] = pInfo.GetPageDesc()
self['minPage'] = pInfo.GetMinPage()
self['maxPage'] = pInfo.GetMaxPage()
self['offsetPage'] = pInfo.GetOffsetPage()
self['fromPage'] = pInfo.GetFromPage()
self['toPage'] = pInfo.GetToPage()
# these values updated after OnOK
self['copies'] = 0
self['deviceName'] = ''
self['driverName'] = ''
self['printAll'] = 0
self['printCollate'] = 0
self['printRange'] = 0
self['printSelection'] = 0
def OnInitDialog(self):
self.pInfo.CreatePrinterDC() # This also sets the hDC of the pInfo structure.
return self._obj_.OnInitDialog()
def OnCancel(self):
del self.pInfo
def OnOK(self):
'''DoModal has finished. Can now access the users choices'''
self._obj_.OnOK()
pInfo = self.pInfo
# user values
flags = pInfo.GetFlags()
self['toFile'] = (flags&win32ui.PD_PRINTTOFILE != 0)
self['direct'] = pInfo.GetDirect()
self['preview'] = pInfo.GetPreview()
self['continuePrinting'] = pInfo.GetContinuePrinting()
self['curPage'] = pInfo.GetCurPage()
self['numPreviewPages'] = pInfo.GetNumPreviewPages()
self['userData'] = pInfo.GetUserData()
self['draw'] = pInfo.GetDraw()
self['pageDesc'] = pInfo.GetPageDesc()
self['minPage'] = pInfo.GetMinPage()
self['maxPage'] = pInfo.GetMaxPage()
self['offsetPage'] = pInfo.GetOffsetPage()
self['fromPage'] = pInfo.GetFromPage()
self['toPage'] = pInfo.GetToPage()
self['copies'] = pInfo.GetCopies()
self['deviceName'] = pInfo.GetDeviceName()
self['driverName'] = pInfo.GetDriverName()
self['printAll'] = pInfo.PrintAll()
self['printCollate'] = pInfo.PrintCollate()
self['printRange'] = pInfo.PrintRange()
self['printSelection'] = pInfo.PrintSelection()
del self.pInfo
class PropertyPage(Dialog):
" Base class for a Property Page"
def __init__( self, id, dllid=None, caption=0 ):
""" id is the resource ID
dllid may be None, a dll object, or a string with a dll name """
self.dll = dllFromDll(dllid)
if self.dll:
oldRes = win32ui.SetResource(self.dll)
if type(id)==type([]):
dlg=win32ui.CreatePropertyPageIndirect(id)
else:
dlg=win32ui.CreatePropertyPage(id, caption)
if self.dll:
win32ui.SetResource(oldRes)
# dont call dialog init!
window.Wnd.__init__(self, dlg)
self.HookCommands()
class PropertySheet(window.Wnd):
def __init__(self, caption, dll=None, pageList=None ):# parent=None, style,etc):
" Initialize a property sheet. pageList is a list of ID's "
# must take a reference to the DLL until InitDialog.
self.dll=dllFromDll(dll)
self.sheet = win32ui.CreatePropertySheet(caption)
window.Wnd.__init__(self, self.sheet)
if not pageList is None:
self.AddPage(pageList)
def OnInitDialog(self):
return self._obj_.OnInitDialog()
def DoModal(self):
if self.dll:
oldRes = win32ui.SetResource(self.dll)
rc = self.sheet.DoModal()
if self.dll:
win32ui.SetResource(oldRes)
return rc
def AddPage(self, pages):
if self.dll:
oldRes = win32ui.SetResource(self.dll)
try: # try list style access
pages[0]
isSeq = 1
except (TypeError,KeyError):
isSeq = 0
if isSeq:
for page in pages:
self.DoAddSinglePage(page)
else:
self.DoAddSinglePage(pages)
if self.dll:
win32ui.SetResource(oldRes)
def DoAddSinglePage(self, page):
"Page may be page, or int ID. Assumes DLL setup "
if type(page)==type(0):
self.sheet.AddPage(win32ui.CreatePropertyPage(page))
else:
self.sheet.AddPage(page)
# define some app utility functions.
def GetSimpleInput(prompt, defValue='', title=None ):
""" displays a dialog, and returns a string, or None if cancelled.
args prompt, defValue='', title=main frames title """
# uses a simple dialog to return a string object.
if title is None: title=win32ui.GetMainFrame().GetWindowText()
# 2to3 insists on converting 'Dialog.__init__' to 'tkinter.dialog...'
DlgBaseClass = Dialog
class DlgSimpleInput(DlgBaseClass):
def __init__(self, prompt, defValue, title ):
self.title=title
DlgBaseClass.__init__(self, win32ui.IDD_SIMPLE_INPUT)
self.AddDDX(win32ui.IDC_EDIT1,'result')
self.AddDDX(win32ui.IDC_PROMPT1, 'prompt')
self._obj_.data['result']=defValue
self._obj_.data['prompt']=prompt
def OnInitDialog(self):
self.SetWindowText(self.title)
return DlgBaseClass.OnInitDialog(self)
dlg=DlgSimpleInput( prompt, defValue, title)
if dlg.DoModal() != win32con.IDOK:
return None
return dlg['result']
|