1

import os

     2

from PLCControler import UndoBuffer

     3

from PythonEditor import PythonEditor

     4

     5

from xml.dom import minidom

     7

import cPickle

     8

     9

from CodeFileTreeNode import CodeFile

    10

    11

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

    12

    13

class PythonFileCTNMixin(CodeFile):

    14

    22

    EditorType = PythonEditor

    23

    24

    def __init__(self):

    25

        CodeFile.__init__(self)

    26

    27

        filepath = self.PythonFileName()

    28

    29

        python_code = PythonClasses["Python"]()

    30

        if os.path.isfile(filepath):

    31

            xmlfile = open(filepath, 'r')

    32

            tree = minidom.parse(xmlfile)

    33

            xmlfile.close()

    34

    35

            for child in tree.childNodes:

    36

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

    37

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

    38

                    self.CodeFile.globals.settext(python_code.gettext())

    39

                    os.remove(filepath)

    40

                    self.CreateCodeFileBuffer(False)

    41

                    self.OnCTNSave()

    42

    43

    def CodeFileName(self):

    44

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

    45

    46

    def PythonFileName(self):

    47

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

    48

    50

    52

        config = self.GetCTRoot().GetProjectConfigNames()[0]

    54

        for variable in self.CodeFile.variables.variable:

    55

            global_name = "%s_%s" % (config.upper(), variable.getname().upper())

    57

                global_name,

    58

                variable.getname(),

    59

                variable.gettype(),

    60

                str(variable.getinitial()))

    61

    66

    80