|
20 | 20 |
|
21 | 21 | from jinja2 import Environment, FileSystemLoader
|
22 | 22 |
|
23 |
| -TEMPLATE_FOLDERS = ["./autogen/main", "./autogen/safer-cluster"] |
24 | 23 | AUTOGEN_NOTE = '// This file was automatically generated from a template in '
|
25 | 24 |
|
26 | 25 |
|
@@ -65,23 +64,26 @@ def template_options(self, base):
|
65 | 64 | 'private_cluster': False,
|
66 | 65 | 'beta_cluster': True,
|
67 | 66 | }),
|
| 67 | +] |
| 68 | + |
| 69 | +SAFER_MODULES = [ |
68 | 70 | Module("./modules/safer-cluster", {
|
69 | 71 | 'module_path': '//modules/safer-cluster',
|
70 |
| - 'safer_cluster': True, |
71 | 72 | }),
|
72 | 73 | ]
|
| 74 | + |
73 | 75 | DEVNULL_FILE = open(os.devnull, 'w')
|
74 | 76 |
|
75 | 77 |
|
76 |
| -def render_modules(template_folder): |
| 78 | +def render_modules(template_folder, modules_list): |
77 | 79 | env = Environment(
|
78 | 80 | keep_trailing_newline=True,
|
79 | 81 | loader=FileSystemLoader(template_folder),
|
80 | 82 | trim_blocks=True,
|
81 | 83 | lstrip_blocks=True,
|
82 | 84 | )
|
83 | 85 | templates = env.list_templates()
|
84 |
| - for module in MODULES: |
| 86 | + for module in modules_list: |
85 | 87 | for template_file in templates:
|
86 | 88 | template = env.get_template(template_file)
|
87 | 89 | if template_file.endswith(".tf.tmpl"):
|
@@ -109,8 +111,9 @@ def render_modules(template_folder):
|
109 | 111 |
|
110 | 112 |
|
111 | 113 | def main(argv):
|
112 |
| - for template_folder in TEMPLATE_FOLDERS: |
113 |
| - render_modules(template_folder) |
| 114 | + render_modules("./autogen/main", MODULES) |
| 115 | + render_modules("./autogen/safer-cluster", SAFER_MODULES) |
| 116 | + |
114 | 117 | DEVNULL_FILE.close()
|
115 | 118 |
|
116 | 119 |
|
|
0 commit comments