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

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

    38

    39

    40

class FileManagementPanel(EditorPanel):

    41

    42

    def _init_Editor(self, parent):

    43

        self.Editor = wx.Panel(parent)

    44

    45

        main_sizer = wx.BoxSizer(wx.HORIZONTAL)

    46

    47

        left_sizer = wx.BoxSizer(wx.VERTICAL)

    48

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

    49

    50

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

    51

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

    52

    53

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

    54

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

    55

    56

        managed_treectrl = self.ManagedDir.GetTreeCtrl()

    57

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

    58

        if self.EnableDragNDrop:

    59

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

    60

    61

        button_sizer = wx.BoxSizer(wx.VERTICAL)

    62

        main_sizer.AddSizer(button_sizer, border=5,

    63

                            flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL)

    64

    65

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

    66

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

    67

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

    68

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

    69

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

    70

            button = wx.lib.buttons.GenBitmapButton(

    71

                self.Editor,

    72

                bitmap=GetBitmap(bitmap),

    73

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

    74

            button.SetToolTipString(help)

    75

            setattr(self, name, button)

    76

            if idx > 0:

    77

                flag = wx.TOP

    78

            else:

    79

                flag = 0

    80

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

    81

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

    82

    83

        right_sizer = wx.BoxSizer(wx.VERTICAL)

    84

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

    85

    86

        if wx.Platform == '__WXMSW__':

    87

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

    88

        else:

    89

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

    90

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

    91

    92

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

    93

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

    94

    95

        system_treectrl = self.SystemDir.GetTreeCtrl()

    96

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

    97

    98

        self.Editor.SetSizer(main_sizer)

    99

   100

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

   101

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

   102

        self.EnableDragNDrop = enable_dragndrop

   103

   104

        if wx.Platform == '__WXMSW__':

   105

            self.HomeDirectory = "/"

   106

        else:

   107

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

   108

   109

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

   110

   111

        self.Controler = controler

   112

   113

        self.EditableFileExtensions = []

   114

        self.EditButton.Hide()

   115

   116

        self.SetIcon(GetBitmap("FOLDER"))

   117

   118

    def __del__(self):

   119

        self.Controler.OnCloseEditor(self)

   120

   121

    def GetTitle(self):

   122

        return _(self.TagName)

   123

   124

    def SetEditableFileExtensions(self, extensions):

   125

        self.EditableFileExtensions = extensions

   126

        if len(self.EditableFileExtensions) > 0:

   127

            self.EditButton.Show()

   128

   129

    def RefreshView(self):

   130

        self.ManagedDir.RefreshTree()

   131

        self.SystemDir.RefreshTree()

   132

        self.RefreshButtonsState()

   133

   134

    def RefreshButtonsState(self):

   135

        managed_filepath = self.ManagedDir.GetPath()

   136

        system_filepath = self.SystemDir.GetPath()

   137

   138

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

   139

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

   140

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

   141

        if len(self.EditableFileExtensions) > 0:

   142

            self.EditButton.Enable(

   143

                os.path.isfile(managed_filepath) and

   144

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

   145

   146

    def OnTreeItemChanged(self, event):

   147

        self.RefreshButtonsState()

   148

        event.Skip()

   149

   150

    def OnDeleteButton(self, event):

   151

        filepath = self.ManagedDir.GetPath()

   152

        if os.path.isfile(filepath):

   153

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

   154

   155

            dialog = wx.MessageDialog(self,

   156

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

   157

                                      _("Delete File"),

   158

                                      wx.YES_NO | wx.ICON_QUESTION)

   159

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

   160

            dialog.Destroy()

   161

   162

            if remove:

   163

                os.remove(filepath)

   164

                self.ManagedDir.RefreshTree()

   165

        event.Skip()

   166

   167

    def OnEditButton(self, event):

   168

        filepath = self.ManagedDir.GetPath()

   169

        if os.path.isfile(filepath) and \

   170

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

   171

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

   172

        event.Skip()

   173

   174

    def CopyFile(self, src, dst):

   175

        if os.path.isfile(src):

   176

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

   177

            if os.path.isfile(dst):

   178

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

   179

            else:

   180

                dst_folder = dst

   181

   182

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

   183

            if os.path.isfile(dst_filepath):

   184

                dialog = wx.MessageDialog(

   185

                    self,

   186

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

   187

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

   188

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

   189

                dialog.Destroy()

   190

            else:

   191

                copy = True

   192

   193

            if copy:

   194

                shutil.copyfile(src, dst_filepath)

   195

                return dst_filepath

   196

        return None

   197

   198

    def OnLeftCopyButton(self, event):

   199

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

   200

        if filepath is not None:

   201

            self.ManagedDir.RefreshTree()

   202

            self.ManagedDir.SetPath(filepath)

   203

        event.Skip()

   204

   205

    def OnRightCopyButton(self, event):

   206

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

   207

        if filepath is not None:

   208

            self.SystemDir.RefreshTree()

   209

            self.SystemDir.SetPath(filepath)

   210

        event.Skip()

   211

   212

    def OnTreeBeginDrag(self, event):

   213

        filepath = self.ManagedDir.GetPath()

   214

        if os.path.isfile(filepath):

   215

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

   216

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

   217

            dragSource = wx.DropSource(self)

   218

            dragSource.SetData(data)

   219

            dragSource.DoDragDrop()