9

from CodeFileTreeNode import CodeFile

    10

    13

class PythonFileCTNMixin(CodeFile):

    18

        CodeFile.__init__(self)

    22

        python_code = PythonClasses["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

    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

        for variable in self.CodeFile.variables.variable:

    57

            text += "%s = %s\n" % (variable.getname(),

    58

                                   str(variable.getinitial()))

    59

        text += "\n"

    60

    61

        # Adding user global variables and routines

    62

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

    63

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

    64

        text += "\n"

    65

    66

        # Adding Beremiz confnode functions

    67

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

    68

        for func, args, return_code, code_object in [

    69

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

    70

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

    71

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

    72

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

    73

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

    74

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

    75

            lines = code_object.gettext().splitlines()

    76

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

    77

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

    78

                    text += "    " + line

    79

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

    80

            else:

    81

                text += "    pass\n\n"

    82

    83

        return text