ba745845d919574752615ddbd2a78c13b8bfd377
[invirt/packages/invirt-base.git] / 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 import sys
6 import argparse
7
8 from mako.template import Template
9 from mako.lookup import TemplateLookup
10
11
12 def render(data):
13     lookup = TemplateLookup(['.', '/etc/apache2/conf.invirt'])
14     return Template(data, lookup=lookup).render()
15
16 def main():
17     parser = argparse.ArgumentParser(description='Render mako template')
18     parser.add_argument('filename', nargs='?', default=None)
19
20     args = parser.parse_args()
21
22     if args.filename == "-" or args.filename is None:
23         fo = sys.stdin
24     else:
25         fo = open(args.filename)
26
27     data = fo.read()
28     print(render(data))
29
30
31 if __name__ == "__main__":
32     main()