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

from __future__ import absolute_import

    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 __del__(self):

   118

        self.Controler.OnCloseEditor(self)

   119

   120

    def GetTitle(self):

   121

        return _(self.TagName)

   122

   123

    def SetEditableFileExtensions(self, extensions):

   124

        self.EditableFileExtensions = extensions

   125

        if len(self.EditableFileExtensions) > 0:

   126

            self.EditButton.Show()

   127

   128

    def RefreshView(self):

   129

        self.ManagedDir.RefreshTree()

   130

        self.SystemDir.RefreshTree()

   131

        self.RefreshButtonsState()

   132

   133

    def RefreshButtonsState(self):

   134

        managed_filepath = self.ManagedDir.GetPath()

   135

        system_filepath = self.SystemDir.GetPath()

   136

   137

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

   138

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

   139

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

   140

        if len(self.EditableFileExtensions) > 0:

   141

            self.EditButton.Enable(

   142

                os.path.isfile(managed_filepath) and

   143

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

   144

   145

    def OnTreeItemChanged(self, event):

   146

        self.RefreshButtonsState()

   147

        event.Skip()

   148

   149

    def OnDeleteButton(self, event):

   150

        filepath = self.ManagedDir.GetPath()

   151

        if os.path.isfile(filepath):

   152

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

   153

   154

            dialog = wx.MessageDialog(self,

   155

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

   156

                                      _("Delete File"),

   157

                                      wx.YES_NO | wx.ICON_QUESTION)

   158

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

   159

            dialog.Destroy()

   160

   161

            if remove:

   162

                os.remove(filepath)

   163

                self.ManagedDir.RefreshTree()

   164

        event.Skip()

   165

   166

    def OnEditButton(self, event):

   167

        filepath = self.ManagedDir.GetPath()

   168

        if os.path.isfile(filepath) and \

   169

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

   170

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

   171

        event.Skip()

   172

   173

    def CopyFile(self, src, dst):

   174

        if os.path.isfile(src):

   175

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

   176

            if os.path.isfile(dst):

   177

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

   178

            else:

   179

                dst_folder = dst

   180

   181

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

   182

            if os.path.isfile(dst_filepath):

   183

                dialog = wx.MessageDialog(

   184

                    self,

   185

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

   186

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

   187

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

   188

                dialog.Destroy()

   189

            else:

   190

                copy = True

   191

   192

            if copy:

   193

                shutil.copyfile(src, dst_filepath)

   194

                return dst_filepath

   195

        return None

   196

   197

    def OnLeftCopyButton(self, event):

   198

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

   199

        if filepath is not None:

   200

            self.ManagedDir.RefreshTree()

   201

            self.ManagedDir.SetPath(filepath)

   202

        event.Skip()

   203

   204

    def OnRightCopyButton(self, event):

   205

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

   206

        if filepath is not None:

   207

            self.SystemDir.RefreshTree()

   208

            self.SystemDir.SetPath(filepath)

   209

        event.Skip()

   210

   211

    def OnTreeBeginDrag(self, event):

   212

        filepath = self.ManagedDir.GetPath()

   213

        if os.path.isfile(filepath):

   214

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

   215

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

   216

            dragSource = wx.DropSource(self)

   217

            dragSource.SetData(data)

   218

            dragSource.DoDragDrop()