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 program is free software; you can redistribute it and/or

    12

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

    13

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

    14

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

    15

#

    16

# This program 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

    19

# GNU General Public License for more details.

    20

#

    21

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

    22

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

    23

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

    24

    25

    26

    27

import os

    28

import shutil

    29

    30

import wx

    31

import wx.lib.buttons

    32

    33

from editors.EditorPanel import EditorPanel

    34

from util.BitmapLibrary import GetBitmap

    35

from controls import FolderTree

    36

    37

    38

class FileManagementPanel(EditorPanel):

    39

    40

    def _init_Editor(self, parent):

    41

        self.Editor = wx.Panel(parent)

    42

    43

        main_sizer = wx.BoxSizer(wx.HORIZONTAL)

    44

    45

        left_sizer = wx.BoxSizer(wx.VERTICAL)

    46

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

    47

    48

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

    49

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

    50

    51

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

    52

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

    53

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

    54

    55

        managed_treectrl = self.ManagedDir.GetTreeCtrl()

    56

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

    57

        if self.EnableDragNDrop:

    58

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

    59

    60

        button_sizer = wx.BoxSizer(wx.VERTICAL)

    61

        main_sizer.Add(button_sizer, border=5,

    62

                            flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL)

    63

    64

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

    65

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

    66

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

    67

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

    68

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

    69

            button = wx.lib.buttons.GenBitmapButton(

    70

                self.Editor,

    71

                bitmap=GetBitmap(bitmap),

    72

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

    73

            button.SetToolTip(help)

    74

            setattr(self, name, button)

    75

            if idx > 0:

    76

                flag = wx.TOP

    77

            else:

    78

                flag = 0

    79

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

    80

            button_sizer.Add(button, border=20, flag=flag)

    81

    82

        right_sizer = wx.BoxSizer(wx.VERTICAL)

    83

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

    84

    85

        if wx.Platform == '__WXMSW__':

    86

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

    87

        else:

    88

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

    89

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

    90

    91

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

    92

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

    93

    94

        system_treectrl = self.SystemDir.GetTreeCtrl()

    95

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

    96

    97

        self.Editor.SetSizer(main_sizer)

    98

    99

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

   100

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

   101

        self.EnableDragNDrop = enable_dragndrop

   102

   103

        if wx.Platform == '__WXMSW__':

   104

            self.HomeDirectory = "/"

   105

        else:

   106

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

   107

   108

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

   109

   110

        self.Controler = controler

   111

   112

        self.EditableFileExtensions = []

   113

        self.EditButton.Hide()

   114

   115

        self.SetIcon(GetBitmap("FOLDER"))

   116

   117

    def GetTitle(self):

   118

        return _(self.TagName)

   119

   120

    def SetEditableFileExtensions(self, extensions):

   121

        self.EditableFileExtensions = extensions

   122

        if len(self.EditableFileExtensions) > 0:

   123

            self.EditButton.Show()

   124

   125

    def RefreshView(self):

   126

        self.ManagedDir.RefreshTree()

   127

        self.SystemDir.RefreshTree()

   128

        self.RefreshButtonsState()

   129

   130

    def RefreshButtonsState(self):

   131

        managed_filepath = self.ManagedDir.GetPath()

   132

        system_filepath = self.SystemDir.GetPath()

   133

   134

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

   135

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

   136

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

   137

        if len(self.EditableFileExtensions) > 0:

   138

            self.EditButton.Enable(

   139

                os.path.isfile(managed_filepath) and

   140

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

   141

   142

    def OnTreeItemChanged(self, event):

   143

        self.RefreshButtonsState()

   144

        event.Skip()

   145

   146

    def OnDeleteButton(self, event):

   147

        filepath = self.ManagedDir.GetPath()

   148

        if os.path.isfile(filepath):

   149

            _folder, filename = os.path.split(filepath)

   150

   151

            dialog = wx.MessageDialog(self,

   152

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

   153

                                      _("Delete File"),

   154

                                      wx.YES_NO | wx.ICON_QUESTION)

   155

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

   156

            dialog.Destroy()

   157

   158

            if remove:

   159

                os.remove(filepath)

   160

                self.ManagedDir.RefreshTree()

   161

        event.Skip()

   162

   163

    def OnEditButton(self, event):

   164

        filepath = self.ManagedDir.GetPath()

   165

        if os.path.isfile(filepath) and \

   166

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

   167

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

   168

        event.Skip()

   169

   170

    def CopyFile(self, src, dst):

   171

        if os.path.isfile(src):

   172

            _src_folder, src_filename = os.path.split(src)

   173

            if os.path.isfile(dst):

   174

                dst_folder, _dst_filename = os.path.split(dst)

   175

            else:

   176

                dst_folder = dst

   177

   178

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

   179

            if os.path.isfile(dst_filepath):

   180

                dialog = wx.MessageDialog(

   181

                    self,

   182

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

   183

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

   184

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

   185

                dialog.Destroy()

   186

            else:

   187

                copy = True

   188

   189

            if copy:

   190

                shutil.copyfile(src, dst_filepath)

   191

                return dst_filepath

   192

        return None

   193

   194

    def OnLeftCopyButton(self, event):

   195

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

   196

        if filepath is not None:

   197

            self.ManagedDir.RefreshTree()

   198

            self.ManagedDir.SetPath(filepath)

   199

        event.Skip()

   200

   201

    def OnRightCopyButton(self, event):

   202

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

   203

        if filepath is not None:

   204

            self.SystemDir.RefreshTree()

   205

            self.SystemDir.SetPath(filepath)

   206

        event.Skip()

   207

   208

    def OnTreeBeginDrag(self, event):

   209

        filepath = self.ManagedDir.GetPath()

   210

        if os.path.isfile(filepath):

   211

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

   212

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

   213

            dragSource = wx.DropSource(self)

   214

            dragSource.SetData(data)

   215

            dragSource.DoDragDrop()