diff -r cdc6393705ce -r 6f0e10085df9 util/FileManagementPanel.py --- /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