Skip to content

Commit 5d6fafc

Browse files
committed
Move safer-cluster modules to their own list in autogen system
1 parent 684d89b commit 5d6fafc

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

helpers/generate_modules/generate_modules.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020

2121
from jinja2 import Environment, FileSystemLoader
2222

23-
TEMPLATE_FOLDERS = ["./autogen/main", "./autogen/safer-cluster"]
2423
AUTOGEN_NOTE = '// This file was automatically generated from a template in '
2524

2625

@@ -65,23 +64,26 @@ def template_options(self, base):
6564
'private_cluster': False,
6665
'beta_cluster': True,
6766
}),
67+
]
68+
69+
SAFER_MODULES = [
6870
Module("./modules/safer-cluster", {
6971
'module_path': '//modules/safer-cluster',
70-
'safer_cluster': True,
7172
}),
7273
]
74+
7375
DEVNULL_FILE = open(os.devnull, 'w')
7476

7577

76-
def render_modules(template_folder):
78+
def render_modules(template_folder, modules_list):
7779
env = Environment(
7880
keep_trailing_newline=True,
7981
loader=FileSystemLoader(template_folder),
8082
trim_blocks=True,
8183
lstrip_blocks=True,
8284
)
8385
templates = env.list_templates()
84-
for module in MODULES:
86+
for module in modules_list:
8587
for template_file in templates:
8688
template = env.get_template(template_file)
8789
if template_file.endswith(".tf.tmpl"):
@@ -109,8 +111,9 @@ def render_modules(template_folder):
109111

110112

111113
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+
114117
DEVNULL_FILE.close()
115118

116119

0 commit comments

Comments
 (0)