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

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

    EditorType = PythonEditor

    16

    17

    def __init__(self):

    18

        CodeFile.__init__(self)

    19

    20

        filepath = self.PythonFileName()

    21

    22

        python_code = PythonClasses["Python"]()

    23

        if os.path.isfile(filepath):

    24

            xmlfile = open(filepath, 'r')

    25

            tree = minidom.parse(xmlfile)

    26

            xmlfile.close()

    27

    28

            for child in tree.childNodes:

    29

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

    30

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

    31

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

    32

                    os.remove(filepath)

    33

                    self.CreateCodeFileBuffer(False)

    34

                    self.OnCTNSave()

    35

    36

    def CodeFileName(self):

    37

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

    38

    39

    def PythonFileName(self):

    40

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

    41

    42

    def GetPythonCode(self):

    43

        current_location = self.GetCurrentLocation()

    44

        # define a unique name for the generated C file

    45

        location_str = "_".join(map(str, current_location))

    46

    47

        text = "## Code generated by Beremiz python mixin confnode\n\n"

    48

    49

        # Adding includes

    50

        text += "## User includes\n"

    51

        text += self.CodeFile.includes.gettext()

    52

        text += "\n"

    53

    54

        # Adding variables

    55

        text += "## User variables reference\n"

    56

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

    57

        for variable in self.CodeFile.variables.variable:

    58

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

    59

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

    60

                global_name,

    61

                variable.getname(),

    62

                variable.gettype(),

    63

                str(variable.getinitial()))

    64

        text += "\n"

    65

    66

        # Adding user global variables and routines

    67

        text += "## User internal user variables and routines\n"

    68

        text += self.CodeFile.globals.gettext()

    69

        text += "\n"

    70

    71

        # Adding Beremiz confnode functions

    72

        text += "## Beremiz confnode functions\n"

    73

        for func, args, return_code, code_object in [

    74

            ("__init_", "*args, **kwargs", 

    75

             "return 0", self.CodeFile.initFunction),

    76

            ("__cleanup_", "", "", self.CodeFile.cleanUpFunction),

    77

            ("__retrieve_", "", "", self.CodeFile.retrieveFunction),

    78

            ("__publish_", "", "", self.CodeFile.publishFunction),]:

    79

            text += "def %s%s(%s):\n" % (func, location_str, args)

    80

            lines = code_object.gettext().splitlines()

    81

            if len(lines) > 0 or return_code != "":

    82

                for line in code_object.gettext().splitlines():

    83

                    text += "    " + line

    84

                text += "    " + return_code + "\n\n"

    85

            else:

    86

                text += "    pass\n\n"

    87

    88

        return text

    89