""" \
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']