1 # Copyright 2013 The Chromium Authors. All rights reserved. 2 # Use of this source code is governed by a BSD-style license that can be 3 # found in the LICENSE file. 4 5 # Based on: 6 # http://src.chromium.org/viewvc/blink/trunk/Source/build/scripts/template_expander.py 7 8 import imp 9 import inspect 10 import os.path 11 import sys 12 13 # Disable lint check for finding modules: 14 # pylint: disable=F0401 15 16 def _GetDirAbove(dirname): 17 """Returns the directory "above" this file containing |dirname| (which must 18 also be "above" this file).""" 19 path = os.path.abspath(__file__) 20 while True: 21 path, tail = os.path.split(path) 22 assert tail 23 if tail == dirname: 24 return path 25 26 try: 27 imp.find_module("jinja2") 28 except ImportError: 29 sys.path.append(os.path.join(_GetDirAbove("mojo"), "third_party")) 30 import jinja2 31 32 33 def ApplyTemplate(mojo_generator, base_dir, path_to_template, params, 34 filters=None, **kwargs): 35 template_directory, template_name = os.path.split(path_to_template) 36 path_to_templates = os.path.join(base_dir, template_directory) 37 loader = jinja2.FileSystemLoader([path_to_templates]) 38 final_kwargs = dict(mojo_generator.GetJinjaParameters()) 39 final_kwargs.update(kwargs) 40 jinja_env = jinja2.Environment(loader=loader, keep_trailing_newline=True, 41 **final_kwargs) 42 jinja_env.globals.update(mojo_generator.GetGlobals()) 43 if filters: 44 jinja_env.filters.update(filters) 45 template = jinja_env.get_template(template_name) 46 return template.render(params) 47 48 49 def UseJinja(path_to_template, **kwargs): 50 # Get the directory of our caller's file. 51 base_dir = os.path.dirname(inspect.getfile(sys._getframe(1))) 52 def RealDecorator(generator): 53 def GeneratorInternal(*args, **kwargs2): 54 parameters = generator(*args, **kwargs2) 55 return ApplyTemplate(args[0], base_dir, path_to_template, parameters, 56 **kwargs) 57 GeneratorInternal.func_name = generator.func_name 58 return GeneratorInternal 59 return RealDecorator 60
