1 # mako/cmd.py 2 # Copyright (C) 2006-2015 the Mako authors and contributors <see AUTHORS file> 3 # 4 # This module is part of Mako and is released under 5 # the MIT License: http://www.opensource.org/licenses/mit-license.php 6 from argparse import ArgumentParser 7 from os.path import isfile, dirname 8 import sys 9 from mako.template import Template 10 from mako.lookup import TemplateLookup 11 from mako import exceptions 12 13 def varsplit(var): 14 if "=" not in var: 15 return (var, "") 16 return var.split("=", 1) 17 18 def _exit(): 19 sys.stderr.write(exceptions.text_error_template().render()) 20 sys.exit(1) 21 22 def cmdline(argv=None): 23 24 parser = ArgumentParser("usage: %prog [FILENAME]") 25 parser.add_argument("--var", default=[], action="append", 26 help="variable (can be used multiple times, use name=value)") 27 parser.add_argument("--template-dir", default=[], action="append", 28 help="Directory to use for template lookup (multiple " 29 "directories may be provided). If not given then if the " 30 "template is read from stdin, the value defaults to be " 31 "the current directory, otherwise it defaults to be the " 32 "parent directory of the file provided.") 33 parser.add_argument('input', nargs='?', default='-') 34 35 options = parser.parse_args(argv) 36 if options.input == '-': 37 lookup_dirs = options.template_dir or ["."] 38 lookup = TemplateLookup(lookup_dirs) 39 try: 40 template = Template(sys.stdin.read(), lookup=lookup) 41 except: 42 _exit() 43 else: 44 filename = options.input 45 if not isfile(filename): 46 raise SystemExit("error: can't find %s" % filename) 47 lookup_dirs = options.template_dir or [dirname(filename)] 48 lookup = TemplateLookup(lookup_dirs) 49 try: 50 template = Template(filename=filename, lookup=lookup) 51 except: 52 _exit() 53 54 kw = dict([varsplit(var) for var in options.var]) 55 try: 56 print(template.render(**kw)) 57 except: 58 _exit() 59 60 61 if __name__ == "__main__": 62 cmdline() 63
