Uploaded Test files

venv/Lib/site-packages/pythonwin/pywin/Demos/hiertest.py

@@ -0,0 +1,104 @@
+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()