def render_action_template(cls, entity_context=""):
global RUNBOOK_ACTION_MAP
LOG.debug("Rendering {} action template".format(cls.__name__))
if not isinstance(cls, ActionType):
raise TypeError("{} is not of type {}".format(cls, action))
# Update entity context
# TODO for now, not adding runbook to context as current mapping -is 1:1
entity_context = entity_context + "_Action_" + cls.__name__
runbook = cls.runbook
runbook_name = getattr(runbook, "name", "") or runbook.__name__
# Note cls.__name__ should be used for call_runbook tasks
RUNBOOK_ACTION_MAP[runbook_name] = cls.__name__
# NOTE Not using main_task_local_reference for now,
# bcz type of main task is "DAG"
levelled_tasks = get_task_order(runbook.tasks)
tasks = []
for task_list in levelled_tasks:
if len(task_list) != 1:
tasks.append(
render_parallel_task_template(
task_list, entity_context, RUNBOOK_ACTION_MAP
)
)
else:
tasks.append(
render_task_template(task_list[0], entity_context, RUNBOOK_ACTION_MAP)
)
variables = []
for variable in runbook.variables:
variables.append(render_variable_template(variable, entity_context))
if not (variables or tasks):
return ""
user_attrs = {
"name": cls.__name__,
"description": cls.__doc__ or "",
"tasks": tasks,
"variables": variables,
}
gui_display_name = getattr(cls, "name", "") or cls.__name__
if gui_display_name != cls.__name__:
user_attrs["gui_display_name"] = gui_display_name
text = render_template(schema_file="action.py.jinja2", obj=user_attrs)
return text.strip()
def render_parallel_task_template(task_list, entity_context,
RUNBOOK_ACTION_MAP):
"""render parallel tasks template"""
rendered_tasks = []
for task in task_list:
rendered_tasks.append(
render_task_template(task, entity_context, RUNBOOK_ACTION_MAP))
user_attrs = {"tasks": rendered_tasks}
text = render_template(schema_file="parallel_task.py.jinja2",
obj=user_attrs)
return text.strip()
def render_variable_template(cls, entity_context):
LOG.debug("Rendering {} variable template".format(cls.__name__))
if not isinstance(cls, VariableType):
raise TypeError("{} is not of type {}".format(cls, VariableType))
# Updating the context of variables
entity_context = entity_context + "_variable_" + cls.__name__
user_attrs = cls.get_user_attrs()
user_attrs["description"] = cls.__doc__ or ""
# Escape new line character. As it is inline parameter for CalmVariable helper
user_attrs["description"] = user_attrs["description"].replace("\n", "\\n")
var_val_type = getattr(cls, "value_type", "STRING")
var_type = ""
schema_file = None
if not cls.options:
var_type = "simple"
else:
options = cls.options.get_dict()
choices = options.get("choices", [])
option_type = options.get("type", "")
if (not choices) and (option_type == "PREDEFINED"):
var_type = "simple"
if cls.regex:
regex = cls.regex.get_dict()
user_attrs["regex"] = regex.get("value", None)
user_attrs["validate_regex"] = regex.get("should_validate", False)
else:
user_attrs["regex"] = None
user_attrs["validate_regex"] = False
if cls.editables:
user_attrs["runtime"] = cls.editables["value"]
else:
user_attrs["runtime"] = False
user_attrs["name"] = cls.__name__
if var_type == "simple":
is_secret = True if user_attrs["type"] == "SECRET" else False
if is_secret:
user_attrs["value"] = get_secret_var_val(entity_context)
if var_val_type == "STRING":
schema_file = "var_simple_secret_string.py.jinja2"
elif var_val_type == "INT":
schema_file = "var_simple_secret_int.py.jinja2"
elif var_val_type == "TIME":
schema_file = "var_simple_secret_time.py.jinja2"
elif var_val_type == "DATE":
schema_file = "var_simple_secret_date.py.jinja2"
elif var_val_type == "DATE_TIME":
schema_file = "var_simple_secret_datetime.py.jinja2"
elif var_val_type == "MULTILINE_STRING":
schema_file = "var_simple_secret_multiline.py.jinja2"
else:
if var_val_type == "STRING":
schema_file = "var_simple_string.py.jinja2"
elif var_val_type == "INT":
schema_file = "var_simple_int.py.jinja2"
elif var_val_type == "TIME":
schema_file = "var_simple_time.py.jinja2"
elif var_val_type == "DATE":
schema_file = "var_simple_date.py.jinja2"
elif var_val_type == "DATE_TIME":
schema_file = "var_simple_datetime.py.jinja2"
elif var_val_type == "MULTILINE_STRING":
user_attrs["value"] = repr(user_attrs["value"])
schema_file = "var_simple_multiline.py.jinja2"
else:
data_type = cls.data_type
options = cls.options.get_dict()
option_type = options.get("type", "PREDEFINED")
if option_type == "PREDEFINED":
user_attrs["choices"] = options.get("choices", [])
if data_type == "BASE":
if var_val_type == "STRING":
schema_file = "var_with_options_predefined_string.py.jinja2"
elif var_val_type == "INT":
schema_file = "var_with_options_predefined_int.py.jinja2"
elif var_val_type == "DATE":
schema_file = "var_with_options_predefined_date.py.jinja2"
elif var_val_type == "TIME":
schema_file = "var_with_options_predefined_time.py.jinja2"
elif var_val_type == "DATE_TIME":
schema_file = "var_with_options_predefined_datetime.py.jinja2"
elif var_val_type == "MULTILINE_STRING":
user_attrs["value"] = repr(user_attrs["value"])
schema_file = "var_with_options_predefined_multiline.py.jinja2"
else:
defaults = cls.value
user_attrs["value"] = defaults.split(",")
if var_val_type == "STRING":
schema_file = "var_with_options_predefined_array_string.py.jinja2"
elif var_val_type == "INT":
schema_file = "var_with_options_predefined_array_int.py.jinja2"
elif var_val_type == "DATE":
schema_file = "var_with_options_predefined_array_date.py.jinja2"
elif var_val_type == "TIME":
schema_file = "var_with_options_predefined_array_time.py.jinja2"
elif var_val_type == "DATE_TIME":
schema_file = "var_with_options_predefined_array_datetime.py.jinja2"
elif var_val_type == "MULTILINE_STRING":
user_attrs["value"] = repr(user_attrs["value"])
schema_file = (
"var_with_options_predefined_array_multiline.py.jinja2"
)
else:
options.pop("choices", None)
task = TaskType.decompile(options)
task.__name__ = "SampleTask"
user_attrs["value"] = render_task_template(
task, entity_context=entity_context
)
if data_type == "BASE":
if var_val_type == "STRING":
schema_file = "var_with_options_fromTask_string.py.jinja2"
elif var_val_type == "INT":
schema_file = "var_with_options_fromTask_int.py.jinja2"
elif var_val_type == "DATE":
schema_file = "var_with_options_fromTask_date.py.jinja2"
elif var_val_type == "TIME":
schema_file = "var_with_options_fromTask_time.py.jinja2"
elif var_val_type == "DATE_TIME":
schema_file = "var_with_options_fromTask_datetime.py.jinja2"
elif var_val_type == "MULTILINE_STRING":
schema_file = "var_with_options_fromTask_multiline.py.jinja2"
else:
if var_val_type == "STRING":
schema_file = "var_with_options_fromTask_array_string.py.jinja2"
elif var_val_type == "INT":
schema_file = "var_with_options_fromTask_array_int.py.jinja2"
elif var_val_type == "DATE":
schema_file = "var_with_options_fromTask_array_date.py.jinja2"
elif var_val_type == "TIME":
schema_file = "var_with_options_fromTask_array_time.py.jinja2"
elif var_val_type == "DATE_TIME":
schema_file = "var_with_options_fromTask_array_datetime.py.jinja2"
elif var_val_type == "MULTILINE_STRING":
schema_file = "var_with_options_fromTask_array_multiline.py.jinja2"
if not schema_file:
raise Exception("Unknown variable type")
text = render_template(schema_file=schema_file, obj=user_attrs)
return text.strip()