def Build( self, projDirPath ):
# Clear all widgets from the sizer
self.bs1.Clear( True )
if projDirPath is not None and os.path.isdir( projDirPath ):
# Build tree control and add it to the sizer
self.dtc = DirTreeCtrl( self, -1, style=
wx.NO_BORDER |
wx.TR_DEFAULT_STYLE |
wx.TR_EDIT_LABELS )
self.dtc.SetRootDir( projDirPath )
self.dtc.Expand( self.dtc.GetRootItem() )
self.bs1.Add( self.dtc, 1, wx.EXPAND )
# Bind tree control events
self.dtc.Bind( wx.EVT_KEY_UP, p3d.wx.OnKeyUp )
self.dtc.Bind( wx.EVT_KEY_DOWN, p3d.wx.OnKeyDown )
self.dtc.Bind( wx.EVT_LEFT_UP, p3d.wx.OnLeftUp )
self.dtc.Bind( wx.EVT_RIGHT_DOWN, self.OnRightDown )
self.dtc.Bind( wx.EVT_RIGHT_UP, self.OnRightUp )
self.dtc.Bind( wx.EVT_MIDDLE_DOWN, self.OnMiddleDown )
self.dtc.Bind( wx.EVT_LEFT_DCLICK, self.OnLeftDClick )
self.dtc.Bind( wx.EVT_TREE_END_LABEL_EDIT, self.OnTreeEndLabelEdit )
else:
# Build and display "project not set" warning
tc = wx.StaticText( self, -1, 'Project directory not set', style=wx.ALIGN_CENTER )
font = tc.GetFont()
font.SetWeight( wx.FONTWEIGHT_BOLD )
tc.SetFont( font )
self.bs1.AddSpacer( 10 )
self.bs1.Add( tc, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 2 )
self.bs1.Layout()
def Build( self, projDirPath ):
# Clear all widgets from the sizer
self.bs1.Clear( True )
if projDirPath is not None and os.path.isdir( projDirPath ):
# Build tree control and add it to the sizer
self.dtc = DirTreeCtrl( self, -1, style=
wx.NO_BORDER |
wx.TR_DEFAULT_STYLE |
wx.TR_EDIT_LABELS )
self.dtc.SetRootDir( projDirPath )
self.dtc.Expand( self.dtc.GetRootItem() )
self.bs1.Add( self.dtc, 1, wx.EXPAND )
# Bind tree control events
self.dtc.Bind( wx.EVT_KEY_UP, p3d.wx.OnKeyUp )
self.dtc.Bind( wx.EVT_KEY_DOWN, p3d.wx.OnKeyDown )
self.dtc.Bind( wx.EVT_LEFT_UP, p3d.wx.OnLeftUp )
self.dtc.Bind( wx.EVT_RIGHT_DOWN, self.OnRightDown )
self.dtc.Bind( wx.EVT_RIGHT_UP, self.OnRightUp )
self.dtc.Bind( wx.EVT_MIDDLE_DOWN, self.OnMiddleDown )
self.dtc.Bind( wx.EVT_LEFT_DCLICK, self.OnLeftDClick )
self.dtc.Bind( wx.EVT_TREE_END_LABEL_EDIT, self.OnTreeEndLabelEdit )
else:
# Build and display "project not set" warning
tc = wx.StaticText( self, -1, 'Project directory not set', style=wx.ALIGN_CENTER )
font = tc.GetFont()
font.SetWeight( wx.FONTWEIGHT_BOLD )
tc.SetFont( font )
self.bs1.AddSpacer( 10 )
self.bs1.Add( tc, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 2 )
self.bs1.Layout()
class ResourcesPanel(wx.Panel):
def __init__(self, *args, **kwargs):
wx.Panel.__init__(self, *args, **kwargs)
self.app = wx.GetApp()
# Bind project file events
pub.subscribe(self.OnUpdate, 'projectFilesAdded')
pub.subscribe(self.OnUpdate, 'projectFilesRemoved')
# Build sizers
self.bs1 = wx.BoxSizer(wx.VERTICAL)
self.SetSizer(self.bs1)
def Build(self, projDirPath):
# Clear all widgets from the sizer
self.bs1.Clear(True)
if projDirPath is not None and os.path.isdir(projDirPath):
# Build tree control and add it to the sizer
self.dtc = DirTreeCtrl(self,
-1,
style=wx.NO_BORDER | wx.TR_DEFAULT_STYLE
| wx.TR_EDIT_LABELS)
self.dtc.SetRootDir(projDirPath)
self.dtc.Expand(self.dtc.GetRootItem())
self.bs1.Add(self.dtc, 1, wx.EXPAND)
# Bind tree control events
self.dtc.Bind(wx.EVT_KEY_UP, p3d.wx.OnKeyUp)
self.dtc.Bind(wx.EVT_KEY_DOWN, p3d.wx.OnKeyDown)
self.dtc.Bind(wx.EVT_LEFT_UP, p3d.wx.OnLeftUp)
self.dtc.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
self.dtc.Bind(wx.EVT_RIGHT_UP, self.OnRightUp)
self.dtc.Bind(wx.EVT_MIDDLE_DOWN, self.OnMiddleDown)
self.dtc.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
self.dtc.Bind(wx.EVT_TREE_END_LABEL_EDIT, self.OnTreeEndLabelEdit)
else:
# Build and display "project not set" warning
tc = wx.StaticText(self,
-1,
'Project directory not set',
style=wx.ALIGN_CENTER)
font = tc.GetFont()
font.SetWeight(wx.FONTWEIGHT_BOLD)
tc.SetFont(font)
self.bs1.AddSpacer(10)
self.bs1.Add(tc, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 2)
self.bs1.Layout()
def OnRightDown(self, evt):
"""
This method does nothing. Oddly enough, *not* binding RightDown seems
to affect RightUp's behaviour, and we only trap half the mouse up
events.
"""
pass
def OnRightUp(self, evt):
# Get the item under the mouse - bail if the item is not ok
itemId = wxUtils.GetClickedItem(self.dtc, evt)
if itemId is None or not itemId.IsOk():
return
menu = wx.Menu()
mItem = wx.MenuItem(menu, wx.NewId(), 'Open in Explorer')
menu.AppendItem(mItem)
wxUtils.IdBind(menu, wx.EVT_MENU, mItem.GetId(), self.OnOpenFile,
itemId)
self.PopupMenu(menu)
menu.Destroy()
def OnOpenFile(self, evt, itemId):
systems = {
'nt': os.startfile,
'posix':
lambda foldername: os.system('xdg-open "%s"' % foldername),
'os2': lambda foldername: os.system('open "%s"' % foldername)
}
filePath = self.dtc.GetItemPath(itemId)
dirPath = os.path.split(filePath)[0]
systems.get(os.name, os.startfile)(dirPath)
def OnMiddleDown(self, evt):
# Get the item under the mouse - bail if the item is not ok
itemId = wxUtils.GetClickedItem(self.dtc, evt)
if itemId is None or not itemId.IsOk():
return
# Select it and start drag and drop operations.
self.dtc.SelectItem(itemId)
self.app.dDropMgr.Start(self, [self.dtc.GetItemPath(itemId)],
self.dtc.GetItemPath(itemId))
def OnLeftDClick(self, evt):
# Load items
itemId = wxUtils.GetClickedItem(self.dtc, evt)
filePath = self.dtc.GetItemPath(itemId)
ext = os.path.splitext(os.path.basename(
self.dtc.GetItemText(itemId)))[1]
if ext == '.xml':
self.app.frame.OnFileOpen(None, filePath)
elif ext == '.py':
os.startfile(filePath)
def OnTreeEndLabelEdit(self, evt):
"""Change the name of the asset in the system."""
# Bail if no valid label is entered
if not evt.GetLabel():
evt.Veto()
return
# Construct new file path and rename
oldPath = self.dtc.GetItemPath(evt.GetItem())
head, tail = os.path.split(oldPath)
newPath = os.path.join(head, evt.GetLabel())
os.rename(oldPath, newPath)
def OnUpdate(self, arg):
"""Rebuild the directory tree."""
self.dtc.Freeze()
def GetExpandedDict():
# Return a dictionary mapping each node path to its tree item.
expDict = {}
for item in self.dtc.GetAllItems():
dir = self.dtc.GetPyData(item)
if dir is not None:
expDict[dir.directory] = (item, self.dtc.IsExpanded(item))
return expDict
# Get map of directory paths to items and expanded states before
# updating
oldItems = GetExpandedDict()
# Rebuild the tree control
rootItem = self.dtc.GetRootItem()
rootDirPath = self.dtc.GetPyData(rootItem).directory
self.dtc._loadDir(rootItem, rootDirPath)
# Get map of directory paths to items and expanded states after
# updating
newItems = GetExpandedDict()
# Set the expanded states back
for dirPath, grp in oldItems.items():
oldItem, oldExp = grp
if dirPath in newItems and oldExp:
newItem, newExp = newItems[dirPath]
self.dtc.Expand(newItem)
self.dtc.Thaw()
class ResourcesPanel( wx.Panel ):
def __init__( self, *args, **kwargs ):
wx.Panel.__init__( self, *args, **kwargs )
# Bind project file events
pub.subscribe( self.OnUpdate, 'projectFilesAdded' )
pub.subscribe( self.OnUpdate, 'projectFilesRemoved' )
# Build sizers
self.bs1 = wx.BoxSizer( wx.VERTICAL )
self.SetSizer( self.bs1 )
def Build( self, projDirPath ):
# Clear all widgets from the sizer
self.bs1.Clear( True )
if projDirPath is not None and os.path.isdir( projDirPath ):
# Build tree control and add it to the sizer
self.dtc = DirTreeCtrl( self, -1, style=
wx.NO_BORDER |
wx.TR_DEFAULT_STYLE |
wx.TR_EDIT_LABELS )
self.dtc.SetRootDir( projDirPath )
self.dtc.Expand( self.dtc.GetRootItem() )
self.bs1.Add( self.dtc, 1, wx.EXPAND )
# Bind tree control events
self.dtc.Bind( wx.EVT_KEY_UP, p3d.wx.OnKeyUp )
self.dtc.Bind( wx.EVT_KEY_DOWN, p3d.wx.OnKeyDown )
self.dtc.Bind( wx.EVT_LEFT_UP, p3d.wx.OnLeftUp )
self.dtc.Bind( wx.EVT_RIGHT_DOWN, self.OnRightDown )
self.dtc.Bind( wx.EVT_RIGHT_UP, self.OnRightUp )
self.dtc.Bind( wx.EVT_MIDDLE_DOWN, self.OnMiddleDown )
self.dtc.Bind( wx.EVT_LEFT_DCLICK, self.OnLeftDClick )
self.dtc.Bind( wx.EVT_TREE_END_LABEL_EDIT, self.OnTreeEndLabelEdit )
else:
# Build and display "project not set" warning
tc = wx.StaticText( self, -1, 'Project directory not set', style=wx.ALIGN_CENTER )
font = tc.GetFont()
font.SetWeight( wx.FONTWEIGHT_BOLD )
tc.SetFont( font )
self.bs1.AddSpacer( 10 )
self.bs1.Add( tc, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 2 )
self.bs1.Layout()
def OnRightDown( self, evt ):
"""
This method does nothing. Oddly enough, *not* binding RightDown seems
to affect RightUp's behaviour, and we only trap half the mouse up
events.
"""
pass
def OnRightUp( self, evt ):
# Get the item under the mouse - bail if the item is not ok
itemId = wxUtils.GetClickedItem( self.dtc, evt )
if itemId is None or not itemId.IsOk():
return
menu = wx.Menu()
mItem = wx.MenuItem( menu, wx.NewId(), 'Open in Explorer' )
menu.AppendItem( mItem )
wxUtils.IdBind( menu, wx.EVT_MENU, mItem.GetId(), self.OnOpenFile, itemId )
self.PopupMenu( menu )
menu.Destroy()
def OnOpenFile( self, evt, itemId ):
systems = {
'nt': os.startfile,
'posix': lambda foldername: os.system( 'xdg-open "%s"' % foldername ),
'os2': lambda foldername: os.system( 'open "%s"' % foldername )
}
filePath = self.dtc.GetItemPath( itemId )
dirPath = os.path.split( filePath )[0]
systems.get( os.name, os.startfile )( dirPath )
def OnMiddleDown( self, evt ):
# Get the item under the mouse - bail if the item is not ok
itemId = wxUtils.GetClickedItem( self.dtc, evt )
if itemId is None or not itemId.IsOk():
return
# Select it
self.dtc.SelectItem( itemId )
# Create a custom data object that contains the full path of the item
do = wx.CustomDataObject( 'filePath' )
do.SetData( str( self.dtc.GetItemPath( itemId ) ) )
# Create the drop source and begin the drag and drop operation
ds = wx.DropSource( self )
ds.SetData( do )
ds.DoDragDrop( wx.Drag_AllowMove )
def OnLeftDClick( self, evt ):
# Load items
itemId = wxUtils.GetClickedItem( self.dtc, evt )
filePath = self.dtc.GetItemPath( itemId )
ext = os.path.splitext( os.path.basename( self.dtc.GetItemText( itemId ) ) )[1]
if ext == '.xml':
wx.GetApp().frame.OnFileOpen( None, filePath )
elif ext == '.py':
os.startfile( filePath )
def OnTreeEndLabelEdit( self, evt ):
"""Change the name of the asset in the system."""
# Bail if no valid label is entered
if not evt.GetLabel():
evt.Veto()
return
# Construct new file path and rename
oldPath = self.dtc.GetItemPath( evt.GetItem() )
head, tail = os.path.split( oldPath )
newPath = os.path.join( head, evt.GetLabel() )
os.rename( oldPath, newPath )
def OnUpdate( self, arg ):
"""Rebuild the directory tree."""
self.dtc.Freeze()
def GetExpandedDict():
# Return a dictionary mapping each node path to its tree item.
expDict = {}
for item in self.dtc.GetAllItems():
dir = self.dtc.GetPyData( item )
if dir is not None:
expDict[dir.directory] = (item, self.dtc.IsExpanded( item ))
return expDict
# Get map of directory paths to items and expanded states before
# updating
oldItems = GetExpandedDict()
# Rebuild the tree control
rootItem = self.dtc.GetRootItem()
rootDirPath = self.dtc.GetPyData( rootItem ).directory
self.dtc._loadDir( rootItem, rootDirPath )
# Get map of directory paths to items and expanded states after
# updating
newItems = GetExpandedDict()
# Set the expanded states back
for dirPath, grp in oldItems.items():
oldItem, oldExp = grp
if dirPath in newItems and oldExp:
newItem, newExp = newItems[dirPath]
self.dtc.Expand( newItem )
self.dtc.Thaw()
