Move JSON event generators into separate modules
[asterisk/asterisk.git] / rest-api-templates / transform.py
1 #
2 # Asterisk -- An open source telephony toolkit.
3 #
4 # Copyright (C) 2013, Digium, Inc.
5 #
6 # David M. Lee, II <dlee@digium.com>
7 #
8 # See http://www.asterisk.org for more information about
9 # the Asterisk project. Please do not directly contact
10 # any of the maintainers of this project for assistance;
11 # the project provides a web site, mailing lists and IRC
12 # channels for your use.
13 #
14 # This program is free software, distributed under the terms of
15 # the GNU General Public License Version 2. See the LICENSE file
16 # at the top of the source tree.
17 #
18
19 import os.path
20 import pystache
21
22
23 class Transform(object):
24     """Transformation for template to code.
25     """
26     def __init__(self, template_file, dest_file_template_str, overwrite=True):
27         """Ctor.
28
29         @param template_file: Filename of the mustache template.
30         @param dest_file_template_str: Destination file name. This is a
31             mustache template, so each resource can write to a unique file.
32         @param overwrite: If True, destination file is ovewritten if it exists.
33         """
34         template_str = unicode(open(template_file, "r").read())
35         self.template = pystache.parse(template_str)
36         dest_file_template_str = unicode(dest_file_template_str)
37         self.dest_file_template = pystache.parse(dest_file_template_str)
38         self.overwrite = overwrite
39
40     def render(self, renderer, model, dest_dir):
41         """Render a model according to this transformation.
42
43         @param render: Pystache renderer.
44         @param model: Model object to render.
45         @param dest_dir: Destination directory to write generated code.
46         """
47         dest_file = pystache.render(self.dest_file_template, model)
48         dest_file = os.path.join(dest_dir, dest_file)
49         if os.path.exists(dest_file) and not self.overwrite:
50             return
51         print "Rendering %s" % dest_file
52         with open(dest_file, "w") as out:
53             out.write(renderer.render(self.template, model))