1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

# This file is part of Beremiz.

     4

# See COPYING file for copyrights details.

     5

     6

from __future__ import absolute_import

     7

import os

     8

from lxml import etree

     9

import util.paths as paths

    10

from plcopen.structures import StdBlckLibs

    11

    12

ScriptDirectory = paths.AbsDir(__file__)

    13

    14

    15

class XSLTModelQuery(object):

    16

    """ a class to handle XSLT queries on project and libs """

    17

    def __init__(self, controller, xsltpath, ext=None):

    18

        # arbitrary set debug to false, updated later

    19

        self.debug = False

    20

    21

        # merge xslt extensions for library access to query specific ones

    22

        xsltext = [

    23

            ("GetProject", lambda *_ignored:

    24

             [controller.GetProject(self.debug)]),

    25

            ("GetStdLibs", lambda *_ignored:

    26

             [lib for lib in StdBlckLibs.values()]),

    27

            ("GetExtensions", lambda *_ignored:

    28

             [ctn["types"] for ctn in controller.ConfNodeTypes])

    29

        ]

    30

    31

        if ext is not None:

    32

            xsltext.extend(ext)

    33

    34

        # parse and compile. "beremiz" arbitrary namespace for extensions

    35

        self.xslt = etree.XSLT(

    36

            etree.parse(

    37

                os.path.join(ScriptDirectory, xsltpath),

    38

                etree.XMLParser()),

    39

            extensions={("beremiz", name): call for name, call in xsltext})

    40

    41

    def _process_xslt(self, root, debug, **kwargs):

    42

        self.debug = debug

    43

        res = self.xslt(root, **{k: etree.XSLT.strparam(v) for k, v in kwargs.iteritems()})

    44

        # print(self.xslt.error_log)

    45

        return res

    46

    47

    48

# -------------------------------------------------------------------------------

    49

#           Helpers functions for translating list of arguments

    50

#                       from xslt to valid arguments

    51

# -------------------------------------------------------------------------------

    52

    53

    54

def _StringValue(x):

    55

    return x

    56

    57

    58

def _BoolValue(x):

    59

    return x in ["true", "0"]

    60

    61

    62

def _translate_args(translations, args):

    63

    return [translate(arg[0]) if len(arg) > 0 else None

    64

            for translate, arg in

    65

            zip(translations, args)]