HGKeeper

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# This file is part of Beremiz, a Integrated Development Environment for

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

# See COPYING file for copyrights details.

# This program is free software; you can redistribute it and/or

# modify it under the terms of the GNU General Public License

# as published by the Free Software Foundation; either version 2

# of the License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

import os

import wx

from util.BitmapLibrary import GetBitmap

DRIVE, FOLDER, FILE = range(3)

def sort_folder(x, y):

if x[1] == y[1]:

return cmp(x[0], y[0])

elif x[1] != FILE:

return -1

else:

return 1

def splitpath(path):

head, tail = os.path.split(path)

if head == "":

return [tail]

elif tail == "":

return splitpath(head)

return splitpath(head) + [tail]

class FolderTree(wx.Panel):

def __init__(self, parent, folder, filter=None, editable=True):

wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL)

main_sizer = wx.BoxSizer(wx.VERTICAL)

self.Tree = wx.TreeCtrl(self,

style=(wx.TR_HAS_BUTTONS |

wx.TR_SINGLE |

wx.SUNKEN_BORDER |

wx.TR_HIDE_ROOT |

wx.TR_LINES_AT_ROOT |

wx.TR_EDIT_LABELS))

if wx.Platform == '__WXMSW__':

self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnTreeItemExpanded, self.Tree)

self.Tree.Bind(wx.EVT_LEFT_DOWN, self.OnTreeLeftDown)

else:

self.Bind(wx.EVT_TREE_ITEM_EXPANDED, self.OnTreeItemExpanded, self.Tree)

self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnTreeItemCollapsed, self.Tree)

self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnTreeBeginLabelEdit, self.Tree)

self.Bind(wx.EVT_TREE_END_LABEL_EDIT, self.OnTreeEndLabelEdit, self.Tree)

main_sizer.AddWindow(self.Tree, 1, flag=wx.GROW)

if filter is not None:

self.Filter = wx.ComboBox(self, style=wx.CB_READONLY)

self.Bind(wx.EVT_COMBOBOX, self.OnFilterChanged, self.Filter)

main_sizer.AddWindow(self.Filter, flag=wx.GROW)

else:

self.Filter = None

self.SetSizer(main_sizer)

self.Folder = folder

self.Editable = editable

self.TreeImageList = wx.ImageList(16, 16)

self.TreeImageDict = {}

for item_type, bitmap in [(DRIVE, "tree_drive"),

(FOLDER, "tree_folder"),

(FILE, "tree_file")]:

self.TreeImageDict[item_type] = self.TreeImageList.Add(GetBitmap(bitmap))

self.Tree.SetImageList(self.TreeImageList)

self.Filters = {}

if self.Filter is not None:

filter_parts = filter.split("|")

for idx in xrange(0, len(filter_parts), 2):

if filter_parts[idx + 1] == "*.*":

self.Filters[filter_parts[idx]] = ""

else:

self.Filters[filter_parts[idx]] = filter_parts[idx + 1].replace("*", "")

self.Filter.Append(filter_parts[idx])

if idx == 0:

self.Filter.SetStringSelection(filter_parts[idx])

self.CurrentFilter = self.Filters[self.Filter.GetStringSelection()]

else:

self.CurrentFilter = ""

def _GetFolderChildren(self, folderpath, recursive=True):

items = []

if wx.Platform == '__WXMSW__' and folderpath == "/":

for c in xrange(ord('a'), ord('z')):

drive = os.path.join("%s:\\" % chr(c))

if os.path.exists(drive):

items.append((drive, DRIVE, self._GetFolderChildren(drive, False)))

else:

try:

files = os.listdir(folderpath)

except Exception:

return []

for filename in files:

if not filename.startswith("."):

filepath = os.path.join(folderpath, filename)

if os.path.isdir(filepath):

if recursive:

children = len(self._GetFolderChildren(filepath, False))

else:

children = 0

items.append((filename, FOLDER, children))

elif (self.CurrentFilter == "" or

os.path.splitext(filename)[1] == self.CurrentFilter):

items.append((filename, FILE, None))

if recursive:

items.sort(sort_folder)

return items

def SetFilter(self, filter):

self.CurrentFilter = filter

def GetTreeCtrl(self):

return self.Tree

def RefreshTree(self):

root = self.Tree.GetRootItem()

if not root.IsOk():

root = self.Tree.AddRoot("")

self.GenerateTreeBranch(root, self.Folder)

def GenerateTreeBranch(self, root, folderpath):

item, item_cookie = self.Tree.GetFirstChild(root)

for idx, (filename, item_type, children) in enumerate(self._GetFolderChildren(folderpath)):

if not item.IsOk():

item = self.Tree.AppendItem(root, filename, self.TreeImageDict[item_type])

if wx.Platform != '__WXMSW__':

item, item_cookie = self.Tree.GetNextChild(root, item_cookie)

elif self.Tree.GetItemText(item) != filename:

item = self.Tree.InsertItemBefore(root, idx, filename, self.TreeImageDict[item_type])

filepath = os.path.join(folderpath, filename)

if item_type != FILE:

if self.Tree.IsExpanded(item):

self.GenerateTreeBranch(item, filepath)

elif children > 0:

self.Tree.SetItemHasChildren(item)

item, item_cookie = self.Tree.GetNextChild(root, item_cookie)

to_delete = []

while item.IsOk():

to_delete.append(item)

item, item_cookie = self.Tree.GetNextChild(root, item_cookie)

for item in to_delete:

self.Tree.Delete(item)

def ExpandItem(self, item):

self.GenerateTreeBranch(item, self.GetPath(item))

self.Tree.Expand(item)

def OnTreeItemActivated(self, event):

self.ExpandItem(event.GetItem())

event.Skip()

def OnTreeLeftDown(self, event):

item, flags = self.Tree.HitTest(event.GetPosition())

if flags & wx.TREE_HITTEST_ONITEMBUTTON and not self.Tree.IsExpanded(item):

self.ExpandItem(item)

else:

event.Skip()

def OnTreeItemExpanded(self, event):

item = event.GetItem()

self.GenerateTreeBranch(item, self.GetPath(item))

event.Skip()

def OnTreeItemCollapsed(self, event):

item = event.GetItem()

self.Tree.DeleteChildren(item)

self.Tree.SetItemHasChildren(item)

event.Skip()

def OnTreeBeginLabelEdit(self, event):

item = event.GetItem()

if self.Editable and not self.Tree.ItemHasChildren(item):

event.Skip()

else:

event.Veto()

def OnTreeEndLabelEdit(self, event):

new_name = event.GetLabel()

if new_name != "":

old_filepath = self.GetPath(event.GetItem())

new_filepath = os.path.join(os.path.split(old_filepath)[0], new_name)

if new_filepath != old_filepath:

if not os.path.exists(new_filepath):

os.rename(old_filepath, new_filepath)

event.Skip()

else:

message = wx.MessageDialog(self,

_("File '%s' already exists!") % new_name,

_("Error"), wx.OK | wx.ICON_ERROR)

message.ShowModal()

message.Destroy()

event.Veto()

else:

event.Skip()

def OnFilterChanged(self, event):

self.CurrentFilter = self.Filters[self.Filter.GetStringSelection()]

self.RefreshTree()

event.Skip()

def _SelectItem(self, root, parts):

if len(parts) == 0:

self.Tree.SelectItem(root)

else:

item, item_cookie = self.Tree.GetFirstChild(root)

while item.IsOk():

if self.Tree.GetItemText(item) == parts[0]:

if self.Tree.ItemHasChildren(item) and \

not self.Tree.IsExpanded(item):

self.Tree.Expand(item)

wx.CallAfter(self._SelectItem, item, parts[1:])

else:

self._SelectItem(item, parts[1:])

return

item, item_cookie = self.Tree.GetNextChild(root, item_cookie)

def SetPath(self, path):

if path.startswith(self.Folder):

root = self.Tree.GetRootItem()

if root.IsOk():

relative_path = path.replace(os.path.join(self.Folder, ""), "")

self._SelectItem(root, splitpath(relative_path))

def GetPath(self, item=None):

if item is None:

item = self.Tree.GetSelection()

if item.IsOk():

filepath = self.Tree.GetItemText(item)

parent = self.Tree.GetItemParent(item)

while parent.IsOk() and parent != self.Tree.GetRootItem():

filepath = os.path.join(self.Tree.GetItemText(parent), filepath)

parent = self.Tree.GetItemParent(parent)

return os.path.join(self.Folder, filepath)

return self.Folder

beremiz