spacepaste

Paste details

reply | raw

  1.  
  2. import os
  3. import wx
  4. import wx.lib.customtreectrl as customtree
  6. try:
  7. from agw import customtreectrl as CT
  8. except ImportError: # if it's not there locally, try the wxPython lib.
  9. import wx.lib.agw.customtreectrl as CT
  11. #---- Recipe Code ----#
  13. class FileBrowser(customtree.CustomTreeCtrl):
  14. FOLDER, \
  15. ERROR, \
  16. FILE = range(3)
  17. def __init__(self, parent, rootdir, *args, **kwargs):
  18. super(FileBrowser, self).__init__(parent,
  19. *args,
  20. **kwargs)
  21. assert os.path.exists(rootdir),\
  22. "Invalid Root Directory!"
  23. assert os.path.isdir(rootdir),\
  24. "rootdir must be a Directory!"
  26. # Attributes
  27. self._il = wx.ImageList(16, 16)
  28. self._root = rootdir
  29. self._rnode = None
  31. rootdir = '../work'
  32. # Setup
  33. for art in (wx.ART_FOLDER, wx.ART_ERROR,
  34. wx.ART_NORMAL_FILE):
  35. bmp = wx.ArtProvider.GetBitmap(art, size=(16,16))
  36. self._il.Add(bmp)
  37. self.SetImageList(self._il)
  38. self._rnode = self.AddRoot(os.path.basename(rootdir),
  39. image=FileBrowser.FOLDER,
  40. data=self._root)
  41. self.SetItemHasChildren(self._rnode, True)
  42. # Use Windows Vista style selections
  43. # self.EnableSelectionVista(True)
  46. # Event Handlers
  47. self.Bind(wx.EVT_TREE_ITEM_EXPANDING, self.OnExpanding)
  48. self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnCollapsed)
  50. def _GetFiles(self, path):
  51. try:
  52. files = [fname for fname in os.listdir(path)
  53. if fname not in ('.', '..')]
  54. except OSError:
  55. files = None
  56. return files
  58. def OnCollapsed(self, event):
  59. item = event.GetItem()
  60. self.DeleteChildren(item)
  62. def OnCompareItems(self, item1, item2):
  63. pydata1 = self.GetPyData(item1)
  64. pydata2 = self.GetPyData(item2)
  65. if os.path.isdir(pydata1) and os.path.isdir(pydata2):
  66. return cmp(pydata1, pydata2)
  67. elif os.path.isdir(pydata1) and os.path.isfile(pydata2):
  68. return -1
  69. elif os.path.isfile(pydata1) and os.path.isdir(pydata2):
  70. return 1
  71. return cmp(pydata1, pydata2)
  73. def OnExpanding(self, event):
  74. item = event.GetItem()
  75. path = self.GetPyData(item)
  76. files = self._GetFiles(path)
  78. if files is None:
  79. self.SetItemImage(item, FileBrowser.ERROR)
  80. self.SetItemHasChildren(item, False)
  81. return
  83. for fname in files:
  84. fullpath = os.path.join(path, fname)
  85. if os.path.isdir(fullpath):
  86. self.AppendDir(item, fullpath)
  87. else:
  88. self.AppendFile(item, fullpath)
  90. self.SortChildren(item)
  91. # pt = self.GetPosition();
  92. # item, flags = self.HitTest(pt)
  93. # if item:
  94. # parent = self.GetItemParent(item)
  95. # if parent.IsOk():
  96. # self.SortChildren(parent)
  98. def AppendDir(self, item, path):
  99. """Add a directory node"""
  100. assert os.path.isdir(path), "Not a valid directory!"
  101. name = os.path.basename(path)
  102. nitem = self.AppendItem(item, name,
  103. image=FileBrowser.FOLDER,
  104. data=path)
  105. self.SetItemHasChildren(nitem, True)
  107. def AppendFile(self, item, path):
  108. """Add a file to a node"""
  109. assert os.path.isfile(path), "Not a valid file!"
  110. name = os.path.basename(path)
  111. self.AppendItem(item, name,
  112. image=FileBrowser.FILE,
  113. data=path)
  115. def GetSelectedPath(self):
  116. """Get the selected path"""
  117. sel = self.GetSelection()
  118. path = self.GetItemPyData(sel)
  119. return path
  121. def GetSelectedPaths(self):
  122. """Get a list of selected paths"""
  123. sels = self.GetSelections()
  124. paths = [self.GetItemPyData(sel)
  125. for sel in sels ]
  126. return paths
  128. #---- End Recipe Code ----#
  130. #---- Sample Application ----#
  132. class MyApp(wx.App):
  133. def OnInit(self):
  134. self.frame = MyFrame(None, title="CustomTreeCtrl", size=(300,400))
  135. self.SetTopWindow(self.frame)
  136. self.frame.Show()
  138. return True
  140. class MyFrame(wx.Frame):
  141. def __init__(self, parent, *args, **kwargs):
  142. super(MyFrame, self).__init__(parent, *args, **kwargs)
  144. # Attributes
  145. self.panel = MyPanel(self)
  147. # Layout
  148. sizer = wx.BoxSizer(wx.VERTICAL)
  149. sizer.Add(self.panel, 1, wx.EXPAND)
  150. self.SetSizer(sizer)
  151. self.CreateStatusBar()
  153. class MyPanel(wx.Panel):
  154. def __init__(self, parent):
  155. super(MyPanel, self).__init__(parent)
  157. # Attributes
  158. self.browser = FileBrowser(self, wx.GetHomeDir())
  160. # Setup
  162. # Layout
  163. sizer = wx.BoxSizer(wx.VERTICAL)
  164. sizer.Add(self.browser, 1, wx.EXPAND)
  165. self.SetSizer(sizer)
  167. # Event Handlers
  168. self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnItemActivated)
  170. def OnItemActivated(self, event):
  171. path = self.browser.GetSelectedPath()
  172. self.GetTopLevelParent().PushStatusText(path)
  174. if __name__ == "__main__":
  175. app = MyApp(False)
  176. app.MainLoop()
  177.  
Powered by spacepaste. Paste removal.