1

import os

     2

from POULibrary import POULibrary

     3

from PythonFileCTNMixin import PythonFileCTNMixin

     4

from util.misc import opjimg

     5

     6

class PythonLibrary(POULibrary):

     7

    def GetLibraryPath(self):

     8

        return os.path.join(os.path.split(__file__)[0], "pous.xml") 

     9

    10

    def Generate_C(self, buildpath, varlist, IECCFLAGS):

    11

    12

        plc_python_filepath = os.path.join(os.path.split(__file__)[0], "plc_python.c")

    13

        plc_python_file = open(plc_python_filepath, 'r')

    14

        plc_python_code = plc_python_file.read()

    15

        plc_python_file.close()

    16

        python_eval_fb_list = []

    17

        for v in varlist:

    18

            if v["vartype"] == "FB" and v["type"] in ["PYTHON_EVAL","PYTHON_POLL"]:

    19

                python_eval_fb_list.append(v)

    20

        python_eval_fb_count = max(1, len(python_eval_fb_list))

    21

    22

        # prepare python code

    23

        plc_python_code = plc_python_code % { "python_eval_fb_count": python_eval_fb_count }

    24

    25

        Gen_Pythonfile_path = os.path.join(buildpath, "py_ext.c")

    26

        pythonfile = open(Gen_Pythonfile_path,'w')

    27

        pythonfile.write(plc_python_code)

    28

        pythonfile.close()

    29

    30

        return (["py_ext"], [(Gen_Pythonfile_path, IECCFLAGS)], True), ""

    31

    32

class PythonFile(PythonFileCTNMixin):

    33

    34

    def GetIconPath(self):

    35

        return opjimg("Pyfile")

    36

    37

    def CTNGenerate_C(self, buildpath, locations):

    38

        current_location = self.GetCurrentLocation()

    39

        # define a unique name for the generated C file

    40

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

    41

    42

        runtimefile_path = os.path.join(buildpath, "runtime_%s.py"%location_str)

    43

        runtimefile = open(runtimefile_path, 'w')

    44

        runtimefile.write(self.GetPythonCode())

    45

        runtimefile.close()

    46

    47

        return [], "", False, ("runtime_%s.py"%location_str, file(runtimefile_path,"rb"))

    48