1 #!/usr/bin/env python

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

     3 

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

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

     6 #

     7 #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8 #

     9 #See COPYING file for copyrights details.

    10 #

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

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

    13 #License as published by the Free Software Foundation; either

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

    15 #

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

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

    18 #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    19 #General Public License for more details.

    20 #

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

    22 #License along with this library; if not, write to the Free Software

    23 #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    24 

    25 import os

    26 import shutil

    27 

    28 import wx

    29 

    30 from EditorPanel import EditorPanel

    31 from util.BitmapLibrary import GetBitmap

    32 

    33 DRIVE, FOLDER, FILE = range(3)

    34 

    35 FILTER = _("All files (*.*)|*.*|CSV files (*.csv)|*.csv")

    36 

    37 def sort_folder(x, y):

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

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

    40     elif x[1] != FILE:

    41         return -1

    42     else:

    43         return 1

    44 

    45 def splitpath(path):

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

    47     if head == "":

    48         return [tail]

    49     elif tail == "":

    50         return splitpath(head)

    51     return splitpath(head) + [tail]

    52     

    53 class FolderTree(wx.Panel):

    54     

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

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

    57         

    58         main_sizer = wx.BoxSizer(wx.VERTICAL)

    59         

    60         self.Tree = wx.TreeCtrl(self, 

    61               style=wx.TR_HAS_BUTTONS|

    62                     wx.TR_SINGLE|

    63                     wx.SUNKEN_BORDER|

    64                     wx.TR_HIDE_ROOT|

    65                     wx.TR_LINES_AT_ROOT|

    66                     wx.TR_EDIT_LABELS)

    67         if wx.Platform == '__WXMSW__':

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

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

    70         else:

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

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

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

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

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

    76         

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

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

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

    80         

    81         self.SetSizer(main_sizer)

    82         

    83         self.Folder = folder

    84         self.Editable = editable

    85         

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

    87         self.TreeImageDict = {}

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

    89                                   (FOLDER, "tree_folder"),

    90                                   (FILE, "tree_file")]:

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

    92         self.Tree.SetImageList(self.TreeImageList)

    93         

    94         self.Filters = {}

    95         filter_parts = filter.split("|")

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

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

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

    99             else:

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

   101             self.Filter.Append(filter_parts[idx])

   102             if idx == 0:

   103                 self.Filter.SetStringSelection(filter_parts[idx])

   104             

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

   106     

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

   108         items = []

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

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

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

   112                 if os.path.exists(drive):

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

   114         else:

   115             try:

   116                 files = os.listdir(folderpath)

   117             except:

   118                 return []

   119             for filename in files:

   120                 if not filename.startswith("."):

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

   122                     if os.path.isdir(filepath):

   123                         if recursive:

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

   125                         else:

   126                             children = 0

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

   128                     elif (self.CurrentFilter == "" or 

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

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

   131         if recursive:

   132             items.sort(sort_folder)

   133         return items

   134     

   135     def GetTreeCtrl(self):

   136         return self.Tree

   137     

   138     def RefreshTree(self):

   139         root = self.Tree.GetRootItem()

   140         if not root.IsOk():

   141             root = self.Tree.AddRoot("")

   142         self.GenerateTreeBranch(root, self.Folder)

   143         

   144     def GenerateTreeBranch(self, root, folderpath):

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

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

   147             if not item.IsOk():

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

   149                 if wx.Platform != '__WXMSW__':

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

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

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

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

   154             if item_type != FILE:

   155                 if self.Tree.IsExpanded(item):

   156                     self.GenerateTreeBranch(item, filepath)

   157                 elif children > 0:

   158                     self.Tree.SetItemHasChildren(item)

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

   160         to_delete = []

   161         while item.IsOk():

   162             to_delete.append(item)

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

   164         for item in to_delete:

   165             self.Tree.Delete(item)

   166 

   167     def ExpandItem(self, item):

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

   169         self.Tree.Expand(item)

   170     

   171     def OnTreeItemActivated(self, event):

   172         self.ExpandItem(event.GetItem())

   173         event.Skip()

   174     

   175     def OnTreeLeftDown(self, event):

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

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

   178             self.ExpandItem(item)

   179         else:

   180             event.Skip()

   181     

   182     def OnTreeItemExpanded(self, event):

   183         item = event.GetItem()

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

   185         event.Skip()

   186 

   187     def OnTreeItemCollapsed(self, event):

   188         item = event.GetItem()

   189         self.Tree.DeleteChildren(item)

   190         self.Tree.SetItemHasChildren(item)

   191         event.Skip()

   192 

   193     def OnTreeBeginLabelEdit(self, event):

   194         item = event.GetItem()

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

   196             event.Skip()

   197         else:

   198             event.Veto()

   199     

   200     def OnTreeEndLabelEdit(self, event):

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

   202         new_filepath = os.path.join(os.path.split(old_filepath)[0], event.GetLabel())

   203         if new_filepath != old_filepath:

   204             if not os.path.exists(new_filepath):

   205                 os.rename(old_filepath, new_filepath)

   206                 event.Skip()

   207             else:

   208                 message =  wx.MessageDialog(self, 

   209                     _("File '%s' already exists!") % event.GetLabel(), 

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

   211                 message.ShowModal()

   212                 message.Destroy()

   213                 event.Veto()

   214     

   215     def OnFilterChanged(self, event):

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

   217         self.RefreshTree()

   218         event.Skip()

   219     

   220     def _SelectItem(self, root, parts):

   221         if len(parts) == 0:

   222             self.Tree.SelectItem(root)

   223         else:

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

   225             while item.IsOk():

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

   227                     if (self.Tree.ItemHasChildren(item) and 

   228                         not self.Tree.IsExpanded(item)):

   229                         self.Tree.Expand(item)

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

   231                     else:

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

   233                     return

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

   235     

   236     def SetPath(self, path):

   237         if path.startswith(self.Folder):

   238             root = self.Tree.GetRootItem()

   239             if root.IsOk():

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

   241                 self._SelectItem(root, splitpath(relative_path))

   242     

   243     def GetPath(self, item=None):

   244         if item is None:

   245             item = self.Tree.GetSelection()

   246         if item.IsOk():

   247             filepath = self.Tree.GetItemText(item)

   248             parent = self.Tree.GetItemParent(item)

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

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

   251                 parent = self.Tree.GetItemParent(parent)

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

   253         return self.Folder

   254 

   255 class FileManagementPanel(EditorPanel):

   256     

   257     def _init_Editor(self, parent):

   258         self.Editor = wx.Panel(parent)

   259         

   260         main_sizer = wx.BoxSizer(wx.HORIZONTAL)

   261         

   262         left_sizer = wx.BoxSizer(wx.VERTICAL)

   263         main_sizer.AddSizer(left_sizer, 1, border=5, flag=wx.GROW|wx.ALL)

   264         

   265         managed_dir_label = wx.StaticText(self.Editor, label=self.TagName + ":")

   266         left_sizer.AddWindow(managed_dir_label, border=5, flag=wx.GROW|wx.BOTTOM)

   267         

   268         self.ManagedDir = FolderTree(self.Editor, self.Folder, FILTER)

   269         left_sizer.AddWindow(self.ManagedDir, 1, flag=wx.GROW)

   270         

   271         managed_treectrl = self.ManagedDir.GetTreeCtrl()

   272         self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeItemChanged, managed_treectrl)

   273         if self.EnableDragNDrop:

   274             self.Bind(wx.EVT_TREE_BEGIN_DRAG, self.OnTreeBeginDrag, managed_treectrl)

   275         

   276         button_sizer = wx.BoxSizer(wx.VERTICAL)

   277         main_sizer.AddSizer(button_sizer, border=5, 

   278               flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL)

   279         

   280         for idx, (name, bitmap, help) in enumerate([

   281                 ("DeleteButton", "remove_element", _("Remove file from left folder")),

   282                 ("LeftCopyButton", "LeftCopy", _("Copy file from right folder to left")),

   283                 ("RightCopyButton", "RightCopy", _("Copy file from left folder to right")),

   284                 ("EditButton", "edit", _("Edit file"))]):

   285             button = wx.lib.buttons.GenBitmapButton(self.Editor, 

   286                   bitmap=GetBitmap(bitmap), 

   287                   size=wx.Size(28, 28), style=wx.NO_BORDER)

   288             button.SetToolTipString(help)

   289             setattr(self, name, button)

   290             if idx > 0:

   291                 flag = wx.TOP

   292             else:

   293                 flag = 0

   294             self.Bind(wx.EVT_BUTTON, getattr(self, "On" + name), button)

   295             button_sizer.AddWindow(button, border=20, flag=flag)

   296         

   297         right_sizer = wx.BoxSizer(wx.VERTICAL)

   298         main_sizer.AddSizer(right_sizer, 1, border=5, flag=wx.GROW|wx.ALL)

   299         

   300         if wx.Platform == '__WXMSW__':

   301             system_dir_label = wx.StaticText(self.Editor, label=_("My Computer:"))

   302         else:

   303             system_dir_label = wx.StaticText(self.Editor, label=_("Home Directory:"))

   304         right_sizer.AddWindow(system_dir_label, border=5, flag=wx.GROW|wx.BOTTOM)

   305         

   306         self.SystemDir = FolderTree(self.Editor, self.HomeDirectory, FILTER, False)

   307         right_sizer.AddWindow(self.SystemDir, 1, flag=wx.GROW)

   308         

   309         system_treectrl = self.SystemDir.GetTreeCtrl()

   310         self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeItemChanged, system_treectrl)

   311         

   312         self.Editor.SetSizer(main_sizer)

   313         

   314     def __init__(self, parent, controler, name, folder, enable_dragndrop=False):

   315         self.Folder = os.path.realpath(folder)

   316         self.EnableDragNDrop = enable_dragndrop

   317         

   318         if wx.Platform == '__WXMSW__':

   319             self.HomeDirectory = "/"

   320         else:

   321             self.HomeDirectory = os.path.expanduser("~")

   322         

   323         EditorPanel.__init__(self, parent, name, None, None)

   324         

   325         self.Controler = controler

   326         

   327         self.EditableFileExtensions = []

   328         self.EditButton.Hide()

   329         

   330         self.SetIcon(GetBitmap("FOLDER"))

   331     

   332     def __del__(self):

   333         self.Controler.OnCloseEditor(self)

   334     

   335     def GetTitle(self):

   336         return self.TagName

   337     

   338     def SetEditableFileExtensions(self, extensions):

   339         self.EditableFileExtensions = extensions

   340         if len(self.EditableFileExtensions) > 0:

   341             self.EditButton.Show()

   342     

   343     def RefreshView(self):

   344         self.ManagedDir.RefreshTree()

   345         self.SystemDir.RefreshTree()

   346         self.RefreshButtonsState()

   347     

   348     def RefreshButtonsState(self):

   349         managed_filepath = self.ManagedDir.GetPath()

   350         system_filepath = self.SystemDir.GetPath()

   351         

   352         self.DeleteButton.Enable(os.path.isfile(managed_filepath))

   353         self.LeftCopyButton.Enable(os.path.isfile(system_filepath))

   354         self.RightCopyButton.Enable(os.path.isfile(managed_filepath))

   355         if len(self.EditableFileExtensions) > 0:

   356             self.EditButton.Enable(

   357                 os.path.isfile(managed_filepath) and

   358                 os.path.splitext(managed_filepath)[1] in self.EditableFileExtensions)

   359     

   360     def OnTreeItemChanged(self, event):

   361         self.RefreshButtonsState()

   362         event.Skip()

   363         

   364     def OnDeleteButton(self, event):

   365         filepath = self.ManagedDir.GetPath()

   366         if os.path.isfile(filepath):

   367             folder, filename = os.path.split(filepath)

   368             

   369             dialog = wx.MessageDialog(self, 

   370                   _("Do you really want to delete the file '%s'?") % filename, 

   371                   _("Delete File"), wx.YES_NO|wx.ICON_QUESTION)

   372             remove = dialog.ShowModal() == wx.ID_YES

   373             dialog.Destroy()

   374             

   375             if remove:

   376                 os.remove(filepath)

   377                 self.ManagedDir.RefreshTree()

   378         event.Skip()

   379 

   380     def OnEditButton(self, event):

   381         filepath = self.ManagedDir.GetPath()

   382         if (os.path.isfile(filepath) and 

   383             os.path.splitext(filepath)[1] in self.EditableFileExtensions):

   384             self.Controler._OpenView(filepath + "::")

   385         event.Skip()

   386         

   387     def CopyFile(self, src, dst):

   388         if os.path.isfile(src):

   389             src_folder, src_filename = os.path.split(src)

   390             if os.path.isfile(dst):

   391                 dst_folder, dst_filename = os.path.split(dst)

   392             else:

   393                 dst_folder = dst

   394             

   395             dst_filepath = os.path.join(dst_folder, src_filename)

   396             if os.path.isfile(dst_filepath):

   397                 dialog = wx.MessageDialog(self, 

   398                       _("The file '%s' already exist.\nDo you want to replace it?") % src_filename, 

   399                       _("Replace File"), wx.YES_NO|wx.ICON_QUESTION)

   400                 copy = dialog.ShowModal() == wx.ID_YES

   401                 dialog.Destroy()

   402             else:

   403                 copy = True

   404                 

   405             if copy:

   406                 shutil.copyfile(src, dst_filepath)

   407                 return dst_filepath

   408         return None

   409 

   410     def OnLeftCopyButton(self, event):

   411         filepath = self.CopyFile(self.SystemDir.GetPath(), self.ManagedDir.GetPath())

   412         if filepath is not None:

   413             self.ManagedDir.RefreshTree()

   414             self.ManagedDir.SetPath(filepath)

   415         event.Skip()

   416 

   417     def OnRightCopyButton(self, event):

   418         filepath = self.CopyFile(self.ManagedDir.GetPath(), self.SystemDir.GetPath())

   419         if filepath is not None:

   420             self.SystemDir.RefreshTree()

   421             self.SystemDir.SetPath(filepath)

   422         event.Skip()

   423     

   424     def OnTreeBeginDrag(self, event):

   425         filepath = self.ManagedDir.GetPath()

   426         if os.path.isfile(filepath):

   427             relative_filepath = filepath.replace(os.path.join(self.Folder, ""), "")

   428             data = wx.TextDataObject(str(("'%s'" % relative_filepath, "Constant")))

   429             dragSource = wx.DropSource(self)

   430             dragSource.SetData(data)

   431             dragSource.DoDragDrop()

   432