--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/util/FileManagementPanel.py Thu Jun 28 12:07:21 2012 +0200
@@ -0,0 +1,369 @@
+#!/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.
+#
+#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
+#
+#See COPYING file for copyrights details.
+#
+#This library 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.1 of the License, or (at your option) any later version.
+#
+#This library 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 library; if not, write to the Free Software
+#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+import os
+import shutil
+
+import wx
+
+from controls import EditorPanel
+from utils.BitmapLibrary import GetBitmap
+
+FILTER = _("All files (*.*)|*.*|CSV files (*.csv)|*.csv")
+
+def sort_folder(x, y):
+ if x[1] == y[1]:
+ return cmp(x[0], y[0])
+ elif x[1]:
+ 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, 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)
+ 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)
+
+ 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)
+
+ self.SetSizer(main_sizer)
+
+ self.Folder = folder
+ self.Editable = editable
+
+ self.TreeImageList = wx.ImageList(16, 16)
+ self.FOLDER_IMAGE = self.TreeImageList.Add(GetBitmap("tree_folder"))
+ self.FILE_IMAGE = self.TreeImageList.Add(GetBitmap("tree_file"))
+ self.Tree.SetImageList(self.TreeImageList)
+
+ self.Filters = {}
+ 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()]
+
+ def _GetFolderChildren(self, folderpath, recursive=True):
+ items = []
+ for filename in os.listdir(folderpath):
+ 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, True, children))
+ elif (self.CurrentFilter == "" or
+ os.path.splitext(filename)[1] == self.CurrentFilter):
+ items.append((filename, False, None))
+ if recursive:
+ items.sort(sort_folder)
+ return items
+
+ 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, isdir, children) in enumerate(self._GetFolderChildren(folderpath)):
+ new = False
+ if not item.IsOk():
+ item = self.Tree.AppendItem(root, filename)
+ if wx.Platform != '__WXMSW__':
+ item, item_cookie = self.Tree.GetNextChild(root, item_cookie)
+ new = True
+ elif self.Tree.GetItemText(item) != filename:
+ item = self.Tree.InsertItemBefore(root, idx, filename)
+ new = True
+ filepath = os.path.join(folderpath, filename)
+ if isdir:
+ if new:
+ self.Tree.SetItemImage(item, self.FOLDER_IMAGE)
+ if children > 0:
+ self.Tree.SetItemHasChildren(item)
+ elif self.Tree.IsExpanded(item):
+ self.GenerateTreeBranch(item, filepath)
+ elif new:
+ self.Tree.SetItemImage(item, self.FILE_IMAGE)
+ 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 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):
+ event.Veto()
+
+ 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
+
+class FileManagementPanel(EditorPanel):
+
+ def _init_Editor(self, parent):
+ self.Editor = wx.Panel(parent)
+
+ main_sizer = wx.BoxSizer(wx.HORIZONTAL)
+
+ left_sizer = wx.BoxSizer(wx.VERTICAL)
+ main_sizer.AddSizer(left_sizer, 1, border=5, flag=wx.GROW|wx.ALL)
+
+ managed_dir_label = wx.StaticText(self.Editor, label=self.TagName + ":")
+ left_sizer.AddWindow(managed_dir_label, border=5, flag=wx.GROW|wx.BOTTOM)
+
+ self.ManagedDir = FolderTree(self.Editor, self.Folder, FILTER)
+ left_sizer.AddWindow(self.ManagedDir, 1, flag=wx.GROW)
+
+ managed_treectrl = self.ManagedDir.GetTreeCtrl()
+ self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeItemChanged, managed_treectrl)
+ if self.EnableDragNDrop:
+ self.Bind(wx.EVT_TREE_BEGIN_DRAG, self.OnTreeBeginDrag, managed_treectrl)
+
+ button_sizer = wx.BoxSizer(wx.VERTICAL)
+ main_sizer.AddSizer(button_sizer, border=5,
+ flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL)
+
+ for idx, (name, bitmap, help) in enumerate([
+ ("DeleteButton", "remove_element", _("Remove file from left folder")),
+ ("LeftCopyButton", "LeftCopy", _("Copy file from right folder to left")),
+ ("RightCopyButton", "RightCopy", _("copy file from left folder to right"))]):
+ button = wx.lib.buttons.GenBitmapButton(self.Editor,
+ bitmap=GetBitmap(bitmap),
+ size=wx.Size(28, 28), style=wx.NO_BORDER)
+ button.SetToolTipString(help)
+ setattr(self, name, button)
+ if idx > 0:
+ flag = wx.TOP
+ else:
+ flag = 0
+ self.Bind(wx.EVT_BUTTON, getattr(self, "On" + name), button)
+ button_sizer.AddWindow(button, border=20, flag=flag)
+
+ right_sizer = wx.BoxSizer(wx.VERTICAL)
+ main_sizer.AddSizer(right_sizer, 1, border=5, flag=wx.GROW|wx.ALL)
+
+ if wx.Platform == '__WXMSW__':
+ system_dir_label = wx.StaticText(self.Editor, label=_("My Computer:"))
+ else:
+ system_dir_label = wx.StaticText(self.Editor, label=_("Home Directory:"))
+ right_sizer.AddWindow(system_dir_label, border=5, flag=wx.GROW|wx.BOTTOM)
+
+ self.SystemDir = FolderTree(self.Editor, self.HomeDirectory, FILTER, False)
+ right_sizer.AddWindow(self.SystemDir, 1, flag=wx.GROW)
+
+ system_treectrl = self.SystemDir.GetTreeCtrl()
+ self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeItemChanged, system_treectrl)
+
+ self.Editor.SetSizer(main_sizer)
+
+ def __init__(self, parent, controler, name, folder, enable_dragndrop=False):
+ self.Folder = os.path.realpath(folder)
+ self.EnableDragNDrop = enable_dragndrop
+ self.HomeDirectory = os.path.expanduser("~")
+
+ EditorPanel.__init__(self, parent, name, None, None)
+
+ self.Controler = controler
+
+ self.SetIcon(GetBitmap("FOLDER"))
+
+ def __del__(self):
+ self.Controler.OnCloseEditor(self)
+
+ def GetTitle(self):
+ return self.TagName
+
+ def RefreshView(self):
+ self.ManagedDir.RefreshTree()
+ self.SystemDir.RefreshTree()
+ self.RefreshButtonsState()
+
+ def RefreshButtonsState(self):
+ managed_filepath = self.ManagedDir.GetPath()
+ system_filepath = self.SystemDir.GetPath()
+
+ self.DeleteButton.Enable(os.path.isfile(managed_filepath))
+ self.LeftCopyButton.Enable(os.path.isfile(system_filepath))
+ self.RightCopyButton.Enable(os.path.isfile(managed_filepath))
+
+ def OnTreeItemChanged(self, event):
+ self.RefreshButtonsState()
+ event.Skip()
+
+ def OnDeleteButton(self, event):
+ filepath = self.ManagedDir.GetPath()
+ if os.path.isfile(filepath):
+ folder, filename = os.path.split(filepath)
+
+ dialog = wx.MessageDialog(self,
+ _("Do you really want to delete the file '%s'?") % filename,
+ _("Delete File"), wx.YES_NO|wx.ICON_QUESTION)
+ remove = dialog.ShowModal() == wx.ID_YES
+ dialog.Destroy()
+
+ if remove:
+ os.remove(filepath)
+ self.ManagedDir.RefreshTree()
+ event.Skip()
+
+ def CopyFile(self, src, dst):
+ if os.path.isfile(src):
+ src_folder, src_filename = os.path.split(src)
+ if os.path.isfile(dst):
+ dst_folder, dst_filename = os.path.split(dst)
+ else:
+ dst_folder = dst
+
+ dst_filepath = os.path.join(dst_folder, src_filename)
+ if os.path.isfile(dst_filepath):
+ dialog = wx.MessageDialog(self,
+ _("The file '%s' already exist.\nDo you want to replace it?") % src_filename,
+ _("Replace File"), wx.YES_NO|wx.ICON_QUESTION)
+ copy = dialog.ShowModal() == wx.ID_YES
+ dialog.Destroy()
+ else:
+ copy = True
+
+ if copy:
+ shutil.copyfile(src, dst_filepath)
+ return dst_filepath
+ return None
+
+ def OnLeftCopyButton(self, event):
+ filepath = self.CopyFile(self.SystemDir.GetPath(), self.ManagedDir.GetPath())
+ if filepath is not None:
+ self.ManagedDir.RefreshTree()
+ self.ManagedDir.SetPath(filepath)
+ event.Skip()
+
+ def OnRightCopyButton(self, event):
+ filepath = self.CopyFile(self.ManagedDir.GetPath(), self.SystemDir.GetPath())
+ if filepath is not None:
+ self.SystemDir.RefreshTree()
+ self.SystemDir.SetPath(filepath)
+ event.Skip()
+
+ def OnTreeBeginDrag(self, event):
+ filepath = self.ManagedDir.GetPath()
+ if os.path.isfile(filepath):
+ relative_filepath = filepath.replace(os.path.join(self.Folder, ""), "")
+ data = wx.TextDataObject(str(("'%s'" % relative_filepath, "Constant")))
+ dragSource = wx.DropSource(self)
+ dragSource.SetData(data)
+ dragSource.DoDragDrop()
+
\ No newline at end of file