1

import os

     2

from PLCControler import UndoBuffer

     3

from PythonEditor import PythonEditor

     4

     5

from xml.dom import minidom

     6

from xmlclass import *

     7

import cPickle

     8

     9

PythonClasses = GenerateClassesFromXSD(os.path.join(os.path.dirname(__file__), "py_ext_xsd.xsd")) 

    10

    11

class PythonFileCTNMixin:

    12

    13

    EditorType = PythonEditor

    14

    15

    def __init__(self):

    16

    17

        self.ConfNodeMethods.insert(0, 

    18

                {"bitmap" : "editPYTHONcode",

    19

                 "name" : _("Edit Python File"), 

    20

                 "tooltip" : _("Edit Python File"),

    21

                 "method" : "_OpenView"},

    22

        )

    23

    24

        filepath = self.PythonFileName()

    25

    26

        self.PythonCode = PythonClasses["Python"]()

    27

        if os.path.isfile(filepath):

    28

            xmlfile = open(filepath, 'r')

    29

            tree = minidom.parse(xmlfile)

    30

            xmlfile.close()

    31

    32

            for child in tree.childNodes:

    33

                if child.nodeType == tree.ELEMENT_NODE and child.nodeName == "Python":

    34

                    self.PythonCode.loadXMLTree(child, ["xmlns", "xmlns:xsi", "xsi:schemaLocation"])

    35

                    self.CreatePythonBuffer(True)

    36

        else:

    37

            self.CreatePythonBuffer(False)

    38

            self.OnCTNSave()

    39

    40

    def PythonFileName(self):

    41

        return os.path.join(self.CTNPath(), "py_ext.xml")

    42

    43

    def GetFilename(self):

    44

        if self.PythonBuffer.IsCurrentSaved():

    45

            return "py_ext"

    46

        else:

    47

            return "~py_ext~"

    48

    49

    def SetPythonCode(self, text):

    50

        self.PythonCode.settext(text)

    51

    52

    def GetPythonCode(self):

    53

        return self.PythonCode.gettext()

    54

    55

    def CTNTestModified(self):

    56

        return self.ChangesToSave or not self.PythonIsSaved()

    57

    58

    def OnCTNSave(self):

    59

        filepath = self.PythonFileName()

    60

    61

        text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n"

    62

        extras = {"xmlns":"http://www.w3.org/2001/XMLSchema",

    63

                  "xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance",

    64

                  "xsi:schemaLocation" : "py_ext_xsd.xsd"}

    65

        text += self.PythonCode.generateXMLText("Python", 0, extras)

    66

    67

        xmlfile = open(filepath,"w")

    68

        xmlfile.write(text.encode("utf-8"))

    69

        xmlfile.close()

    70

    71

        self.MarkPythonAsSaved()

    72

        return True

    73

    74

#-------------------------------------------------------------------------------

    75

#                      Current Buffering Management Functions

    76

#-------------------------------------------------------------------------------

    77

    78

    """

    79

    Return a copy of the project

    80

    """

    81

    def Copy(self, model):

    82

        return cPickle.loads(cPickle.dumps(model))

    83

    84

    def CreatePythonBuffer(self, saved):

    85

        self.Buffering = False

    86

        self.PythonBuffer = UndoBuffer(cPickle.dumps(self.PythonCode), saved)

    87

    88

    def BufferPython(self):

    89

        self.PythonBuffer.Buffering(cPickle.dumps(self.PythonCode))

    90

    91

    def StartBuffering(self):

    92

        self.Buffering = True

    93

    94

    def EndBuffering(self):

    95

        if self.Buffering:

    96

            self.PythonBuffer.Buffering(cPickle.dumps(self.PythonCode))

    97

            self.Buffering = False

    98

    99

    def MarkPythonAsSaved(self):

   100

        self.EndBuffering()

   101

        self.PythonBuffer.CurrentSaved()

   102

   103

    def PythonIsSaved(self):

   104

        return self.PythonBuffer.IsCurrentSaved() and not self.Buffering

   105

   106

    def LoadPrevious(self):

   107

        self.EndBuffering()

   108

        self.PythonCode = cPickle.loads(self.PythonBuffer.Previous())

   109

   110

    def LoadNext(self):

   111

        self.PythonCode = cPickle.loads(self.PythonBuffer.Next())

   112

   113

    def GetBufferState(self):

   114

        first = self.PythonBuffer.IsFirst() and not self.Buffering

   115

        last = self.PythonBuffer.IsLast()

   116

        return not first, not last

   117