1

import os

     2

from PLCControler import UndoBuffer

     3

from PythonEditor import PythonEditor

     4

     5

from xml.dom import minidom

     6

from xmlclass import GenerateClassesFromXSD

     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

    15

    CODEFILE_NAME = "PyFile"

    16

    SECTIONS_NAMES = [

    17

        "globals",

    18

        "init",

    19

        "cleanup",

    20

        "start",

    21

        "stop"]

    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

    49

    def GetSectionsCode(self):

    50

    51

        # Generate Beremiz python runtime variables code

    52

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

    53

        variables_str = ""

    54

        for variable in self.CodeFile.variables.variable:

    55

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

    56

            variables_str += "# global_var:%s python_var:%s type:%s initial:%s\n" % (

    57

                global_name,

    58

                variable.getname(),

    59

                variable.gettype(),

    60

                str(variable.getinitial()))

    61

    62

        sections_code = {

    63

            "variables": variables_str,

    64

            "globals": self.CodeFile.globals.gettext().strip()

    65

        }

    66

    67

        # Generate Beremiz python runtime functions code

    68

        for section in self.SECTIONS_NAMES:

    69

            if section != "globals":

    70

                code_object = getattr(self.CodeFile, section)

    71

                section_str = ""

    72

                lines = code_object.gettext().strip().splitlines()

    73

                if len(lines) > 0:

    74

                    for line in lines:

    75

                        section_str += "    " + line + "\n"

    76

                    section_str += "\n"

    77

                sections_code[section] = section_str

    78

    79

        return sections_code

    80