robotframework-onap/eteutils/TemplatingEngine.py

   1 # Copyright 2019 AT&T Intellectual Property. All rights reserved.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #         http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15
  16 from jinja2 import Environment, FileSystemLoader, select_autoescape
  17 from robot import utils
  18
  19
  20 class TemplatingEngine(object):
  21     """TemplateImporter is common resource for templating with strings."""
  22
  23     def __init__(self):
  24         self._cache = utils.ConnectionCache('No Jinja Environments created')
  25
  26     def create_environment(self, alias, templates_folder):
  27         jinja_env = Environment(
  28             loader=FileSystemLoader(templates_folder),
  29             autoescape=select_autoescape(['html', 'xml'])
  30         )
  31         self._cache.register(jinja_env, alias=alias)
  32
  33     def apply_template(self, alias, template_location, values):
  34         """returns a string that is the jinja template in template_location filled in via the dictionary in values """
  35         template = self._cache.switch(alias).get_template(template_location)
  36         return template.render(values)