1 import os

     2 from util import opjimg

     3 from PLCControler import UndoBuffer

     4 from PythonEditor import PythonEditor

     5 

     6 from xml.dom import minidom

     7 from xmlclass import *

     8 import cPickle

     9 

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

    11 

    12 class PythonFileCTNMixin:

    13     

    14     EditorType = PythonEditor

    15     

    16     def __init__(self):

    17         

    18         self.ConfNodeMethods.insert(0, 

    19                 {"bitmap" : opjimg("editPYTHONcode"),

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

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

    22                  "method" : "_OpenView"},

    23         )

    24 

    25         filepath = self.PythonFileName()

    26         

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

    28         if os.path.isfile(filepath):

    29             xmlfile = open(filepath, 'r')

    30             tree = minidom.parse(xmlfile)

    31             xmlfile.close()

    32             

    33             for child in tree.childNodes:

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

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

    36                     self.CreatePythonBuffer(True)

    37         else:

    38             self.CreatePythonBuffer(False)

    39             self.OnCTNSave()

    40 

    41     def PythonFileName(self):

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

    43 

    44     def GetFilename(self):

    45         if self.PythonBuffer.IsCurrentSaved():

    46             return "py_ext"

    47         else:

    48             return "~py_ext~"

    49 

    50     def SetPythonCode(self, text):

    51         self.PythonCode.settext(text)

    52         

    53     def GetPythonCode(self):

    54         return self.PythonCode.gettext()

    55     

    56     def CTNTestModified(self):

    57         return self.ChangesToSave or not self.PythonIsSaved()

    58     

    59     def OnCTNSave(self):

    60         filepath = self.PythonFileName()

    61         

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

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

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

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

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

    67 

    68         xmlfile = open(filepath,"w")

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

    70         xmlfile.close()

    71         

    72         self.MarkPythonAsSaved()

    73         return True

    74         

    75 #-------------------------------------------------------------------------------

    76 #                      Current Buffering Management Functions

    77 #-------------------------------------------------------------------------------

    78 

    79     """

    80     Return a copy of the project

    81     """

    82     def Copy(self, model):

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

    84 

    85     def CreatePythonBuffer(self, saved):

    86         self.Buffering = False

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

    88 

    89     def BufferPython(self):

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

    91     

    92     def StartBuffering(self):

    93         self.Buffering = True

    94         

    95     def EndBuffering(self):

    96         if self.Buffering:

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

    98             self.Buffering = False

    99     

   100     def MarkPythonAsSaved(self):

   101         self.EndBuffering()

   102         self.PythonBuffer.CurrentSaved()

   103     

   104     def PythonIsSaved(self):

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

   106         

   107     def LoadPrevious(self):

   108         self.EndBuffering()

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

   110     

   111     def LoadNext(self):

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

   113     

   114     def GetBufferState(self):

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

   116         last = self.PythonBuffer.IsLast()

   117         return not first, not last

   118