import win32ui
import os
import commctrl

from pywin.tools import hierlist
from pywin.mfc import docview, window

# directory listbox
# This has obvious limitations - doesnt track subdirs, etc.  Demonstrates
# simple use of Python code for querying the tree as needed.
# Only use strings, and lists of strings (from curdir())
class DirHierList(hierlist.HierList):
	def __init__(self, root, listBoxID = win32ui.IDC_LIST1):
		hierlist.HierList.__init__(self, root, win32ui.IDB_HIERFOLDERS, listBoxID)
	def GetText(self, item):
		return os.path.basename(item)
	def GetSubList(self, item):
		if os.path.isdir(item):
			ret = [os.path.join(item, fname) for fname in os.listdir(item)]
		else:
			ret = None
		return ret
	# if the item is a dir, it is expandable.
	def IsExpandable(self, item):
		return os.path.isdir(item)
	def GetSelectedBitmapColumn(self, item):
		return self.GetBitmapColumn(item)+6 # Use different color for selection

class TestDocument(docview.Document):
	def __init__(self, template):
		docview.Document.__init__(self, template)
		self.hierlist = hierlist.HierListWithItems(HLIFileDir("\\"), win32ui.IDB_HIERFOLDERS, win32ui.AFX_IDW_PANE_FIRST)

class HierListView(docview.TreeView):
	def OnInitialUpdate(self):
		rc = self._obj_.OnInitialUpdate()
		self.hierList = self.GetDocument().hierlist
		self.hierList.HierInit(self.GetParent())
		self.hierList.SetStyle(commctrl.TVS_HASLINES | commctrl.TVS_LINESATROOT | commctrl.TVS_HASBUTTONS)
		return rc

class HierListFrame(window.MDIChildWnd):
	pass

def GetTestRoot():
	tree1 = ('Tree 1',[('Item 1','Item 1 data'),'Item 2',3])
	tree2 = ('Tree 2',[('Item 2.1','Item 2 data'),'Item 2.2',2.3])
	return ('Root',[tree1,tree2,'Item 3'])

def demoboth():
	template = docview.DocTemplate(win32ui.IDR_PYTHONTYPE, TestDocument, HierListFrame, HierListView)
	template.OpenDocumentFile(None).SetTitle("Hierlist demo")
	
	demomodeless()

def demomodeless():
	testList2=DirHierList("\\")
	dlg=hierlist.HierDialog('hier list test',testList2)
	dlg.CreateWindow()

def demodlg	():
	testList2=DirHierList("\\")
	dlg=hierlist.HierDialog('hier list test',testList2)
	dlg.DoModal()

def demo():
	template = docview.DocTemplate(win32ui.IDR_PYTHONTYPE, TestDocument, HierListFrame, HierListView)
	template.OpenDocumentFile(None).SetTitle("Hierlist demo")

#
# Demo/Test for HierList items.
#
# Easy to make a better directory program.
#
class HLIFileDir(hierlist.HierListItem):
	def __init__( self, filename ):
		self.filename = filename
		hierlist.HierListItem.__init__(self)
	def GetText(self):
		try:
			return "%-20s %d bytes" % (os.path.basename(self.filename), os.stat(self.filename)[6])
		except os.error as details:
			return "%-20s - %s" % (self.filename, details[1])

	def IsExpandable(self):
		return os.path.isdir(self.filename)
	def GetSubList(self):
		ret = []
		for newname in os.listdir(self.filename):
			if newname not in ['.', '..']:
				ret.append( HLIFileDir( os.path.join(self.filename,newname ) ) )
		return ret


def demohli():
	template = docview.DocTemplate(win32ui.IDR_PYTHONTYPE, TestDocument, hierlist.HierListFrame, hierlist.HierListView)
	template.OpenDocumentFile(None).SetTitle("Hierlist demo")

if __name__=='__main__':
	import demoutils
	if demoutils.HaveGoodGUI():
		demoboth()
	else:
		demodlg()