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.
106 lines
2.2 KiB
106 lines
2.2 KiB
10 years ago
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
"""Simple Mako renderer.
|
||
|
|
|
||
|
|
Just a wrapper around the mako rendering library.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
import getopt
|
||
|
|
import imp
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
from mako.lookup import TemplateLookup
|
||
|
|
from mako.runtime import Context
|
||
|
|
from mako.template import Template
|
||
|
|
import simplejson
|
||
|
|
import bunch
|
||
|
|
|
||
|
|
|
||
|
|
# Imports a plugin
|
||
|
|
def import_plugin(name):
|
||
|
|
_, base_ex = os.path.split(name)
|
||
|
|
base, _ = os.path.splitext(base_ex)
|
||
|
|
|
||
|
|
with open(name, 'r') as plugin_file:
|
||
|
|
plugin_code = plugin_file.read()
|
||
|
|
plugin_module = imp.new_module(base)
|
||
|
|
exec plugin_code in plugin_module.__dict__
|
||
|
|
return plugin_module
|
||
|
|
|
||
|
|
|
||
|
|
def out(msg):
|
||
|
|
print >> sys.stderr, msg
|
||
|
|
|
||
|
|
|
||
|
|
def showhelp():
|
||
|
|
out('mako-renderer.py [-o out] [-m cache] [-d dict] [-d dict...] template')
|
||
|
|
|
||
|
|
|
||
|
|
def main(argv):
|
||
|
|
got_input = False
|
||
|
|
module_directory = None
|
||
|
|
dictionary = {}
|
||
|
|
json_dict = {}
|
||
|
|
got_output = False
|
||
|
|
output_file = sys.stdout
|
||
|
|
plugins = []
|
||
|
|
|
||
|
|
try:
|
||
|
|
opts, args = getopt.getopt(argv, 'hm:d:o:p:')
|
||
|
|
except getopt.GetoptError:
|
||
|
|
out('Unknown option')
|
||
|
|
showhelp()
|
||
|
|
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_file = open(arg, 'w')
|
||
|
|
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 == '-d':
|
||
|
|
dict_file = open(arg, 'r')
|
||
|
|
bunch.merge_json(json_dict, simplejson.loads(dict_file.read()))
|
||
|
|
dict_file.close()
|
||
|
|
elif opt == '-p':
|
||
|
|
plugins.append(import_plugin(arg))
|
||
|
|
|
||
|
|
for plugin in plugins:
|
||
|
|
plugin.mako_plugin(json_dict)
|
||
|
|
|
||
|
|
for k, v in json_dict.items():
|
||
|
|
dictionary[k] = bunch.to_bunch(v)
|
||
|
|
|
||
|
|
ctx = Context(output_file, **dictionary)
|
||
|
|
|
||
|
|
for arg in args:
|
||
|
|
got_input = True
|
||
|
|
template = Template(filename=arg,
|
||
|
|
module_directory=module_directory,
|
||
|
|
lookup=TemplateLookup(directories=['.']))
|
||
|
|
template.render_context(ctx)
|
||
|
|
|
||
|
|
if not got_input:
|
||
|
|
out('Got nothing to do')
|
||
|
|
showhelp()
|
||
|
|
|
||
|
|
output_file.close()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main(sys.argv[1:])
|