#!/usr/bin/env python

from wxPython import wx
import sys, os
from stat import *

GlobalObjList = []

class Obj:
	def __init__(self, obj):
		self.obj = obj
		# Uncomment next line to eliminate crash.
		# GlobalObjList.append(self)

	def Name(self):
		head, tail = os.path.split(self.obj)
		if tail:
			return tail
		else:
			return head
		
	def HasChildren(self):
		return os.path.isdir(self.obj)
	
	def Children(self):
		objList = os.listdir(self.obj)
		objList.sort()
		objList = map(lambda obj,parent=self.obj: os.path.join(parent,obj),
					  objList)
		objectList = map(Obj, objList)
		return objectList
	
	def __str__(self):
		return self.obj
	
	def __repr__(self):
		return self.obj

	def __del__(self):
		print 'del', self.obj
		

#----------------------------------------------------------------------

class pyTree(wx.wxTreeCtrl):

	def __init__(self, parent, id, obj):
		wx.wxTreeCtrl.__init__(self, parent, id)
		self.root = self.AddRoot(obj.Name(), -1, -1, wx.wxTreeItemData(''))
		self.SetPyData(self.root, obj)
		if obj.HasChildren():
			self.SetItemHasChildren(self.root, wx.TRUE)
		wx.EVT_TREE_ITEM_EXPANDING(self, self.GetId(), self.OnItemExpanding)
		wx.EVT_TREE_ITEM_COLLAPSED(self, self.GetId(), self.OnItemCollapsed)
		wx.EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged)
		self.output = None

	def SetOutput(self, output):
		self.output = output

	def OnItemExpanding(self,event):
		item = event.GetItem()
		obj = self.GetPyData(item)
		children = obj.Children()
		for child in children:
			new_item = self.AppendItem(item, child.Name(), -1, -1,
									   wx.wxTreeItemData(''))
			self.SetPyData(new_item, child)
			if child.HasChildren():
				self.SetItemHasChildren(new_item, wx.TRUE)

	def OnItemCollapsed(self, event):
		item = event.GetItem()
		self.DeleteChildren(item)

	def OnSelChanged(self, event):
		if not self.output:
			return
		obj = self.GetPyData( event.GetItem() )
		apply(self.output, (`obj`,))



#----------------------------------------------------------------------
if __name__ == '__main__':

	class MyFrame(wx.wxFrame):

		def __init__(self):
			wx.wxFrame.__init__(self, wx.NULL, -1, 'PyTreeItemData Test',
			 					wx.wxDefaultPosition, wx.wxSize(600,500))
			split = wx.wxSplitterWindow(self, -1)
			
			if sys.platform == 'win32':
				tree = pyTree(split, -1, Obj('C:\\'))
			else:
				tree = pyTree(split, -1, Obj('/'))
				
			text = wx.wxTextCtrl(split, -1, '', wx.wxDefaultPosition,
								 wx.wxDefaultSize, wx.wxTE_MULTILINE)
			split.SplitVertically(tree, text, 200)
			tree.SetOutput(text.SetValue)
			tree.SelectItem(tree.root)

	class MyApp(wx.wxApp):
		 
		def OnInit(self):
			frame = MyFrame()
			frame.Show(wx.TRUE)
			self.SetTopWindow(frame)
			return wx.TRUE

	app = MyApp(0)
	app.MainLoop()