laurent@782: #!/usr/bin/env python
laurent@782: # -*- coding: utf-8 -*-
laurent@782: 
laurent@782: #This file is part of Beremiz, a Integrated Development Environment for
laurent@782: #programming IEC 61131-3 automates supporting plcopen standard and CanFestival. 
laurent@782: #
laurent@782: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
laurent@782: #
laurent@782: #See COPYING file for copyrights details.
laurent@782: #
laurent@782: #This library is free software; you can redistribute it and/or
laurent@782: #modify it under the terms of the GNU General Public
laurent@782: #License as published by the Free Software Foundation; either
laurent@782: #version 2.1 of the License, or (at your option) any later version.
laurent@782: #
laurent@782: #This library is distributed in the hope that it will be useful,
laurent@782: #but WITHOUT ANY WARRANTY; without even the implied warranty of
laurent@782: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
laurent@782: #General Public License for more details.
laurent@782: #
laurent@782: #You should have received a copy of the GNU General Public
laurent@782: #License along with this library; if not, write to the Free Software
laurent@782: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
laurent@782: 
laurent@782: import os
laurent@782: import shutil
laurent@782: 
laurent@782: import wx
laurent@782: 
laurent@782: from controls import EditorPanel
laurent@782: from utils.BitmapLibrary import GetBitmap
laurent@782: 
laurent@782: FILTER = _("All files (*.*)|*.*|CSV files (*.csv)|*.csv")
laurent@782: 
laurent@782: def sort_folder(x, y):
laurent@782:     if x[1] == y[1]:
laurent@782:         return cmp(x[0], y[0])
laurent@782:     elif x[1]:
laurent@782:         return -1
laurent@782:     else:
laurent@782:         return 1
laurent@782: 
laurent@782: def splitpath(path):
laurent@782:     head, tail = os.path.split(path)
laurent@782:     if head == "":
laurent@782:         return [tail]
laurent@782:     elif tail == "":
laurent@782:         return splitpath(head)
laurent@782:     return splitpath(head) + [tail]
laurent@782:     
laurent@782: class FolderTree(wx.Panel):
laurent@782:     
laurent@782:     def __init__(self, parent, folder, filter, editable=True):
laurent@782:         wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL)
laurent@782:         
laurent@782:         main_sizer = wx.BoxSizer(wx.VERTICAL)
laurent@782:         
laurent@782:         self.Tree = wx.TreeCtrl(self, 
laurent@782:               style=wx.TR_HAS_BUTTONS|
laurent@782:                     wx.TR_SINGLE|
laurent@782:                     wx.SUNKEN_BORDER|
laurent@782:                     wx.TR_HIDE_ROOT|
laurent@782:                     wx.TR_LINES_AT_ROOT|
laurent@782:                     wx.TR_EDIT_LABELS)
laurent@782:         self.Bind(wx.EVT_TREE_ITEM_EXPANDED, self.OnTreeItemExpanded, self.Tree)
laurent@782:         self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnTreeItemCollapsed, self.Tree)
laurent@782:         self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnTreeBeginLabelEdit, self.Tree)
laurent@782:         self.Bind(wx.EVT_TREE_END_LABEL_EDIT, self.OnTreeEndLabelEdit, self.Tree)
laurent@782:         main_sizer.AddWindow(self.Tree, 1, flag=wx.GROW)
laurent@782:         
laurent@782:         self.Filter = wx.ComboBox(self, style=wx.CB_READONLY)
laurent@782:         self.Bind(wx.EVT_COMBOBOX, self.OnFilterChanged, self.Filter)
laurent@782:         main_sizer.AddWindow(self.Filter, flag=wx.GROW)
laurent@782:         
laurent@782:         self.SetSizer(main_sizer)
laurent@782:         
laurent@782:         self.Folder = folder
laurent@782:         self.Editable = editable
laurent@782:         
laurent@782:         self.TreeImageList = wx.ImageList(16, 16)
laurent@782:         self.FOLDER_IMAGE = self.TreeImageList.Add(GetBitmap("tree_folder"))
laurent@782:         self.FILE_IMAGE = self.TreeImageList.Add(GetBitmap("tree_file"))
laurent@782:         self.Tree.SetImageList(self.TreeImageList)
laurent@782:         
laurent@782:         self.Filters = {}
laurent@782:         filter_parts = filter.split("|")
laurent@782:         for idx in xrange(0, len(filter_parts), 2):
laurent@782:             if filter_parts[idx + 1] == "*.*":
laurent@782:                 self.Filters[filter_parts[idx]] = ""
laurent@782:             else:
laurent@782:                 self.Filters[filter_parts[idx]] = filter_parts[idx + 1].replace("*", "")
laurent@782:             self.Filter.Append(filter_parts[idx])
laurent@782:             if idx == 0:
laurent@782:                 self.Filter.SetStringSelection(filter_parts[idx])
laurent@782:             
laurent@782:         self.CurrentFilter = self.Filters[self.Filter.GetStringSelection()]
laurent@782:     
laurent@782:     def _GetFolderChildren(self, folderpath, recursive=True):
laurent@782:         items = []
laurent@782:         for filename in os.listdir(folderpath):
laurent@782:             if not filename.startswith("."):
laurent@782:                 filepath = os.path.join(folderpath, filename)
laurent@782:                 if os.path.isdir(filepath):
laurent@782:                     if recursive:
laurent@782:                         children = len(self._GetFolderChildren(filepath, False))
laurent@782:                     else:
laurent@782:                         children = 0
laurent@782:                     items.append((filename, True, children))
laurent@782:                 elif (self.CurrentFilter == "" or 
laurent@782:                       os.path.splitext(filename)[1] == self.CurrentFilter):
laurent@782:                     items.append((filename, False, None))
laurent@782:         if recursive:
laurent@782:             items.sort(sort_folder)
laurent@782:         return items
laurent@782:     
laurent@782:     def GetTreeCtrl(self):
laurent@782:         return self.Tree
laurent@782:     
laurent@782:     def RefreshTree(self):
laurent@782:         root = self.Tree.GetRootItem()
laurent@782:         if not root.IsOk():
laurent@782:             root = self.Tree.AddRoot("")
laurent@782:         self.GenerateTreeBranch(root, self.Folder)
laurent@782:         
laurent@782:     def GenerateTreeBranch(self, root, folderpath):
laurent@782:         item, item_cookie = self.Tree.GetFirstChild(root)
laurent@782:         for idx, (filename, isdir, children) in enumerate(self._GetFolderChildren(folderpath)):
laurent@782:             new = False
laurent@782:             if not item.IsOk():
laurent@782:                 item = self.Tree.AppendItem(root, filename)
laurent@782:                 if wx.Platform != '__WXMSW__':
laurent@782:                     item, item_cookie = self.Tree.GetNextChild(root, item_cookie)
laurent@782:                 new = True
laurent@782:             elif self.Tree.GetItemText(item) != filename:
laurent@782:                 item = self.Tree.InsertItemBefore(root, idx, filename)
laurent@782:                 new = True
laurent@782:             filepath = os.path.join(folderpath, filename)
laurent@782:             if isdir:
laurent@782:                 if new:
laurent@782:                     self.Tree.SetItemImage(item, self.FOLDER_IMAGE)
laurent@782:                     if children > 0:
laurent@782:                         self.Tree.SetItemHasChildren(item)
laurent@782:                 elif self.Tree.IsExpanded(item):
laurent@782:                     self.GenerateTreeBranch(item, filepath)
laurent@782:             elif new:
laurent@782:                 self.Tree.SetItemImage(item, self.FILE_IMAGE)
laurent@782:             item, item_cookie = self.Tree.GetNextChild(root, item_cookie)
laurent@782:         to_delete = []
laurent@782:         while item.IsOk():
laurent@782:             to_delete.append(item)
laurent@782:             item, item_cookie = self.Tree.GetNextChild(root, item_cookie)
laurent@782:         for item in to_delete:
laurent@782:             self.Tree.Delete(item)
laurent@782: 
laurent@782:     def OnTreeItemExpanded(self, event):
laurent@782:         item = event.GetItem()
laurent@782:         self.GenerateTreeBranch(item, self.GetPath(item))
laurent@782:         event.Skip()
laurent@782: 
laurent@782:     def OnTreeItemCollapsed(self, event):
laurent@782:         item = event.GetItem()
laurent@782:         self.Tree.DeleteChildren(item)
laurent@782:         self.Tree.SetItemHasChildren(item)
laurent@782:         event.Skip()
laurent@782: 
laurent@782:     def OnTreeBeginLabelEdit(self, event):
laurent@782:         item = event.GetItem()
laurent@782:         if self.Editable and not self.Tree.ItemHasChildren(item):
laurent@782:             event.Skip()
laurent@782:         else:
laurent@782:             event.Veto()
laurent@782:     
laurent@782:     def OnTreeEndLabelEdit(self, event):
laurent@782:         event.Veto()
laurent@782:     
laurent@782:     def OnFilterChanged(self, event):
laurent@782:         self.CurrentFilter = self.Filters[self.Filter.GetStringSelection()]
laurent@782:         self.RefreshTree()
laurent@782:         event.Skip()
laurent@782:     
laurent@782:     def _SelectItem(self, root, parts):
laurent@782:         if len(parts) == 0:
laurent@782:             self.Tree.SelectItem(root)
laurent@782:         else:
laurent@782:             item, item_cookie = self.Tree.GetFirstChild(root)
laurent@782:             while item.IsOk():
laurent@782:                 if self.Tree.GetItemText(item) == parts[0]:
laurent@782:                     if (self.Tree.ItemHasChildren(item) and 
laurent@782:                         not self.Tree.IsExpanded(item)):
laurent@782:                         self.Tree.Expand(item)
laurent@782:                         wx.CallAfter(self._SelectItem, item, parts[1:])
laurent@782:                     else:
laurent@782:                         self._SelectItem(item, parts[1:])
laurent@782:                     return
laurent@782:                 item, item_cookie = self.Tree.GetNextChild(root, item_cookie)
laurent@782:     
laurent@782:     def SetPath(self, path):
laurent@782:         if path.startswith(self.Folder):
laurent@782:             root = self.Tree.GetRootItem()
laurent@782:             if root.IsOk():
laurent@782:                 relative_path = path.replace(os.path.join(self.Folder, ""), "")
laurent@782:                 self._SelectItem(root, splitpath(relative_path))
laurent@782:     
laurent@782:     def GetPath(self, item=None):
laurent@782:         if item is None:
laurent@782:             item = self.Tree.GetSelection()
laurent@782:         if item.IsOk():
laurent@782:             filepath = self.Tree.GetItemText(item)
laurent@782:             parent = self.Tree.GetItemParent(item)
laurent@782:             while parent.IsOk() and parent != self.Tree.GetRootItem():
laurent@782:                 filepath = os.path.join(self.Tree.GetItemText(parent), filepath)
laurent@782:                 parent = self.Tree.GetItemParent(parent)
laurent@782:             return os.path.join(self.Folder, filepath)
laurent@782:         return self.Folder
laurent@782: 
laurent@782: class FileManagementPanel(EditorPanel):
laurent@782:     
laurent@782:     def _init_Editor(self, parent):
laurent@782:         self.Editor = wx.Panel(parent)
laurent@782:         
laurent@782:         main_sizer = wx.BoxSizer(wx.HORIZONTAL)
laurent@782:         
laurent@782:         left_sizer = wx.BoxSizer(wx.VERTICAL)
laurent@782:         main_sizer.AddSizer(left_sizer, 1, border=5, flag=wx.GROW|wx.ALL)
laurent@782:         
laurent@782:         managed_dir_label = wx.StaticText(self.Editor, label=self.TagName + ":")
laurent@782:         left_sizer.AddWindow(managed_dir_label, border=5, flag=wx.GROW|wx.BOTTOM)
laurent@782:         
laurent@782:         self.ManagedDir = FolderTree(self.Editor, self.Folder, FILTER)
laurent@782:         left_sizer.AddWindow(self.ManagedDir, 1, flag=wx.GROW)
laurent@782:         
laurent@782:         managed_treectrl = self.ManagedDir.GetTreeCtrl()
laurent@782:         self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeItemChanged, managed_treectrl)
laurent@782:         if self.EnableDragNDrop:
laurent@782:             self.Bind(wx.EVT_TREE_BEGIN_DRAG, self.OnTreeBeginDrag, managed_treectrl)
laurent@782:         
laurent@782:         button_sizer = wx.BoxSizer(wx.VERTICAL)
laurent@782:         main_sizer.AddSizer(button_sizer, border=5, 
laurent@782:               flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL)
laurent@782:         
laurent@782:         for idx, (name, bitmap, help) in enumerate([
laurent@782:                 ("DeleteButton", "remove_element", _("Remove file from left folder")),
laurent@782:                 ("LeftCopyButton", "LeftCopy", _("Copy file from right folder to left")),
laurent@782:                 ("RightCopyButton", "RightCopy", _("copy file from left folder to right"))]):
laurent@782:             button = wx.lib.buttons.GenBitmapButton(self.Editor, 
laurent@782:                   bitmap=GetBitmap(bitmap), 
laurent@782:                   size=wx.Size(28, 28), style=wx.NO_BORDER)
laurent@782:             button.SetToolTipString(help)
laurent@782:             setattr(self, name, button)
laurent@782:             if idx > 0:
laurent@782:                 flag = wx.TOP
laurent@782:             else:
laurent@782:                 flag = 0
laurent@782:             self.Bind(wx.EVT_BUTTON, getattr(self, "On" + name), button)
laurent@782:             button_sizer.AddWindow(button, border=20, flag=flag)
laurent@782:         
laurent@782:         right_sizer = wx.BoxSizer(wx.VERTICAL)
laurent@782:         main_sizer.AddSizer(right_sizer, 1, border=5, flag=wx.GROW|wx.ALL)
laurent@782:         
laurent@782:         if wx.Platform == '__WXMSW__':
laurent@782:             system_dir_label = wx.StaticText(self.Editor, label=_("My Computer:"))
laurent@782:         else:
laurent@782:             system_dir_label = wx.StaticText(self.Editor, label=_("Home Directory:"))
laurent@782:         right_sizer.AddWindow(system_dir_label, border=5, flag=wx.GROW|wx.BOTTOM)
laurent@782:         
laurent@782:         self.SystemDir = FolderTree(self.Editor, self.HomeDirectory, FILTER, False)
laurent@782:         right_sizer.AddWindow(self.SystemDir, 1, flag=wx.GROW)
laurent@782:         
laurent@782:         system_treectrl = self.SystemDir.GetTreeCtrl()
laurent@782:         self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeItemChanged, system_treectrl)
laurent@782:         
laurent@782:         self.Editor.SetSizer(main_sizer)
laurent@782:         
laurent@782:     def __init__(self, parent, controler, name, folder, enable_dragndrop=False):
laurent@782:         self.Folder = os.path.realpath(folder)
laurent@782:         self.EnableDragNDrop = enable_dragndrop
laurent@782:         self.HomeDirectory = os.path.expanduser("~")
laurent@782:         
laurent@782:         EditorPanel.__init__(self, parent, name, None, None)
laurent@782:         
laurent@782:         self.Controler = controler
laurent@782:         
laurent@782:         self.SetIcon(GetBitmap("FOLDER"))
laurent@782:     
laurent@782:     def __del__(self):
laurent@782:         self.Controler.OnCloseEditor(self)
laurent@782:     
laurent@782:     def GetTitle(self):
laurent@782:         return self.TagName
laurent@782:     
laurent@782:     def RefreshView(self):
laurent@782:         self.ManagedDir.RefreshTree()
laurent@782:         self.SystemDir.RefreshTree()
laurent@782:         self.RefreshButtonsState()
laurent@782:     
laurent@782:     def RefreshButtonsState(self):
laurent@782:         managed_filepath = self.ManagedDir.GetPath()
laurent@782:         system_filepath = self.SystemDir.GetPath()
laurent@782:         
laurent@782:         self.DeleteButton.Enable(os.path.isfile(managed_filepath))
laurent@782:         self.LeftCopyButton.Enable(os.path.isfile(system_filepath))
laurent@782:         self.RightCopyButton.Enable(os.path.isfile(managed_filepath))
laurent@782:     
laurent@782:     def OnTreeItemChanged(self, event):
laurent@782:         self.RefreshButtonsState()
laurent@782:         event.Skip()
laurent@782:         
laurent@782:     def OnDeleteButton(self, event):
laurent@782:         filepath = self.ManagedDir.GetPath()
laurent@782:         if os.path.isfile(filepath):
laurent@782:             folder, filename = os.path.split(filepath)
laurent@782:             
laurent@782:             dialog = wx.MessageDialog(self, 
laurent@782:                   _("Do you really want to delete the file '%s'?") % filename, 
laurent@782:                   _("Delete File"), wx.YES_NO|wx.ICON_QUESTION)
laurent@782:             remove = dialog.ShowModal() == wx.ID_YES
laurent@782:             dialog.Destroy()
laurent@782:             
laurent@782:             if remove:
laurent@782:                 os.remove(filepath)
laurent@782:                 self.ManagedDir.RefreshTree()
laurent@782:         event.Skip()
laurent@782: 
laurent@782:     def CopyFile(self, src, dst):
laurent@782:         if os.path.isfile(src):
laurent@782:             src_folder, src_filename = os.path.split(src)
laurent@782:             if os.path.isfile(dst):
laurent@782:                 dst_folder, dst_filename = os.path.split(dst)
laurent@782:             else:
laurent@782:                 dst_folder = dst
laurent@782:             
laurent@782:             dst_filepath = os.path.join(dst_folder, src_filename)
laurent@782:             if os.path.isfile(dst_filepath):
laurent@782:                 dialog = wx.MessageDialog(self, 
laurent@782:                       _("The file '%s' already exist.\nDo you want to replace it?") % src_filename, 
laurent@782:                       _("Replace File"), wx.YES_NO|wx.ICON_QUESTION)
laurent@782:                 copy = dialog.ShowModal() == wx.ID_YES
laurent@782:                 dialog.Destroy()
laurent@782:             else:
laurent@782:                 copy = True
laurent@782:                 
laurent@782:             if copy:
laurent@782:                 shutil.copyfile(src, dst_filepath)
laurent@782:                 return dst_filepath
laurent@782:         return None
laurent@782: 
laurent@782:     def OnLeftCopyButton(self, event):
laurent@782:         filepath = self.CopyFile(self.SystemDir.GetPath(), self.ManagedDir.GetPath())
laurent@782:         if filepath is not None:
laurent@782:             self.ManagedDir.RefreshTree()
laurent@782:             self.ManagedDir.SetPath(filepath)
laurent@782:         event.Skip()
laurent@782: 
laurent@782:     def OnRightCopyButton(self, event):
laurent@782:         filepath = self.CopyFile(self.ManagedDir.GetPath(), self.SystemDir.GetPath())
laurent@782:         if filepath is not None:
laurent@782:             self.SystemDir.RefreshTree()
laurent@782:             self.SystemDir.SetPath(filepath)
laurent@782:         event.Skip()
laurent@782:     
laurent@782:     def OnTreeBeginDrag(self, event):
laurent@782:         filepath = self.ManagedDir.GetPath()
laurent@782:         if os.path.isfile(filepath):
laurent@782:             relative_filepath = filepath.replace(os.path.join(self.Folder, ""), "")
laurent@782:             data = wx.TextDataObject(str(("'%s'" % relative_filepath, "Constant")))
laurent@782:             dragSource = wx.DropSource(self)
laurent@782:             dragSource.SetData(data)
laurent@782:             dragSource.DoDragDrop()
laurent@782: