mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
173 lines
5.8 KiB
173 lines
5.8 KiB
4 years ago
|
#!/usr/bin/env python3
|
||
8 years ago
|
# Copyright 2015 gRPC authors.
|
||
10 years ago
|
#
|
||
8 years ago
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
# you may not use this file except in compliance with the License.
|
||
|
# You may obtain a copy of the License at
|
||
10 years ago
|
#
|
||
8 years ago
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
10 years ago
|
#
|
||
8 years ago
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
# See the License for the specific language governing permissions and
|
||
|
# limitations under the License.
|
||
10 years ago
|
"""Simple Mako renderer.
|
||
|
|
||
|
Just a wrapper around the mako rendering library.
|
||
|
"""
|
||
|
|
||
|
import getopt
|
||
4 years ago
|
import glob
|
||
4 years ago
|
import importlib.util
|
||
10 years ago
|
import os
|
||
4 years ago
|
import pickle
|
||
9 years ago
|
import shutil
|
||
10 years ago
|
import sys
|
||
4 years ago
|
from typing import List
|
||
10 years ago
|
|
||
4 years ago
|
import yaml
|
||
4 years ago
|
from mako import exceptions
|
||
10 years ago
|
from mako.lookup import TemplateLookup
|
||
|
from mako.runtime import Context
|
||
|
from mako.template import Template
|
||
4 years ago
|
|
||
4 years ago
|
PROJECT_ROOT = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..",
|
||
|
"..")
|
||
|
# TODO(lidiz) find a better way for plugins to reference each other
|
||
|
sys.path.append(os.path.join(PROJECT_ROOT, 'tools', 'buildgen', 'plugins'))
|
||
10 years ago
|
|
||
|
|
||
4 years ago
|
def out(msg: str) -> None:
|
||
4 years ago
|
print(msg, file=sys.stderr)
|
||
10 years ago
|
|
||
|
|
||
4 years ago
|
def showhelp() -> None:
|
||
7 years ago
|
out('mako-renderer.py [-o out] [-m cache] [-P preprocessed_input] [-d dict] [-d dict...]'
|
||
|
' [-t template] [-w preprocessed_output]')
|
||
10 years ago
|
|
||
|
|
||
4 years ago
|
def render_template(template: Template, context: Context) -> None:
|
||
|
"""Render the mako template with given context.
|
||
|
|
||
|
Prints an error template to indicate where and what in the template caused
|
||
|
the render failure.
|
||
|
"""
|
||
|
try:
|
||
|
template.render_context(context)
|
||
|
except:
|
||
|
out(exceptions.text_error_template().render())
|
||
|
raise
|
||
|
|
||
|
|
||
|
def main(argv: List[str]) -> None:
|
||
7 years ago
|
got_input = False
|
||
|
module_directory = None
|
||
|
preprocessed_output = None
|
||
|
dictionary = {}
|
||
|
json_dict = {}
|
||
|
got_output = False
|
||
|
output_name = None
|
||
|
got_preprocessed_input = False
|
||
|
output_merged = None
|
||
|
|
||
|
try:
|
||
4 years ago
|
opts, args = getopt.getopt(argv, 'hM:m:o:t:P:')
|
||
7 years ago
|
except getopt.GetoptError:
|
||
|
out('Unknown option')
|
||
10 years ago
|
showhelp()
|
||
7 years ago
|
sys.exit(2)
|
||
|
|
||
|
for opt, arg in opts:
|
||
|
if opt == '-h':
|
||
|
out('Displaying showhelp')
|
||
|
showhelp()
|
||
|
sys.exit()
|
||
|
elif opt == '-o':
|
||
|
if got_output:
|
||
|
out('Got more than one output')
|
||
|
showhelp()
|
||
|
sys.exit(3)
|
||
|
got_output = True
|
||
|
output_name = arg
|
||
|
elif opt == '-m':
|
||
|
if module_directory is not None:
|
||
|
out('Got more than one cache directory')
|
||
|
showhelp()
|
||
|
sys.exit(4)
|
||
|
module_directory = arg
|
||
|
elif opt == '-M':
|
||
|
if output_merged is not None:
|
||
|
out('Got more than one output merged path')
|
||
|
showhelp()
|
||
|
sys.exit(5)
|
||
|
output_merged = arg
|
||
|
elif opt == '-P':
|
||
|
assert not got_preprocessed_input
|
||
|
assert json_dict == {}
|
||
4 years ago
|
with open(arg, 'rb') as dict_file:
|
||
7 years ago
|
dictionary = pickle.load(dict_file)
|
||
|
got_preprocessed_input = True
|
||
|
|
||
|
cleared_dir = False
|
||
|
for arg in args:
|
||
|
got_input = True
|
||
|
with open(arg) as f:
|
||
4 years ago
|
srcs = list(yaml.load_all(f.read(), Loader=yaml.FullLoader))
|
||
7 years ago
|
for src in srcs:
|
||
4 years ago
|
if isinstance(src, str):
|
||
7 years ago
|
assert len(srcs) == 1
|
||
5 years ago
|
template = Template(src,
|
||
|
filename=arg,
|
||
|
module_directory=module_directory,
|
||
|
lookup=TemplateLookup(directories=['.']))
|
||
7 years ago
|
with open(output_name, 'w') as output_file:
|
||
4 years ago
|
render_template(template, Context(output_file,
|
||
|
**dictionary))
|
||
7 years ago
|
else:
|
||
|
# we have optional control data: this template represents
|
||
|
# a directory
|
||
|
if not cleared_dir:
|
||
|
if not os.path.exists(output_name):
|
||
|
pass
|
||
|
elif os.path.isfile(output_name):
|
||
|
os.unlink(output_name)
|
||
|
else:
|
||
|
shutil.rmtree(output_name, ignore_errors=True)
|
||
|
cleared_dir = True
|
||
|
items = []
|
||
|
if 'foreach' in src:
|
||
|
for el in dictionary[src['foreach']]:
|
||
|
if 'cond' in src:
|
||
|
args = dict(dictionary)
|
||
|
args['selected'] = el
|
||
|
if not eval(src['cond'], {}, args):
|
||
|
continue
|
||
|
items.append(el)
|
||
|
assert items
|
||
|
else:
|
||
|
items = [None]
|
||
|
for item in items:
|
||
|
args = dict(dictionary)
|
||
|
args['selected'] = item
|
||
|
item_output_name = os.path.join(
|
||
|
output_name,
|
||
|
Template(src['output_name']).render(**args))
|
||
|
if not os.path.exists(os.path.dirname(item_output_name)):
|
||
|
os.makedirs(os.path.dirname(item_output_name))
|
||
|
template = Template(
|
||
|
src['template'],
|
||
|
filename=arg,
|
||
|
module_directory=module_directory,
|
||
|
lookup=TemplateLookup(directories=['.']))
|
||
|
with open(item_output_name, 'w') as output_file:
|
||
4 years ago
|
render_template(template, Context(output_file, **args))
|
||
7 years ago
|
|
||
|
if not got_input and not preprocessed_output:
|
||
|
out('Got nothing to do')
|
||
10 years ago
|
showhelp()
|
||
7 years ago
|
|
||
10 years ago
|
|
||
|
if __name__ == '__main__':
|
||
7 years ago
|
main(sys.argv[1:])
|