Add functions to support recursive validation for extra_vars

This commit is contained in:
Philip Douglass
2022-12-15 09:46:42 -05:00
parent fcf56950b3
commit ad4e257fdb

View File

@@ -856,11 +856,7 @@ class CredentialTypeInjectorField(JSONSchemaField):
template_name = template_name.split('.')[1] template_name = template_name.split('.')[1]
setattr(valid_namespace['tower'].filename, template_name, 'EXAMPLE_FILENAME') setattr(valid_namespace['tower'].filename, template_name, 'EXAMPLE_FILENAME')
for type_, injector in value.items(): def validate_template_string(tmpl):
if type_ == 'env':
for key in injector.keys():
self.validate_env_var_allowed(key)
for key, tmpl in injector.items():
try: try:
sandbox.ImmutableSandboxedEnvironment(undefined=StrictUndefined).from_string(tmpl).render(valid_namespace) sandbox.ImmutableSandboxedEnvironment(undefined=StrictUndefined).from_string(tmpl).render(valid_namespace)
except UndefinedError as e: except UndefinedError as e:
@@ -878,6 +874,24 @@ class CredentialTypeInjectorField(JSONSchemaField):
params={'value': value}, params={'value': value},
) )
def validate_extra_vars(node):
if isinstance(node, dict):
return {validate_extra_vars(k): validate_extra_vars(v) for k, v in node.items()}
elif isinstance(node, list):
return [validate_extra_vars(x) for x in node]
else:
validate_template_string(node)
for type_, injector in value.items():
if type_ == 'env':
for key in injector.keys():
self.validate_env_var_allowed(key)
if type_ == 'extra_vars':
validate_extra_vars(injector)
else:
for key, tmpl in injector.items():
validate_template_string(tmpl)
class AskForField(models.BooleanField): class AskForField(models.BooleanField):
""" """