scripts/invirt-mako-render

   1 #!/usr/bin/env python3
   2 # Modified from mako-render by gdb.
   3 # TODO: set the lookup path dynamically.
   4
   5 def render(data):
   6
   7     lookup = TemplateLookup(['.', '/etc/apache2/conf.invirt'])
   8     return Template(data, lookup=lookup).render()
   9 import sys
  10 import argparse
  11
  12 def main(argv=None):
  13 from mako.template import Template
  14 from mako.lookup import TemplateLookup
  15
  16     if argv is None:
  17         argv = sys.argv
  18
  19
  20     parser = OptionParser("usage: %prog [FILENAME]")
  21
  22     opts, args = parser.parse_args(argv[1:])
  23     if len(args) not in (0, 1):
  24         parser.error("wrong number of arguments") # Will exit
  25
  26     if (len(args) == 0) or (args[0] == "-"):
  27         fo = stdin
  28     else:
  29         filename = args[0]
  30         if not isfile(filename):
  31             raise SystemExit("error: can't find %s" % filename)
  32         fo = open(filename)
  33
  34     data = fo.read()
  35     print render(data)
  36
  37 if __name__ == "__main__":
  38     main()