105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
|
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()
|