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 class XSLTModelQuery(object):

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

    16     def __init__(self, controller, xsltpath, ext = []):

    17         # arbitrary set debug to false, updated later

    18         self.debug = False

    19 

    20         # merge xslt extensions for library access to query specific ones

    21         xsltext = [

    22             ("GetProject", lambda *_ignored: 

    23                 controller.GetProject(self.debug)),

    24             ("GetStdLibs", lambda *_ignored: 

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

    26             ("GetExtensions", lambda *_ignored: 

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

    28         ] + ext

    29 

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

    31         self.xslt = etree.XSLT(

    32             etree.parse(

    33                 os.path.join(ScriptDirectory, xsltpath),

    34                 etree.XMLParser()),

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

    36 

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

    38         self.debug = debug

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