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

from XSLTransform import XSLTransform

    12

    13

ScriptDirectory = paths.AbsDir(__file__)

    14

    15

class XSLTModelQuery(XSLTransform):

    16

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

    17

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

    18

    19

        # arbitrary set debug to false, updated later

    20

        self.debug = False

    21

    22

        # merge xslt extensions for library access to query specific ones

    23

        xsltext = [

    24

            ("GetProject", lambda *_ignored:

    25

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

    26

            ("GetStdLibs", lambda *_ignored:

    27

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

    28

            ("GetExtensions", lambda *_ignored:

    29

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

    30

        ]

    31

    32

        if ext is not None:

    33

            xsltext.extend(ext)

    34

    35

        XSLTransform.__init__(self,

    36

                           os.path.join(ScriptDirectory, xsltpath),

    37

                           xsltext)

    38

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

    39

        self.debug = debug

    40

        return self.transform(root, **kwargs)

    41

    42

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

    43

#           Helpers functions for translating list of arguments

    44

#                       from xslt to valid arguments

    45

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

    46

    47

    48

def _StringValue(x):

    49

    return x

    50

    51

    52

def _BoolValue(x):

    53

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

    54

    55

    56

def _translate_args(translations, args):

    57

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

    58

            for translate, arg in

    59

            zip(translations, args)]