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

        filepath = self.PythonFileName()

    18

    19

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

    20

        if os.path.isfile(filepath):

    21

            xmlfile = open(filepath, 'r')

    22

            tree = minidom.parse(xmlfile)

    23

            xmlfile.close()

    24

    25

            for child in tree.childNodes:

    26

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

    27

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

    28

                    self.CreatePythonBuffer(True)

    29

        else:

    30

            self.CreatePythonBuffer(False)

    31

            self.OnCTNSave()

    32

    33

    def PythonFileName(self):

    34

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

    35

    36

    def GetFilename(self):

    37

        if self.PythonBuffer.IsCurrentSaved():

    38

            return "py_ext"

    39

        else:

    40

            return "~py_ext~"

    41

    42

    def SetPythonCode(self, text):

    43

        self.PythonCode.settext(text)

    44

    45

    def GetPythonCode(self):

    46

        return self.PythonCode.gettext()

    47

    48

    def CTNTestModified(self):

    49

        return self.ChangesToSave or not self.PythonIsSaved()

    50

    51

    def OnCTNSave(self):

    52

        filepath = self.PythonFileName()

    53

    54

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

    55

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

    56

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

    57

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

    58

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

    59

    60

        xmlfile = open(filepath,"w")

    61

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

    62

        xmlfile.close()

    63

    64

        self.MarkPythonAsSaved()

    65

        return True

    66

    67

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

    68

#                      Current Buffering Management Functions

    69

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

    70

    71

    """

    72

    Return a copy of the project

    73

    """

    74

    def Copy(self, model):

    75

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

    76

    77

    def CreatePythonBuffer(self, saved):

    78

        self.Buffering = False

    79

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

    80

    81

    def BufferPython(self):

    82

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

    83

    84

    def StartBuffering(self):

    85

        self.Buffering = True

    86

    87

    def EndBuffering(self):

    88

        if self.Buffering:

    89

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

    90

            self.Buffering = False

    91

    92

    def MarkPythonAsSaved(self):

    93

        self.EndBuffering()

    94

        self.PythonBuffer.CurrentSaved()

    95

    96

    def PythonIsSaved(self):

    97

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

    98

    99

    def LoadPrevious(self):

   100

        self.EndBuffering()

   101

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

   102

   103

    def LoadNext(self):

   104

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

   105

   106

    def GetBufferState(self):

   107

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

   108

        last = self.PythonBuffer.IsLast()

   109

        return not first, not last

   110