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

import os

    26

import shutil

    27

    28

import wx

    29

import wx.lib.buttons

    30

    31

from EditorPanel import EditorPanel

    32

from util.BitmapLibrary import GetBitmap

    33

from controls import FolderTree

    34

    35

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

    36

    37

class FileManagementPanel(EditorPanel):

    38

    39

    def _init_Editor(self, parent):

    40

        self.Editor = wx.Panel(parent)

    41

    42

        main_sizer = wx.BoxSizer(wx.HORIZONTAL)

    43

    44

        left_sizer = wx.BoxSizer(wx.VERTICAL)

    45

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

    46

    47

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

    48

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

    49

    50

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

    51

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

    52

    53

        managed_treectrl = self.ManagedDir.GetTreeCtrl()

    54

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

    55

        if self.EnableDragNDrop:

    56

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

    57

    58

        button_sizer = wx.BoxSizer(wx.VERTICAL)

    59

        main_sizer.AddSizer(button_sizer, border=5, 

    60

              flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL)

    61

    62

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

    63

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

    64

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

    65

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

    66

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

    67

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

    68

                  bitmap=GetBitmap(bitmap), 

    69

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

    70

            button.SetToolTipString(help)

    71

            setattr(self, name, button)

    72

            if idx > 0:

    73

                flag = wx.TOP

    74

            else:

    75

                flag = 0

    76

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

    77

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

    78

    79

        right_sizer = wx.BoxSizer(wx.VERTICAL)

    80

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

    81

    82

        if wx.Platform == '__WXMSW__':

    83

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

    84

        else:

    85

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

    86

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

    87

    88

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

    89

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

    90

    91

        system_treectrl = self.SystemDir.GetTreeCtrl()

    92

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

    93

    94

        self.Editor.SetSizer(main_sizer)

    95

    96

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

    97

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

    98

        self.EnableDragNDrop = enable_dragndrop

    99

   100

        if wx.Platform == '__WXMSW__':

   101

            self.HomeDirectory = "/"

   102

        else:

   103

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

   104

   105

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

   106

   107

        self.Controler = controler

   108

   109

        self.EditableFileExtensions = []

   110

        self.EditButton.Hide()

   111

   112

        self.SetIcon(GetBitmap("FOLDER"))

   113

   114

    def __del__(self):

   115

        self.Controler.OnCloseEditor(self)

   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"), wx.YES_NO|wx.ICON_QUESTION)

   154

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

   155

            dialog.Destroy()

   156

   157

            if remove:

   158

                os.remove(filepath)

   159

                self.ManagedDir.RefreshTree()

   160

        event.Skip()

   161

   162

    def OnEditButton(self, event):

   163

        filepath = self.ManagedDir.GetPath()

   164

        if (os.path.isfile(filepath) and 

   165

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

   166

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

   167

        event.Skip()

   168

   169

    def CopyFile(self, src, dst):

   170

        if os.path.isfile(src):

   171

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

   172

            if os.path.isfile(dst):

   173

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

   174

            else:

   175

                dst_folder = dst

   176

   177

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

   178

            if os.path.isfile(dst_filepath):

   179

                dialog = wx.MessageDialog(self, 

   180

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

   181

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

   182

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

   183

                dialog.Destroy()

   184

            else:

   185

                copy = True

   186

   187

            if copy:

   188

                shutil.copyfile(src, dst_filepath)

   189

                return dst_filepath

   190

        return None

   191

   192

    def OnLeftCopyButton(self, event):

   193

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

   194

        if filepath is not None:

   195

            self.ManagedDir.RefreshTree()

   196

            self.ManagedDir.SetPath(filepath)

   197

        event.Skip()

   198

   199

    def OnRightCopyButton(self, event):

   200

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

   201

        if filepath is not None:

   202

            self.SystemDir.RefreshTree()

   203

            self.SystemDir.SetPath(filepath)

   204

        event.Skip()

   205

   206

    def OnTreeBeginDrag(self, event):

   207

        filepath = self.ManagedDir.GetPath()

   208

        if os.path.isfile(filepath):

   209

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

   210

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

   211

            dragSource = wx.DropSource(self)

   212

            dragSource.SetData(data)

   213

            dragSource.DoDragDrop()

   214