ari: Fix #include to match generated headers for snakeCase resource files
[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 filecmp
20 import os.path
21 import pystache
22 import shutil
23 import tempfile
24
25
26 class Transform(object):
27     """Transformation for template to code.
28     """
29     def __init__(self, template_file, dest_file_template_str, overwrite=True):
30         """Ctor.
31
32         @param template_file: Filename of the mustache template.
33         @param dest_file_template_str: Destination file name. This is a
34             mustache template, so each resource can write to a unique file.
35         @param overwrite: If True, destination file is ovewritten if it exists.
36         """
37         template_str = unicode(open(template_file, "r").read())
38         self.template = pystache.parse(template_str)
39         dest_file_template_str = unicode(dest_file_template_str)
40         self.dest_file_template = pystache.parse(dest_file_template_str)
41         self.overwrite = overwrite
42
43     def render(self, renderer, model, dest_dir):
44         """Render a model according to this transformation.
45
46         @param render: Pystache renderer.
47         @param model: Model object to render.
48         @param dest_dir: Destination directory to write generated code.
49         """
50         dest_file = pystache.render(self.dest_file_template, model)
51         dest_file = os.path.join(dest_dir, dest_file)
52         dest_exists = os.path.exists(dest_file)
53         if dest_exists and not self.overwrite:
54             return
55         tmp_file = tempfile.mkstemp()
56         with tempfile.NamedTemporaryFile() as out:
57             out.write(renderer.render(self.template, model))
58             out.flush()
59
60             if not dest_exists or not filecmp.cmp(out.name, dest_file):
61                 print "Writing %s" % dest_file
62                 shutil.copyfile(out.name, dest_file)