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

    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.AddSizer(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.AddWindow(managed_dir_label, border=5, flag=wx.GROW | wx.BOTTOM)

    50

    51

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

    52

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

    53

    54

        managed_treectrl = self.ManagedDir.GetTreeCtrl()

    55

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

    56

        if self.EnableDragNDrop:

    57

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

    58

    59

        button_sizer = wx.BoxSizer(wx.VERTICAL)

    60

        main_sizer.AddSizer(button_sizer, border=5,

    61

                            flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL)

    62

    63

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

    64

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

    65

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

    66

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

    67

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

    68

            button = wx.lib.buttons.GenBitmapButton(

    69

                self.Editor,

    70

                bitmap=GetBitmap(bitmap),

    71

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

    72

            button.SetToolTipString(help)

    73

            setattr(self, name, button)

    74

            if idx > 0:

    75

                flag = wx.TOP

    76

            else:

    77

                flag = 0

    78

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

    79

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

    80

    81

        right_sizer = wx.BoxSizer(wx.VERTICAL)

    82

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

    83

    84

        if wx.Platform == '__WXMSW__':

    85

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

    86

        else:

    87

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

    88

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

    89

    90

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

    91

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

    92

    93

        system_treectrl = self.SystemDir.GetTreeCtrl()

    94

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

    95

    96

        self.Editor.SetSizer(main_sizer)

    97

    98

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

    99

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

   100

        self.EnableDragNDrop = enable_dragndrop

   101

   102

        if wx.Platform == '__WXMSW__':

   103

            self.HomeDirectory = "/"

   104

        else:

   105

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

   106

   107

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

   108

   109

        self.Controler = controler

   110

   111

        self.EditableFileExtensions = []

   112

        self.EditButton.Hide()

   113

   114

        self.SetIcon(GetBitmap("FOLDER"))

   115

   116

    def __del__(self):

   117

        self.Controler.OnCloseEditor(self)

   118

   119

    def GetTitle(self):

   120

        return _(self.TagName)

   121

   122

    def SetEditableFileExtensions(self, extensions):

   123

        self.EditableFileExtensions = extensions

   124

        if len(self.EditableFileExtensions) > 0:

   125

            self.EditButton.Show()

   126

   127

    def RefreshView(self):

   128

        self.ManagedDir.RefreshTree()

   129

        self.SystemDir.RefreshTree()

   130

        self.RefreshButtonsState()

   131

   132

    def RefreshButtonsState(self):

   133

        managed_filepath = self.ManagedDir.GetPath()

   134

        system_filepath = self.SystemDir.GetPath()

   135

   136

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

   137

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

   138

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

   139

        if len(self.EditableFileExtensions) > 0:

   140

            self.EditButton.Enable(

   141

                os.path.isfile(managed_filepath) and

   142

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

   143

   144

    def OnTreeItemChanged(self, event):

   145

        self.RefreshButtonsState()

   146

        event.Skip()

   147

   148

    def OnDeleteButton(self, event):

   149

        filepath = self.ManagedDir.GetPath()

   150

        if os.path.isfile(filepath):

   151

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

   152

   153

            dialog = wx.MessageDialog(self,

   154

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

   155

                                      _("Delete File"),

   156

                                      wx.YES_NO | wx.ICON_QUESTION)

   157

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

   158

            dialog.Destroy()

   159

   160

            if remove:

   161

                os.remove(filepath)

   162

                self.ManagedDir.RefreshTree()

   163

        event.Skip()

   164

   165

    def OnEditButton(self, event):

   166

        filepath = self.ManagedDir.GetPath()

   167

        if os.path.isfile(filepath) and \

   168

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

   169

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

   170

        event.Skip()

   171

   172

    def CopyFile(self, src, dst):

   173

        if os.path.isfile(src):

   174

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

   175

            if os.path.isfile(dst):

   176

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

   177

            else:

   178

                dst_folder = dst

   179

   180

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

   181

            if os.path.isfile(dst_filepath):

   182

                dialog = wx.MessageDialog(

   183

                    self,

   184

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

   185

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

   186

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

   187

                dialog.Destroy()

   188

            else:

   189

                copy = True

   190

   191

            if copy:

   192

                shutil.copyfile(src, dst_filepath)

   193

                return dst_filepath

   194

        return None

   195

   196

    def OnLeftCopyButton(self, event):

   197

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

   198

        if filepath is not None:

   199

            self.ManagedDir.RefreshTree()

   200

            self.ManagedDir.SetPath(filepath)

   201

        event.Skip()

   202

   203

    def OnRightCopyButton(self, event):

   204

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

   205

        if filepath is not None:

   206

            self.SystemDir.RefreshTree()

   207

            self.SystemDir.SetPath(filepath)

   208

        event.Skip()

   209

   210

    def OnTreeBeginDrag(self, event):

   211

        filepath = self.ManagedDir.GetPath()

   212

        if os.path.isfile(filepath):

   213

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

   214

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

   215

            dragSource = wx.DropSource(self)

   216

            dragSource.SetData(data)

   217

            dragSource.DoDragDrop()