Uploaded Test files

venv/Lib/site-packages/pythonwin/pywin/mfc/dialog.py

@@ -0,0 +1,241 @@
+""" \
+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 __bool__(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 list(self.data.keys())
+	def items(self): return list(self.data.items())
+	def values(self): return list(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']